From bce756b9aa51dfdfa2c2e6cdef4f88f660b80148 Mon Sep 17 00:00:00 2001 From: Parker Date: Mon, 26 Feb 2024 20:55:05 -0600 Subject: Overhaul: Flask -> FastAPI --- app/func/link/renew.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 app/func/link/renew.py (limited to 'app/func/link/renew.py') diff --git a/app/func/link/renew.py b/app/func/link/renew.py new file mode 100644 index 0000000..bcf0550 --- /dev/null +++ b/app/func/link/renew.py @@ -0,0 +1,23 @@ +import sqlalchemy +import datetime + +from db import engine + +""" +Renew a specified link so that the user can continue logging through that URL +Adds 7 days from the current date +""" +def renew_link(link, owner): + with engine.begin() as conn: + try: + link_owner = conn.execute(sqlalchemy.text('SELECT owner FROM links WHERE link = :link'), [{'link': link}]).fetchone()[0] + except TypeError: + return 404 + + if owner == link_owner: + with engine.begin() as conn: + 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 link, expire_date + else: + return 401 \ No newline at end of file -- cgit v1.2.3-70-g09d2