Userid-chatid-fix

.env.example

@@ -1,4 +1,6 @@
 TG_BOT_TOKEN=""
 MONGODB_USERNAME=""
 MONGODB_PASSWORD=""
-MONGODB_PORT=27017
+MONGODB_PORT=27017
+
+MONGODB_HOST="mongodb://login:password@ip:port" # If you don't use docker compose

main.py

@@ -1,10 +1,6 @@
 import asyncio
 
-from src.bot import bot, dp
-
-
-async def main():
-    await dp.start_polling(bot)
+from src.bot import main
 
 if __name__ == "__main__":
     asyncio.run(main())

random_prizes.py

@@ -1,24 +1,19 @@
-from dotenv import load_dotenv
-import os
 import argparse
-import pymongo
+import os
 import random
 
+import pymongo
+from dotenv import load_dotenv
+
 load_dotenv()
 
-MONGODB_USERNAME = os.getenv("MONGODB_USERNAME")
-MONGODB_PASSWORD = os.getenv("MONGODB_PASSWORD")
-MONGODB_PORT = os.getenv("MONGODB_PORT")
-MONGODB_IP = os.getenv("MONGODB_IP")
+MONGODB_HOST = os.getenv("MONGODB_HOST")
 
 parser = argparse.ArgumentParser()
 parser.add_argument("count", type=int)
 args = parser.parse_args()
 
-MONGO_HOST = (
-    f"mongodb://{MONGODB_USERNAME}:{MONGODB_PASSWORD}@{MONGODB_IP}:{MONGODB_PORT}"
-)
-client = pymongo.MongoClient(MONGO_HOST)
+client = pymongo.MongoClient(MONGODB_HOST)
 database = client.get_database("cu_graph_bot")
 collection = database.get_collection("users")
 users = ["@" + i["username"] for i in collection.find({"_links.4": {"$exists": True}})]