TheLovinator/ttvdrops
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update import_drops command to accept multiple JSON file paths

Browse source
This commit is contained in:
Joakim Hellsén 2025-09-05 01:04:30 +02:00
commit 97df0b342b
1 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

31
twitch/management/commands/import_drops.py
View file

@ -58,7 +58,8 @@ class Command(BaseCommand):
parser: The command argument parser. parser: The command argument parser.
""" """
parser.add_argument( parser.add_argument(
"path", "paths",
nargs="+",
type=str, type=str,
help="Path to the JSON file or directory containing JSON files.", help="Path to the JSON file or directory containing JSON files.",
) )
@ -79,23 +80,25 @@ class Command(BaseCommand):
CommandError: If the file/directory doesn't exist, isn't a JSON file, CommandError: If the file/directory doesn't exist, isn't a JSON file,
or has an invalid JSON structure. or has an invalid JSON structure.
""" """
path: Path = Path(options["path"]) paths: list[str] = options["paths"]
processed_dir: str = options["processed_dir"] processed_dir: str = options["processed_dir"]
processed_path: Path = path / processed_dir for p in paths:
processed_path.mkdir(exist_ok=True) path: Path = Path(p)
processed_path: Path = path / processed_dir
processed_path.mkdir(exist_ok=True)
if not path.exists(): if not path.exists():
msg: str = f"Path {path} does not exist" msg: str = f"Path {path} does not exist"
raise CommandError(msg) raise CommandError(msg)
if path.is_file(): if path.is_file():
self._process_file(file_path=path, processed_path=processed_path) self._process_file(file_path=path, processed_path=processed_path)
elif path.is_dir(): elif path.is_dir():
self._process_directory(directory=path, processed_path=processed_path) self._process_directory(directory=path, processed_path=processed_path)
else: else:
msg = f"Path {path} is neither a file nor a directory" msg = f"Path {path} is neither a file nor a directory"
raise CommandError(msg) raise CommandError(msg)
def _process_directory(self, directory: Path, processed_path: Path) -> None: def _process_directory(self, directory: Path, processed_path: Path) -> None:
"""Process all JSON files in a directory using parallel processing. """Process all JSON files in a directory using parallel processing.