<reponame>videetssinghai/URL_Shortner
from django.core.management.base import BaseCommand, CommandError
from shortener.models import getBit

class Command(BaseCommand):
    help = 'Closes the specified poll for voting'

    def add_arguments(self, parser):
        parser.add_argument('poll_id', nargs='+', type=int)

    def handle(self, *args, **options):
        return getBit.objects.refresh_codes()


