NDOJ/judge/management/commands/camo.py
2022-05-14 12:57:27 -05:00

16 lines
484 B
Python

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"]))