uid.management.commands.truncate_image_tables

Complexity

Total Complexity

3

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	Command.handle()	0	14	2
A	Command.add_arguments()	0	7	1

import logging

from django.core.management import BaseCommand

from uid.models import truncate_database, truncate_filled_tables

# Get an instance of a logger
logger = logging.getLogger(__name__)


class Command(BaseCommand):
    help = """Truncate uid tables and reset counters"""

    def add_arguments(self, parser):
        # Named (optional) arguments
        parser.add_argument(
            '--all',
            action='store_true',
            dest='all',
            help='Truncate all uid tables and reset counters',
        )

    def handle(self, *args, **options):
        # HINT: maybe UID or InjectTools could be more informative than
        # uid suffix?

        logger.info("Starting truncate_image_tables")

        if options['all'] is True:
            truncate_database()

        else:
            truncate_filled_tables()

        # debug
        logger.info("Done!")