small fixes and tests
This commit is contained in:
		@ -65,31 +65,32 @@ class Command (BaseCommand):
 | 
			
		||||
 | 
			
		||||
    def add_arguments (self, parser):
 | 
			
		||||
        parser.formatter_class=RawTextHelpFormatter
 | 
			
		||||
 | 
			
		||||
        now = tz.datetime.today()
 | 
			
		||||
 | 
			
		||||
        group = parser.add_argument_group('action')
 | 
			
		||||
        group.add_argument(
 | 
			
		||||
            '--update', action='store_true',
 | 
			
		||||
            help = 'generate (unconfirmed) diffusions for the given month. '
 | 
			
		||||
                   'These diffusions must be confirmed manually by changing '
 | 
			
		||||
                   'their type to "normal"')
 | 
			
		||||
            help='generate (unconfirmed) diffusions for the given month. '
 | 
			
		||||
                 'These diffusions must be confirmed manually by changing '
 | 
			
		||||
                 'their type to "normal"')
 | 
			
		||||
        group.add_argument(
 | 
			
		||||
            '--clean', action='store_true',
 | 
			
		||||
            help = 'remove unconfirmed diffusions older than the given month')
 | 
			
		||||
            help='remove unconfirmed diffusions older than the given month')
 | 
			
		||||
 | 
			
		||||
        group.add_argument(
 | 
			
		||||
            '--check', action='store_true',
 | 
			
		||||
            help = 'check future unconfirmed diffusions from the given date '
 | 
			
		||||
                   'agains\'t schedules and remove it if that do not match any '
 | 
			
		||||
                   'schedule')
 | 
			
		||||
            help='check future unconfirmed diffusions from the given date '
 | 
			
		||||
                 'agains\'t schedules and remove it if that do not match any '
 | 
			
		||||
                 'schedule')
 | 
			
		||||
 | 
			
		||||
        group = parser.add_argument_group(
 | 
			
		||||
            'date')
 | 
			
		||||
        group.add_argument('--year', type=int, default=now.year,
 | 
			
		||||
                            help='used by update, default is today\'s year')
 | 
			
		||||
        group.add_argument('--month', type=int, default=now.month,
 | 
			
		||||
                            help='used by update, default is today\'s month')
 | 
			
		||||
        group.add_argument(
 | 
			
		||||
            '--year', type=int, default=now.year,
 | 
			
		||||
            help='used by update, default is today\'s year')
 | 
			
		||||
        group.add_argument(
 | 
			
		||||
            '--month', type=int, default=now.month,
 | 
			
		||||
            help='used by update, default is today\'s month')
 | 
			
		||||
 | 
			
		||||
    def handle (self, *args, **options):
 | 
			
		||||
        date = tz.datetime(year = options.get('year'),
 | 
			
		||||
 | 
			
		||||
@ -48,7 +48,8 @@ class Command (BaseCommand):
 | 
			
		||||
        )
 | 
			
		||||
        parser.add_argument(
 | 
			
		||||
            '-s', '--scan', action='store_true',
 | 
			
		||||
            help='Scan programs directories for changes'
 | 
			
		||||
            help='Scan programs directories for changes, plus check for a '
 | 
			
		||||
                 ' matching episode on sounds that have not been yet assigned'
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -101,6 +102,7 @@ class Command (BaseCommand):
 | 
			
		||||
            self.report(program, path, 'no diffusion found for the given date')
 | 
			
		||||
            return
 | 
			
		||||
        diffusion = diffusion[0]
 | 
			
		||||
        print(diffusion, sound_info)
 | 
			
		||||
        return diffusion.episode or None
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
@ -163,7 +165,7 @@ class Command (BaseCommand):
 | 
			
		||||
                        episode.sounds.add(sound)
 | 
			
		||||
                        episode.save()
 | 
			
		||||
 | 
			
		||||
        self.check_sounds(Sound.objects.filter(path__startswith == subdir))
 | 
			
		||||
        self.check_sounds(Sound.objects.filter(path__startswith = subdir))
 | 
			
		||||
 | 
			
		||||
    def check_quality (self, check = False):
 | 
			
		||||
        """
 | 
			
		||||
@ -179,7 +181,6 @@ class Command (BaseCommand):
 | 
			
		||||
        else:
 | 
			
		||||
            files = [ sound.path for sound in sounds.filter(removed = False) ]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
        print('start quality check...')
 | 
			
		||||
        cmd = quality_check.Command()
 | 
			
		||||
        cmd.handle( files = files,
 | 
			
		||||
 | 
			
		||||
		Reference in New Issue
	
	Block a user