NDOJ/judge/management/commands/camo.py

17 lines
484 B
Python
Raw Normal View History

2020-01-21 06:35:58 +00:00
from django.core.management.base import BaseCommand, CommandError
from judge.utils.camo import client as camo_client
class Command(BaseCommand):
help = 'obtains the camo url for the specified url'
def add_arguments(self, parser):
parser.add_argument('url', help='url to use camo on')
def handle(self, *args, **options):
if camo_client is None:
raise CommandError('Camo not available')
print(camo_client.image_url(options['url']))