quick change for current date [ci-skip]

sfc-gh-jdemlow · Aug 22, 2024 · 552ea17 · 552ea17
1 parent ec5f172
commit 552ea17
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/all_in_one_app.py b/all_in_one_app.py
@@ -133,7 +133,11 @@ def run_command(query):
             STORAGE_BYTES as total_active_bytes,
             STAGE_BYTES as total_stage_bytes,
             FAILSAFE_BYTES as total_failsafe_bytes
-        FROM snowflake.account_usage.storage_usage
+        FROM (
+            SELECT * 
+            FROM snowflake.account_usage.storage_usage
+            WHERE usage_date < CURRENT_DATE()
+        )
         WHERE USAGE_DATE = (SELECT MAX(USAGE_DATE) FROM snowflake.account_usage.storage_usage)
     )
     SELECT 

diff --git a/storage/forecast.py b/storage/forecast.py
@@ -9,7 +9,12 @@ def generate_storage_forecast(training_days, predicted_days):
     SELECT 
         TO_TIMESTAMP_NTZ(usage_date) AS usage_date,
         storage_bytes / POWER(1024, 3) AS storage_gb
-    FROM snowflake.account_usage.storage_usage
+    FROM   
+    (
+        SELECT * 
+        FROM snowflake.account_usage.storage_usage
+        WHERE usage_date < CURRENT_DATE()
+    )
     WHERE TO_TIMESTAMP_NTZ(usage_date) < DATEADD(day, -{training_days}, CURRENT_DATE());
     """)
 

diff --git a/streamlit_app.py b/streamlit_app.py
@@ -10,7 +10,6 @@
 from storage.forecast import generate_storage_forecast
 from storage.recommendations import generate_recommendations, display_recommendations
 
-
 # Initialize session state
 if 'storage_data' not in st.session_state:
     st.session_state.storage_data = None
@@ -75,7 +74,7 @@
             STAGE_BYTES as total_stage_bytes,
             FAILSAFE_BYTES as total_failsafe_bytes
         FROM snowflake.account_usage.storage_usage
-        WHERE USAGE_DATE = (SELECT MAX(USAGE_DATE) FROM snowflake.account_usage.storage_usage)
+        WHERE USAGE_DATE = DATEADD(day, -1, (SELECT MAX(USAGE_DATE) FROM snowflake.account_usage.storage_usage))
     )
     SELECT 
         ROUND(total_active_bytes / POWER(1024, 3), 1) AS "Active Storage (GB)",