fix: missing import, typo

main
rrr-marble 5 years ago
parent 6e0940aaf2
commit f8ef88d442

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

Loading…
Cancel
Save