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/records.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 app/func/link/records.py (limited to 'app/func/link/records.py') diff --git a/app/func/link/records.py b/app/func/link/records.py new file mode 100644 index 0000000..56bb6d2 --- /dev/null +++ b/app/func/link/records.py @@ -0,0 +1,23 @@ +import sqlalchemy + +from db import engine + +""" +Retrieve all records associated with a specific link +""" +def get_link_records(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: + records = conn.execute(sqlalchemy.text('SELECT timestamp, ip, location, browser, os, user_agent, isp FROM records WHERE owner = :owner and link = :link'), [{'owner': owner, 'link': link}]).fetchall() + if not records: + return 204 + else: + return 401 + + return records \ No newline at end of file -- cgit v1.2.3-70-g09d2