parker/linklogger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Save expire_date as datetime rather than string

Browse Source
This commit is contained in:
Parker M. 2024-02-25 17:46:34 -06:00
parent fd90921c10
commit fdc9939f57
No known key found for this signature in database
GPG Key ID: 95CD2E0C7E329F2A
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/func/newlink.py
View File

@ -29,7 +29,9 @@ def generate_link(request, owner):
while True:
try:
link = ''.join(random.choices(choices, k=5))
conn.execute(sqlalchemy.text('INSERT INTO links(owner, link, redirect_link, expire_date) VALUES (:owner, :link, :redirect_link, :expire_date)'), [{'owner': owner, 'link': link, 'redirect_link': redirect_link, 'expire_date': (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y')}])
expire_date = (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y')
expire_date = datetime.datetime.strptime(expire_date, '%d/%m/%Y')
conn.execute(sqlalchemy.text('INSERT INTO links(owner, link, redirect_link, expire_date) VALUES (:owner, :link, :redirect_link, :expire_date)'), [{'owner': owner, 'link': link, 'redirect_link': redirect_link, 'expire_date': expire_date}])
conn.commit()
break
except exc.IntegrityError:

6
app/func/renew_link.py
View File

@ -16,7 +16,9 @@ def renew_link(link, owner):
if owner == link_owner:
with engine.begin() as conn:
conn.execute(sqlalchemy.text('UPDATE links SET expire_date = :expire_date WHERE link = :link'), [{'expire_date': (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y'), 'link': link}])
return f'Link renewed, now expires on {(datetime.datetime.now() + datetime.timedelta(days=7)).strftime("%d/%m/%Y")}', 200
expire_date = (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y')
expire_date = datetime.datetime.strptime(expire_date, '%d/%m/%Y')
conn.execute(sqlalchemy.text('UPDATE links SET expire_date = :expire_date WHERE link = :link'), [{'expire_date': expire_date, 'link': link}])
return f'Link renewed, now expires on {expire_date}', 200
else:
return 'You are not the owner of this link', 401