heierbtc/lnbits
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

check uploaded image size in API

Browse Source
This commit is contained in:
Tiago Vasconcelos 2023-01-09 12:31:12 +00:00
parent 4d5c7133bc
commit 44fa30fdc3
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
lnbits/extensions/market/views_api.py
View File

@ -113,6 +113,23 @@ async def api_market_product_create(
if stall.currency != "sat": if stall.currency != "sat":
data.price *= settings.fiat_base_multiplier data.price *= settings.fiat_base_multiplier
if data.image:
image_is_url = data.image.startswith("https://") or data.image.startswith(
"http://"
)
if not image_is_url:
def size(b64string):
return int((len(b64string) * 3) / 4 - b64string.count("=", -2))
image_size = size(data.image) / 1024
if image_size > 100:
raise HTTPException(
status_code=HTTPStatus.BAD_REQUEST,
detail=f"Image size is too big, {int(image_size)}Kb. Max: 100kb, Compress the image at https://tinypng.com, or use an URL.",
)
if product_id: if product_id:
product = await get_market_product(product_id) product = await get_market_product(product_id)
if not product: if not product: