clean junk data
This commit is contained in:
55
scripts/delete_junk_samples.py
Normal file
55
scripts/delete_junk_samples.py
Normal file
@@ -0,0 +1,55 @@
|
||||
from firebase_admin import firestore
|
||||
import os
|
||||
import firebase_admin
|
||||
from datetime import datetime, timedelta
|
||||
from google.cloud.firestore_v1.base_query import FieldFilter
|
||||
|
||||
def delete_documents_in_batch(collection_ref, batch_size):
|
||||
# Query to get all documents
|
||||
query = collection_ref.order_by('testTime', direction=firestore.Query.DESCENDING).limit(batch_size)
|
||||
documents = query.stream()
|
||||
|
||||
batch = collection_ref._client.batch() # Internal batch object
|
||||
|
||||
# Delete documents in batch
|
||||
count = 0
|
||||
for doc in documents:
|
||||
# print(doc.to_dict)
|
||||
batch.delete(doc.reference)
|
||||
count += 1
|
||||
|
||||
# Commit the batch
|
||||
batch.commit()
|
||||
print(f'Deleted {count} documents in batch')
|
||||
|
||||
|
||||
def delete_latest_documents(collection_name, num_documents, batch_size):
|
||||
|
||||
file_path = os.getcwd() + '/keys' + '/hpos-af3cc-firebase-adminsdk.json'
|
||||
cred = firebase_admin.credentials.Certificate(fr'{file_path}')
|
||||
|
||||
firebase_admin.initialize_app(cred)
|
||||
|
||||
# Initialize Firestore client
|
||||
db = firestore.client()
|
||||
|
||||
# Reference to the collection
|
||||
collection_ref = db.collection(collection_name)
|
||||
|
||||
# Delete documents in batches
|
||||
remaining_documents = num_documents
|
||||
while remaining_documents > 0:
|
||||
current_batch_size = min(remaining_documents, batch_size)
|
||||
delete_documents_in_batch(collection_ref, current_batch_size)
|
||||
remaining_documents -= current_batch_size
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Specify your Firestore collection name
|
||||
collection_name = 'testData'
|
||||
|
||||
num_documents = 10000
|
||||
batch_size = 500
|
||||
|
||||
# Call the function to delete the latest documents in batches
|
||||
delete_latest_documents(collection_name, num_documents, batch_size)
|
||||
Reference in New Issue
Block a user