aiolabs/withdraw
8
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix updating the uses (#6)

Browse source 
* update the usescsv db row when uses change
* add a created column for visual aid of the created links
* allow decrease uses
This commit is contained in:
Tiago Vasconcelos 2023-09-24 18:32:00 +01:00 • committed by GitHub
commit d8b6ac40bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 27 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
views_api.py
View file

@ -113,7 +113,27 @@ async def api_link_create_or_update(
raise HTTPException(
detail="Not your withdraw link.", status_code=HTTPStatus.FORBIDDEN
)
link = await update_withdraw_link(link_id, **data.dict())
data_dict = data.dict()
if link.uses > data.uses:
if data.uses - link.used <= 0:
raise HTTPException(
detail="Cannot reduce uses below current used.", status_code=HTTPStatus.BAD_REQUEST
)
numbers = link.usescsv.split(",")
usescsv = ",".join(numbers[:data.uses - link.used])
data_dict["usescsv"] = usescsv
if link.uses < data.uses:
numbers = link.usescsv.split(",")
current_number = int(numbers[-1])
while len(numbers) < (data.uses - link.used):
current_number += 1
numbers.append(str(current_number))
usescsv = ",".join(numbers)
data_dict["usescsv"] = usescsv
link = await update_withdraw_link(link_id, **data_dict)
else:
link = await create_withdraw_link(wallet_id=wallet.wallet.id, data=data)
assert link