Changes to db: fixed error in __init__, looking for more precise errors
This commit is contained in:
@ -1,5 +1,5 @@
|
||||
import sqlmodel
|
||||
import mariadb
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
from sqlalchemy.future import Engine
|
||||
|
||||
from . import models
|
||||
@ -21,7 +21,7 @@ def add_account(
|
||||
with sqlmodel.Session(engine) as session:
|
||||
session.add(account)
|
||||
session.commit()
|
||||
except Exception:
|
||||
except IntegrityError:
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
@ -34,7 +34,7 @@ def add_master_pass(engine: Engine, user_id: int, salt: bytes, passwd: bytes) ->
|
||||
with sqlmodel.Session(engine) as session:
|
||||
session.add(master_pass)
|
||||
session.commit()
|
||||
except Exception:
|
||||
except IntegrityError:
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
|
Reference in New Issue
Block a user