from django.core.management.base import BaseCommand, CommandError
import os
import shutil
from django.conf import settings


class Command(BaseCommand):
    help = 'Clear log files'

    def add_arguments(self, parser):
        parser.add_argument(
            '--log_name',
            type=str,
            default='__all__',
            dest='log_name',
            help='указать имя конкретного файла'
        )

    def handle(self, *args, **options):
        if options['log_name'] == '__all__':
            try:
                shutil.rmtree(settings.BASE_DIR + '/logs')
            except FileNotFoundError:
                pass

            os.mkdir(settings.BASE_DIR + '/logs')

        else:
            os.remove(settings.BASE_DIR + '/logs/' + options['log_name'])