Skip to content

Commit d004610

Browse files
committed
Refactor dashboard to use Datastore instead of Firestore for data retrieval
1 parent b829ff3 commit d004610

1 file changed

Lines changed: 10 additions & 7 deletions

File tree

app.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from flask import Flask, render_template
2-
from google.cloud import firestore
2+
from google.cloud import datastore
33
import os
44
from dotenv import load_dotenv
55
import logging
@@ -16,13 +16,16 @@
1616
@app.route("/")
1717
def dashboard():
1818
try:
19-
logger.info("Attempting to connect to Firestore...")
20-
db = firestore.Client()
19+
logger.info("Attempting to connect to Datastore...")
20+
# Use datastore client instead of firestore since project is in Datastore mode
21+
db = datastore.Client()
2122

22-
logger.info("Fetching runs from Firestore...")
23-
runs = db.collection("runs").order_by("timestamp", direction=firestore.Query.DESCENDING).limit(10).get()
24-
run_data = [r.to_dict() for r in runs]
25-
logger.info(f"Retrieved {len(run_data)} runs from Firestore")
23+
logger.info("Fetching runs from Datastore...")
24+
query = db.query(kind="runs")
25+
query.order = ["-timestamp"]
26+
runs = list(query.fetch(limit=10))
27+
run_data = [dict(r) for r in runs]
28+
logger.info(f"Retrieved {len(run_data)} runs from Datastore")
2629

2730
logger.info("Fetching anomalies from Firestore...")
2831
anomalies = db.collection("anomalies").order_by("run_id", direction=firestore.Query.DESCENDING).limit(5).get()

0 commit comments

Comments
 (0)