From 436ddf6f23242eb709b591cd5e9cbf1553f8d390 Mon Sep 17 00:00:00 2001
From: Joris
Date: Mon, 20 May 2024 09:40:11 +0200
Subject: Allow to upload file and download from given link

---
 src/db.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 src/db.py

(limited to 'src/db.py')

diff --git a/src/db.py b/src/db.py
new file mode 100644
index 0000000..8aa20f8
--- /dev/null
+++ b/src/db.py
@@ -0,0 +1,20 @@
+import secrets
+
+def insert_file(conn, filename: str, expiration_days: int, content_length: int):
+    cur = conn.cursor()
+    file_id = secrets.token_urlsafe()
+    cur.execute(
+        'INSERT INTO files(id, filename, created, expires, content_length) VALUES(?, ?, datetime(), datetime(datetime(), ?), ?)',
+        (file_id, filename, f'+{expiration_days} days', content_length)
+    )
+    conn.commit()
+    return file_id
+
+def get_file(conn, file_id: str):
+    cur = conn.cursor()
+    res = cur.execute(
+        'SELECT filename, expires, content_length FROM files WHERE id = ?',
+        (file_id,)
+    )
+    return res.fetchone()
+
-- 
cgit v1.2.3