Gerar dados falsos para testar

utilsv2/mongo.py

@@ -3,14 +3,13 @@ from typing import Any
 
 from pymongo import MongoClient
 from pymongo.collection import Collection
+from pymongo.cursor import Cursor
 from pymongo.errors import ConnectionFailure
 
 try:
     from utilsv2.log import logger
-    from utilsv2.parser import massage_magnitudes
 except ModuleNotFoundError:
     from log import logger
-    from parser import massage_magnitudes
 
 logger = logging.getLogger(__name__)
 
@@ -32,8 +31,6 @@ def add_events(
     db = client["main"]
     coll = db[collection]
 
-    data = massage_magnitudes(data)
-
     _res = coll.insert_many(data)
 
     if _res.acknowledged:
@@ -65,7 +62,24 @@ def close(client: MongoClient) -> None:
     logger.info("Closed the DB.")
 
 
+def query_all(cli: MongoClient, collection: str) -> Any:
+    coll: Collection = cli.main[collection]
+
+    result = coll.find({})
+
+    for doc in result:
+        print(doc)
+
+
+def filter_query(cli: MongoClient, collection: str, filter_by):
+    coll: Collection = cli.main[collection]
+
+    res = coll.find({""})
+
+
 if __name__ == "__main__":
     v = connect("mongodb://localhost:27017")
 
+    query_all(v, "quakes")
+
     close(v)