lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Reformat using black

Browse source
This commit is contained in:
cuom1999 2022-05-14 12:57:27 -05:00
parent efee4ad081
commit a87fb49918
221 changed files with 19127 additions and 7310 deletions
Download patch file Download diff file Expand all files Collapse all files

35
judge/utils/camo.py
View file

@ -10,39 +10,44 @@ class CamoClient(object):
"""Based on https://github.com/sionide21/camo-client"""
def __init__(self, server, key, excluded=(), https=False):
self.server = server.rstrip('/')
self.server = server.rstrip("/")
self.key = key
self.https = https
self.excluded = excluded
def image_url(self, url):
return '%s/%s/%s' % (self.server,
hmac.new(utf8bytes(self.key), utf8bytes(url), sha1).hexdigest(),
utf8bytes(url).hex())
return "%s/%s/%s" % (
self.server,
hmac.new(utf8bytes(self.key), utf8bytes(url), sha1).hexdigest(),
utf8bytes(url).hex(),
)
def rewrite_url(self, url):
if url.startswith(self.server) or url.startswith(self.excluded):
return url
elif url.startswith(('http://', 'https://')):
elif url.startswith(("http://", "https://")):
return self.image_url(url)
elif url.startswith('//'):
return self.rewrite_url(('https:' if self.https else 'http:') + url)
elif url.startswith("//"):
return self.rewrite_url(("https:" if self.https else "http:") + url)
else:
return url
def update_tree(self, doc):
for img in doc.xpath('.//img'):
for attr in ('src', 'data-src'):
for img in doc.xpath(".//img"):
for attr in ("src", "data-src"):
if img.get(attr):
img.set(attr, self.rewrite_url(img.get(attr)))
for obj in doc.xpath('.//object'):
if obj.get('data'):
obj.set('data', self.rewrite_url(obj.get('data')))
for obj in doc.xpath(".//object"):
if obj.get("data"):
obj.set("data", self.rewrite_url(obj.get("data")))
if settings.DMOJ_CAMO_URL and settings.DMOJ_CAMO_KEY:
client = CamoClient(settings.DMOJ_CAMO_URL, key=settings.DMOJ_CAMO_KEY,
excluded=settings.DMOJ_CAMO_EXCLUDE,
https=settings.DMOJ_CAMO_HTTPS)
client = CamoClient(
settings.DMOJ_CAMO_URL,
key=settings.DMOJ_CAMO_KEY,
excluded=settings.DMOJ_CAMO_EXCLUDE,
https=settings.DMOJ_CAMO_HTTPS,
)
else:
client = None