From 6fd3fe78cb04d40cd94b39f2ab0b12d3cf398b01 Mon Sep 17 00:00:00 2001
From: osiris account <osiris137@protonmail.com>
Date: Sun, 12 Mar 2023 13:11:37 -0700
Subject: [PATCH] add libraries and clean up

---
 token-scanner-api/src/server/database.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/token-scanner-api/src/server/database.py b/token-scanner-api/src/server/database.py
index 856c12e..f431414 100644
--- a/token-scanner-api/src/server/database.py
+++ b/token-scanner-api/src/server/database.py
@@ -44,9 +44,7 @@ async def retrieve_top_balances(top_number: int, env_vars: dict) -> list:
     """Retrieve top balances from the database."""
 
     collection = _get_db_collection(env_vars)
-    # todo: need to sort? how to optimzie to not return all?
-    # something like for post in posts.find({"date": {"$lt": d}}).sort("author"):
-    top_balances = collection.find()
+    top_balances = collection.find().sort({"balance": {"$lt": top_number}}, pymongo.DESCENDING)
 
     result = []
     counter = 0