Save expire_date as datetime.date

This commit is contained in:
Parker M. 2024-02-25 17:52:10 -06:00
parent fdc9939f57
commit 236f050f54
No known key found for this signature in database
GPG Key ID: 95CD2E0C7E329F2A
2 changed files with 2 additions and 4 deletions

View File

@ -29,8 +29,7 @@ def generate_link(request, owner):
while True:
try:
link = ''.join(random.choices(choices, k=5))
expire_date = (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y')
expire_date = datetime.datetime.strptime(expire_date, '%d/%m/%Y')
expire_date = datetime.datetime.date(datetime.datetime.now()) + datetime.timedelta(days=7)
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

View File

@ -16,8 +16,7 @@ def renew_link(link, owner):
if owner == link_owner:
with engine.begin() as conn:
expire_date = (datetime.datetime.now() + datetime.timedelta(days=7)).strftime('%d/%m/%Y')
expire_date = datetime.datetime.strptime(expire_date, '%d/%m/%Y')
expire_date = datetime.datetime.date(datetime.datetime.now()) + datetime.timedelta(days=7)
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: