From 7cd2fd47f7ac3f1da82a8f289c1758ae70750309 Mon Sep 17 00:00:00 2001 From: Pritimay Sarkar Date: Thu, 26 Oct 2023 12:12:05 +0530 Subject: [PATCH] add date range for buffers --- scripts/buffers.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/scripts/buffers.py b/scripts/buffers.py index 5c84917..4a795e9 100644 --- a/scripts/buffers.py +++ b/scripts/buffers.py @@ -7,32 +7,29 @@ import pandas as pd from datetime import datetime import sys -# Initialize Firebase Admin SDK cred = credentials.Certificate(os.getcwd() + '/' + 'keys/hpos-prod-firebase-adminsdk.json') # Replace with your own service account key path firebase_admin.initialize_app(cred) -# Get a reference to the Firestore database db = firestore.client() -# Specify the collections -buffers_collection = db.collection("buffers") +start_date = sys.argv[1] +end_date = sys.argv[2] + +buffers_collection = db.collection("buffers").where(filter=FieldFilter("testTime", ">=", start_date)).where(filter=FieldFilter("testTime", "<", end_date)) docs = buffers_collection.stream() -# Prepare data to store in CSV data = [] for doc in docs: doc_data = doc.to_dict() data.append(doc_data) -# Convert the data to a DataFrame df = pd.DataFrame(data) print(df) print(df.size) # print("duplicates", len(df['_id']) - len(df['_id'].drop_duplicates())) -# Save the DataFrame to a CSV file output_filename = f'buffers_{datetime.today().strftime("%d_%m_%Y_%H_%M")}.xlsx' downloads_dir = os.path.join(os.path.expanduser("~"), "Downloads") @@ -44,6 +41,5 @@ writer.close() print(f"Data saved to '{output_path}'") -# Close the Firebase Admin SDK firebase_admin.delete_app(firebase_admin.get_app())