67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
from pathlib import Path
|
|
|
|
from django.core.files import File
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from umap.models import Pictogram
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Import pictograms from a folder"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument("path")
|
|
parser.add_argument(
|
|
"--attribution",
|
|
required=True,
|
|
help="Attribution of the imported pictograms",
|
|
)
|
|
parser.add_argument(
|
|
"--extensions",
|
|
help="Optional list of extensins to process",
|
|
nargs="+",
|
|
default=[".svg"],
|
|
)
|
|
parser.add_argument(
|
|
"--exclude",
|
|
help="Optional list of files or dirs to exclude",
|
|
nargs="+",
|
|
default=["font"],
|
|
)
|
|
parser.add_argument(
|
|
"--force", action="store_true", help="Update picto if it already exists."
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
self.path = Path(options["path"])
|
|
self.attribution = options["attribution"]
|
|
self.extensions = options["extensions"]
|
|
self.force = options["force"]
|
|
self.exclude = options["exclude"]
|
|
self.handle_directory(self.path)
|
|
|
|
def handle_directory(self, path):
|
|
for filename in path.iterdir():
|
|
if filename.name in self.exclude:
|
|
continue
|
|
if filename.is_dir():
|
|
self.handle_directory(filename)
|
|
continue
|
|
if filename.suffix in self.extensions:
|
|
name = filename.stem
|
|
picto = Pictogram.objects.filter(name=name).last()
|
|
if picto:
|
|
if not self.force:
|
|
self.stdout.write(
|
|
f"⚠ Pictogram with name '{name}' already exists. Skipping."
|
|
)
|
|
continue
|
|
else:
|
|
picto = Pictogram()
|
|
picto.name = name
|
|
if path.name != self.path.name: # Subfolders only
|
|
picto.category = path.name
|
|
picto.attribution = self.attribution
|
|
with (filename).open("rb") as f:
|
|
picto.pictogram.save(filename.name, File(f), save=True)
|
|
self.stdout.write(f"✔ Imported pictogram {filename}.")
|