1 files changed, 20 insertions, 0 deletions
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()
+