diff options
Diffstat (limited to 'src/soc_collector/db.py')
| -rw-r--r-- | src/soc_collector/db.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/soc_collector/db.py b/src/soc_collector/db.py index d601a82..f537f4a 100644 --- a/src/soc_collector/db.py +++ b/src/soc_collector/db.py @@ -21,7 +21,7 @@ from bson import ObjectId class SearchInput(BaseModel): """Handle search data for HTTP request""" - search: Optional[Dict[str, Any]] + filter: Dict[str, Any] limit: int = 25 skip: int = 0 @@ -68,9 +68,9 @@ class DBClient: """ data: List[Dict[str, Any]] = [] - cursor = self.collection.find(search_data.search) + cursor = self.collection.find(search_data.filter) - cursor.sort({"ip": ASCENDING, "timestamp": DESCENDING}).limit(search_data.limit).skip(search_data.skip) + cursor.sort([("ip", ASCENDING), ("timestamp", DESCENDING)]).limit(search_data.limit).skip(search_data.skip) try: async for document in cursor: |