klmp200/Sith
1
0
Fork 0
mirror of https://github.com/ae-utbm/sith.git synced 2025-04-17 03:20:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

better error handling and corrections

Browse Source
This commit is contained in:
Kenneth SOARES 2025-04-09 23:53:18 +02:00
parent 9c27055a5e
commit 8ab9429774
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
core/management/commands/add_promo_logo.py
View File

@ -2,7 +2,7 @@ import pathlib
from django.apps import apps from django.apps import apps
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from PIL import Image from PIL import Image, UnidentifiedImageError
class Command(BaseCommand): class Command(BaseCommand):
@ -13,7 +13,7 @@ class Command(BaseCommand):
def handle(self, number: int, path: pathlib.Path, force: int, *args, **options): def handle(self, number: int, path: pathlib.Path, force: int, *args, **options):
if not path.exists() or path.is_dir(): if not path.exists() or path.is_dir():
self.stdout.write("input file does not exist.") self.stderr.write(f"{path} is not a file or does not exist")
return return
dest_path = ( dest_path = (
@ -27,13 +27,23 @@ class Command(BaseCommand):
if dest_path.exists() and not force: if dest_path.exists() and not force:
over = input("File already exists, do you want to overwrite it? (y/N):") over = input("File already exists, do you want to overwrite it? (y/N):")
if over.lower() != "y": if over.lower() != "y":
self.stdout.write("exiting")
return return
try: try:
with Image.open(path) as im: im = Image.open(path)
try:
im.resize((120, 120)).save(dest_path, format="PNG") im.resize((120, 120)).save(dest_path, format="PNG")
self.stdout.write(
f"Promo logo moved and resized successfully at {dest_path}"
)
except IOError as ioe:
self.stderr.write(ioe)
self.stdout.write( except FileNotFoundError:
f"Promo logo moved and resized successfully at {dest_path}" self.stderr.write("image file not found")
) except TypeError:
except IOError as ioe: self.stderr.write("wrong image format")
self.stderr.write(ioe) except ValueError:
self.stderr.write("wrong mode or fp")
except UnidentifiedImageError:
self.stderr.write("image cannot be opened and identified.")