tidy, use new nowcasting_datamodel

requirements.txt

@@ -1,5 +1,5 @@
 altair==4.2.2
-nowcasting_datamodel==1.5.19
+nowcasting_datamodel==1.5.25
 pvsite-datamodel==1.0.1
 numpy==1.24.1
 pandas==1.5.3

src/users.py

@@ -4,6 +4,10 @@
 import os
 from nowcasting_datamodel.connection import DatabaseConnection
 from nowcasting_datamodel.models.api import UserSQL, APIRequestSQL
+from nowcasting_datamodel.read.read_user import (
+    get_all_last_api_request,
+    get_api_requests_for_one_user,
+)
 
 from plots.users import make_api_requests_plot
 
@@ -24,21 +28,18 @@ def user_page():
         value=datetime.today() - timedelta(days=31),
     )
     end_time = st.sidebar.date_input(
-        "End Date", min_value=datetime.today() - timedelta(days=365), max_value=datetime.today()
+        "End Date",
+        min_value=datetime.today() - timedelta(days=365),
+        max_value=datetime.today() + timedelta(days=1),
+        value=datetime.today() + timedelta(days=1),
     )
 
     # get last call from the database
     url = os.environ["DB_URL"]
     connection = DatabaseConnection(url=url, echo=True)
     with connection.get_session() as session:
 
-        last_requests_sql = (
-            session.query(APIRequestSQL)
-            .distinct(APIRequestSQL.user_uuid)
-            .join(UserSQL)
-            .order_by(APIRequestSQL.user_uuid, APIRequestSQL.created_utc.desc())
-            .all()
-        )
+        last_requests_sql = get_all_last_api_request(session=session)
 
         last_request = [
             (last_request_sql.user.email, last_request_sql.created_utc)
@@ -56,13 +57,8 @@ def user_page():
 
     # get all calls for selected user
     with connection.get_session() as session:
-        api_requests_sql = (
-            session.query(APIRequestSQL)
-            .join(UserSQL)
-            .where(UserSQL.email == email_selected)
-            .where(APIRequestSQL.created_utc >= start_time)
-            .where(APIRequestSQL.created_utc <= end_time)
-            .all()
+        api_requests_sql = get_api_requests_for_one_user(
+            session=session, email=email_selected, start_datetime=start_time, end_datetime=end_time
         )
 
         api_requests = [
@@ -73,5 +69,3 @@ def user_page():
 
     fig = make_api_requests_plot(api_requests, email_selected, end_time, start_time)
     st.plotly_chart(fig, theme="streamlit")
-
-