Skip to content

Commit 4fd7f07

Browse files
authored
Merge pull request #34 from crane-cloud/ch/del-database
chore: ensure exclusion of deleted database records from listings
2 parents 878aae7 + 9c5dc7f commit 4fd7f07

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

app/routes.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,10 @@ def fetch_database_stats(access_token: str = Depends(security), db: Session = De
2929
total = 0
3030

3131
for flavour in database_flavours:
32-
databases = db.query(Database).filter_by(
33-
database_flavour_name=flavour['name']).all()
32+
databases = db.query(Database).filter(
33+
Database.database_flavour_name == flavour['name'],
34+
Database.deleted == False
35+
).all()
3436

3537
database_count = len(databases)
3638
dbs_per_flavour[f"{flavour['name']}_db_count"] = database_count
@@ -71,6 +73,9 @@ def get_all_databases(
7173
query = query.filter(
7274
Database.database_flavour_name == database_flavour_name)
7375

76+
# ensure at this point that deleted databases are not part of the query for the above filters
77+
query = query.filter(Database.deleted == False)
78+
7479
total_count = query.count()
7580
total_pages = (total_count + per_page - 1) // per_page
7681

@@ -181,9 +186,6 @@ def create_database(database: DatabaseFlavor, access_token: str = Depends(securi
181186
return SimpleNamespace(status_code=201, data={"database": database})
182187

183188

184-
185-
186-
187189
@router.post("/databases/{database_id}/enable")
188190
def enable_database(database_id: str, access_token: str = Depends(security), db: Session = Depends(get_db)):
189191
current_user = get_current_user(access_token.credentials)
@@ -621,7 +623,6 @@ def database_graph_data(start: Optional[str] = Query(description="Start date for
621623
return {"status_code": 200, 'data': {'metadata': metadata, 'graph_data': db_info}}
622624

623625

624-
625626
@router.get("/databases/{database_id}")
626627
def single_database(database_id: str, access_token: str = Depends(security), db: Session = Depends(get_db)):
627628
current_user = get_current_user(access_token.credentials)

0 commit comments

Comments
 (0)