Skip to content

Commit a0fe0f9

Browse files
committed
refactor(backend): migrate to sqlalchemy 2.0 uuid, declarative base and col
1 parent 7807d5b commit a0fe0f9

5 files changed

Lines changed: 103 additions & 122 deletions

File tree

src/pypsa_app/backend/api/deps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def get_accessible_network(
177177
178178
Raises 404 if not found or inaccessible.
179179
"""
180-
network = db.query(Network).filter(Network.id == str(network_id)).first()
180+
network = db.query(Network).filter(Network.id == network_id).first()
181181
if not network:
182182
raise HTTPException(404, "Network not found")
183183

src/pypsa_app/backend/api/routes/admin.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def update_network_admin(
243243
network = (
244244
db.query(Network)
245245
.options(joinedload(Network.owner))
246-
.filter(Network.id == str(network_id))
246+
.filter(Network.id == network_id)
247247
.first()
248248
)
249249

@@ -296,7 +296,7 @@ def delete_network_admin(
296296
admin: User = Depends(require_permission(Permission.NETWORKS_MANAGE_ALL)),
297297
) -> dict:
298298
"""Delete any network (admin only)"""
299-
network = db.query(Network).filter(Network.id == str(network_id)).first()
299+
network = db.query(Network).filter(Network.id == network_id).first()
300300
if not network:
301301
raise HTTPException(404, "Network not found")
302302

src/pypsa_app/backend/api/routes/networks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def get_network(
170170
network = (
171171
db.query(Network)
172172
.options(joinedload(Network.owner))
173-
.filter(Network.id == str(network_id))
173+
.filter(Network.id == network_id)
174174
.first()
175175
)
176176

@@ -194,7 +194,7 @@ def update_network(
194194
network = (
195195
db.query(Network)
196196
.options(joinedload(Network.owner))
197-
.filter(Network.id == str(network_id))
197+
.filter(Network.id == network_id)
198198
.first()
199199
)
200200

src/pypsa_app/backend/database.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from sqlalchemy import create_engine
2-
from sqlalchemy.orm import declarative_base, sessionmaker
2+
from sqlalchemy.orm import DeclarativeBase, sessionmaker
33

44
from pypsa_app.backend.settings import (
55
DB_MAX_OVERFLOW,
@@ -30,7 +30,9 @@
3030

3131
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
3232

33-
Base = declarative_base()
33+
34+
class Base(DeclarativeBase):
35+
pass
3436

3537

3638
from pypsa_app.backend import models # noqa: E402, F401

0 commit comments

Comments
 (0)