mirror of
https://github.com/ajurna/cbwebreader.git
synced 2025-12-06 14:17:19 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from typing import Optional, Union
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import User
|
|
from django.core.management.base import BaseCommand, CommandParser
|
|
from loguru import logger
|
|
|
|
from comic.models import ComicBook, Directory
|
|
from comic.processing import generate_directory
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Scan directories to Update Comic DB"
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.OUTPUT = False
|
|
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
parser.add_argument(
|
|
'--out',
|
|
action='store_true',
|
|
help='Output to console',
|
|
)
|
|
|
|
def handle(self, *args, **options) -> None:
|
|
self.OUTPUT = options.get('out', False)
|
|
self.scan_directory()
|
|
|
|
def scan_directory(self, user: Optional[User] = None, directory: Optional[Directory] = None) -> None:
|
|
if not user:
|
|
user_model = get_user_model()
|
|
user = user_model.objects.first()
|
|
for item in generate_directory(user, directory):
|
|
item: Union[Directory, ComicBook]
|
|
if item.type == 'Directory':
|
|
logger.info(item)
|
|
self.scan_directory(user, item)
|