rewardsystem/autocredits.py

@@ -2,14 +2,11 @@ async def get_all_users(client):
     users = client['nova-core']['users']
     return users
         
-
-async def update_credits(pymongo_client, settings=None):
-    users = await get_all_users(pymongo_client)
-
+async def update_credits(users, settings=None):
     if not settings:
-        users.update_many({}, {"$inc": {"credits": 2500}})
+        users.update_many({}, {"$inc": {"credits": 250}})
 
     else:
         for key, value in settings.items():
-            users.update_many(
-                {'level': key}, {"$inc": {"credits": int(value)}})
+            users.update_many({'role': key}, {"$inc": {"credits": int(value)}})
+            print(f'Updated {key} to {value}')

rewardsystem/main.py

@@ -1,48 +1,59 @@
-import asyncio
-from settings import roles
-import autocredits
-import aiohttp
-from dotenv import load_dotenv
 import os
+import asyncio
+import aiohttp
 import pymongo
 
+import autocredits
+
+from dotenv import load_dotenv
+
+from settings import roles
+
 load_dotenv()
 
-CONNECTION_STRING = os.getenv("MONGO_URI")
-
+pymongo_client = pymongo.MongoClient(os.getenv('MONGO_URI'))
 
 async def main():
-    pymongo_client = pymongo.MongoClient(CONNECTION_STRING)
-
-    await update_roles(pymongo_client)
-    await autocredits.update_credits(pymongo_client, roles)
-
-
-async def update_roles(pymongo_client):
-    async with aiohttp.ClientSession() as session:
-        try:
-            async with session.get('http://0.0.0.0:3224/get_roles') as response:
-                data = await response.json()
-        except aiohttp.ClientError as e:
-            print(f"Error: {e}")
-            return
-
-    lvlroles = [f"lvl{lvl}" for lvl in range(10, 110, 10)]
-    discord_users = data
     users = await autocredits.get_all_users(pymongo_client)
 
-    for user in users.find():
+    await update_roles(users)
+    await autocredits.update_credits(users, roles)
+
+async def update_roles(users):
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.get('http://localhost:3224/get_roles') as response:
+                discord_users = await response.json()
+        except aiohttp.ClientError as e:
+            raise ValueError('Could not get roles.') from exc
+
+    lvlroles = [f'lvl{lvl}' for lvl in range(10, 110, 10)] + ['']
+
+    users = await autocredits.get_all_users(pymongo_client)
+
+    filtered_users = users.find({'role': {'$in': lvlroles}})
+    
+    bulk_updates = []
+    for user in filtered_users:
         discord = str(user['auth']['discord'])
 
         for id_, roles in discord_users.items():
             if id_ == discord:
                 for role in lvlroles:
+                    print(2, id_)
                     if role in roles:
-                        users.update_one({'auth.discord': int(discord)}, {
-                                         '$set': {'level': role}})
-                        print(f"Updated {discord} to {role}")
+                        print(0, id_)
+                        bulk_updates.append(pymongo.UpdateOne(
+                            {'auth.discord': int(discord)},
+                            {'$set': {'role': role}})
+                        )
+                        print(1, id_)
+                        print(f'Updated {id_} to {role}')
+                        break
 
-    return users
+    if bulk_updates:
+        with pymongo_client:
+            users.bulk_write(bulk_updates)
 
 if __name__ == "__main__":
     asyncio.run(main())

rewardsystem/settings.py

@@ -1,13 +1,13 @@
 roles = {
-    '': '2500',
-    'lvl10': '2800',
-    'lvl20': '3100',
-    'lvl30': '3400',
-    'lvl40': '3700',
-    'lvl50': '4000',
-    'lvl60': '4300',
-    'lvl70': '4600',
-    'lvl80': '4900',
-    'lvl90': '5200',
-    'lvl100': '5500',
+    '': '250',
+    'lvl10': '280',
+    'lvl20': '310',
+    'lvl30': '340',
+    'lvl40': '370',
+    'lvl50': '400',
+    'lvl60': '430',
+    'lvl70': '460',
+    'lvl80': '490',
+    'lvl90': '520',
+    'lvl100': '550',
 }