fix: missing import, typo

pull/6/head
rrr-marble 5 years ago
parent 6e0940aaf2
commit f8ef88d442

@ -1,4 +1,6 @@
from fastapi import FastAPI from fastapi import FastAPI
from uuid import uuid4
import sqlite3 import sqlite3
# use database residing here # use database residing here
@ -31,7 +33,7 @@ async def new_session():
except sqlite3.IntegrityError as e: except sqlite3.IntegrityError as e:
if i < tries - 1 and str(e) == "UNIQUE constraint failed: sessions.cookie": if i < tries - 1 and str(e) == "UNIQUE constraint failed: sessions.cookie":
continue continue
else if str(e) == "UNIQUE constraint failed: sessions.cookie": elif str(e) == "UNIQUE constraint failed: sessions.cookie":
return {"cookie": "error"} return {"cookie": "error"}
else: else:
raise raise
@ -115,7 +117,7 @@ async def rate_picture(cookie: str, picture_id: int, mark: int):
""", """,
{"imgid": picture_id, "sessionid": sessionid["sessionid"], "mark": mark}, {"imgid": picture_id, "sessionid": sessionid["sessionid"], "mark": mark},
) )
cur.commit() con.commit()
except sqlite3.IntegrityError as e: except sqlite3.IntegrityError as e:
if str(e) == "UNIQUE constraint failed: marks.imgid, marks.sessionid": if str(e) == "UNIQUE constraint failed: marks.imgid, marks.sessionid":
return {"status": "already rated"} return {"status": "already rated"}

Loading…
Cancel
Save