flifloo/BetterTwitFix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Image check in lambda handler

Browse source
This commit is contained in:
Dylan 2022-05-26 16:13:31 +01:00
parent 77b972a1b2
commit ec1f4dc284
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
combineImg/__init__.py
View file

@ -117,6 +117,9 @@ def genImageFromURL(urlArray):
def lambda_handler(event, context): def lambda_handler(event, context):
images = event["queryStringParameters"].get("imgs","").split(",") images = event["queryStringParameters"].get("imgs","").split(",")
for img in images:
if not img.startswith("https://pbs.twimg.com"):
return {'statusCode':400,'body':'Invalid image URL'}
combined = genImageFromURL(images) combined = genImageFromURL(images)
buffered = BytesIO() buffered = BytesIO()
combined.save(buffered,format="JPEG",quality=60) combined.save(buffered,format="JPEG",quality=60)