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()