From de0b9cf8d358da15fa314cbc6864027e421eb4a1 Mon Sep 17 00:00:00 2001 From: rrr-marble Date: Fri, 2 Jul 2021 15:57:32 +0300 Subject: [PATCH] add: save session initialization time --- main.py | 8 +++++--- util/import_photos.py | 1 + 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 02db192..be12b54 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,6 @@ from fastapi import FastAPI +from datetime import datetime from uuid import uuid4 import sqlite3 @@ -18,16 +19,17 @@ cur = con.cursor() # NB! single is enough for now, we might require multiple la async def new_session(): """Start a new session""" # add session to the database + time = datetime.utcnow().replace(microsecond=0).isoformat() tries = 3 # something is very wrong with our random, if we miss 3 times for i in range(tries): try: # generate a cookie cookie = uuid4().hex cur.execute( - """INSERT INTO sessions(cookie) - VALUES(:cookie) + """INSERT INTO sessions(cookie, time) + VALUES(:cookie, :time) """, - {"cookie": cookie}, + {"cookie": cookie, "time": time}, ) con.commit() except sqlite3.IntegrityError as e: diff --git a/util/import_photos.py b/util/import_photos.py index af4d120..59aca94 100644 --- a/util/import_photos.py +++ b/util/import_photos.py @@ -138,6 +138,7 @@ def check_database(database_path: str): ( sessionid INTEGER PRIMARY KEY, cookie TEXT UNIQUE NOT NULL, + time TEXT, description TEXT )""" )