alex/nova-api
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-api.git synced 2024-11-25 16:43:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Updated main.py

Browse source
This commit is contained in:
Leander 2023-08-07 14:55:17 +02:00 committed by GitHub
parent 9a1a9d9e3e
commit 3f1db2dd3e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
rewardsystem/main.py
View file

@ -1,59 +1,48 @@
import os
import asyncio
import aiohttp
import pymongo
import autocredits
from dotenv import load_dotenv
from settings import roles
import autocredits
import aiohttp
from dotenv import load_dotenv
import os
import pymongo
load_dotenv()
pymongo_client = pymongo.MongoClient(os.getenv('MONGO_URI'))
CONNECTION_STRING = os.getenv("MONGO_URI")
async def main():
users = await autocredits.get_all_users(pymongo_client)
pymongo_client = pymongo.MongoClient(CONNECTION_STRING)
await update_roles(users)
await autocredits.update_credits(users, roles)
await update_roles(pymongo_client)
await autocredits.update_credits(pymongo_client, roles)
async def update_roles(users):
async def update_roles(pymongo_client):
async with aiohttp.ClientSession() as session:
try:
async with session.get('http://localhost:3224/get_roles') as response:
discord_users = await response.json()
async with session.get('http://0.0.0.0:3224/get_roles') as response:
data = 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)] + ['']
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)
filtered_users = users.find({'role': {'$in': lvlroles}})
bulk_updates = []
for user in filtered_users:
for user in users.find():
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:
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
users.update_one({'auth.discord': int(discord)}, {
'$set': {'level': role}})
print(f"Updated {discord} to {role}")
if bulk_updates:
with pymongo_client:
users.bulk_write(bulk_updates)
return users
if __name__ == "__main__":
asyncio.run(main())
asyncio.run(main())