diff --git a/app/func/newlink.py b/app/func/newlink.py index 9cc220f..ec45111 100644 --- a/app/func/newlink.py +++ b/app/func/newlink.py @@ -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 diff --git a/app/func/renew_link.py b/app/func/renew_link.py index bbb3de2..9d31c33 100644 --- a/app/func/renew_link.py +++ b/app/func/renew_link.py @@ -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: