fix: add DatabaseError if select dev_eui does not return correctly
This commit is contained in:
parent
e887b1d357
commit
914235064c
1 changed files with 5 additions and 2 deletions
|
|
@ -26,7 +26,10 @@ class PgDeviceRepository(DeviceRepository):
|
|||
cur.execute(
|
||||
"SELECT device_id FROM device WHERE device_eui = %s", (dev_eui,)
|
||||
)
|
||||
return str(cur.fetchone()[0]) # type: ignore
|
||||
row = cur.fetchone()
|
||||
if row is None:
|
||||
raise DatabaseError(f"Device {dev_eui} introuvable après insertion")
|
||||
return str(row[0])
|
||||
except psycopg2.DatabaseError as e:
|
||||
raise DatabaseError(f"Erreur de création du device {dev_eui}") from e
|
||||
|
||||
|
|
@ -38,6 +41,6 @@ class PgDeviceRepository(DeviceRepository):
|
|||
cur.execute(query)
|
||||
rows = cur.fetchall()
|
||||
except psycopg2.DatabaseError as e:
|
||||
raise DatabaseError(f"Erreur d'accès aux devices : {e}") from e
|
||||
raise DatabaseError("Erreur d'accès aux devices") from e
|
||||
|
||||
return [Device(device_id=r[0], device_eui=r[1]) for r in rows]
|
||||
Loading…
Add table
Add a link
Reference in a new issue