From f97552b076f056fb94a6a3e20933ed6c8246d354 Mon Sep 17 00:00:00 2001 From: Aevann Date: Wed, 6 Sep 2023 17:25:01 +0300 Subject: [PATCH] handle error --- files/routes/users.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/files/routes/users.py b/files/routes/users.py index 85de2a5f9..66e8a6d4a 100644 --- a/files/routes/users.py +++ b/files/routes/users.py @@ -7,6 +7,7 @@ from collections import Counter import gevent import qrcode from sqlalchemy.orm import aliased, load_only +from sqlalchemy.exc import IntegrityError from files.classes import * from files.classes.leaderboard import Leaderboard @@ -33,8 +34,14 @@ def _add_profile_view(vid, uid): if view: view.last_view_utc = int(time.time()) else: view = ViewerRelationship(viewer_id=vid, user_id=uid) db.add(view) + + try: + db.commit() + except IntegrityError as e: + db.rollback() + if not str(e).startswith('(psycopg2.errors.UniqueViolation) duplicate key value violates unique constraint "'): + print(STARS + e + STARS, flush=True) - db.commit() db.close() stdout.flush()