From ff17350222cce6af80a1f61e54a828f821bfccb4 Mon Sep 17 00:00:00 2001
From: Andrew Ridgway <ar17787@gmail.com>
Date: Fri, 12 Jul 2024 21:00:24 +1000
Subject: [PATCH] get ph and dates

---
 src/flask/chart_test.py        |  5 +++++
 src/flask/charts/PoolCharts.py | 17 +++++++++++++++++
 src/flask/charts/__init__.py   |  0
 src/flask/mongo/query_db.py    |  2 +-
 4 files changed, 23 insertions(+), 1 deletion(-)
 create mode 100644 src/flask/chart_test.py
 create mode 100644 src/flask/charts/PoolCharts.py
 create mode 100644 src/flask/charts/__init__.py

diff --git a/src/flask/chart_test.py b/src/flask/chart_test.py
new file mode 100644
index 0000000..4039c81
--- /dev/null
+++ b/src/flask/chart_test.py
@@ -0,0 +1,5 @@
+from charts import PoolCharts
+
+chart = PoolCharts.PoolCharts()
+
+chart.ph_line_chart()
diff --git a/src/flask/charts/PoolCharts.py b/src/flask/charts/PoolCharts.py
new file mode 100644
index 0000000..0924a1e
--- /dev/null
+++ b/src/flask/charts/PoolCharts.py
@@ -0,0 +1,17 @@
+import sys
+
+from bokeh.plotting import figure, show
+
+from mongo.get_conn import db_conn
+
+class PoolCharts():
+    def __init__(self):
+        self.db = db_conn()
+
+
+    def ph_line_chart(self):
+        ph = self.db.pool_db
+        query = {}
+        data = ph.find({}).sort("date", -1).limit(10)
+        for record in data:
+            print (f'{record["date"]},{record["ph"]}')
diff --git a/src/flask/charts/__init__.py b/src/flask/charts/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/flask/mongo/query_db.py b/src/flask/mongo/query_db.py
index c8aa476..17c1491 100644
--- a/src/flask/mongo/query_db.py
+++ b/src/flask/mongo/query_db.py
@@ -23,7 +23,7 @@ class pool_query:
             queried record
         """
         query = { "test_user" : f"{test_user}", "date" : f"{date}"}
-        record = self.db.real_db.find_one(query)
+        record = self.db.pool_db.find_one(query)
         if record:
             self.existing_record = record
             return True