diff --git a/api/config/credits.yml b/api/config/credits.yml index fa2449b..7025e66 100644 --- a/api/config/credits.yml +++ b/api/config/credits.yml @@ -18,7 +18,6 @@ bonuses: helper: 0.4 booster: 0.5 -# discord reward 0.99^lvl? rewards: - day: 250 + day: 250 # scales with level. More in ../../credit_management/settings.py \ No newline at end of file diff --git a/credit_management/autocredits.py b/credit_management/autocredits.py new file mode 100644 index 0000000..e30fb23 --- /dev/null +++ b/credit_management/autocredits.py @@ -0,0 +1,12 @@ +async def get_all_users(client): + users = client['nova-core']['users'] + return users + +async def update_credits(users, settings = None): + if not settings: + users.update_many({}, {"$inc": {"credits": 250}}) + + else: + for key, value in settings.items(): + users.update_many({'role': key}, {"$inc": {"credits": int(value)}}) + print(f"Updated {key} to {value}") \ No newline at end of file diff --git a/credit_management/main.py b/credit_management/main.py new file mode 100644 index 0000000..f6a0148 --- /dev/null +++ b/credit_management/main.py @@ -0,0 +1,51 @@ +import asyncio +from settings import roles +import autocredits +import aiohttp +from dotenv import load_dotenv +import os +import pymongo + +load_dotenv() + +CONNECTION_STRING = os.getenv("CONNECTION_STRING") +pymongo_client = pymongo.MongoClient(CONNECTION_STRING) + +async def main(): + users = await autocredits.get_all_users(pymongo_client) + + 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:50000/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) + + 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: + if role in roles: + bulk_updates.append(pymongo.UpdateOne({'auth.discord': int(discord)}, {'$set': {'role': role}})) + print(f"Updated {id_} to {role}") + break + if bulk_updates: + with pymongo_client: + users.bulk_write(bulk_updates) + +if __name__ == "__main__": + asyncio.run(main()) \ No newline at end of file diff --git a/credit_management/role_bot.py b/credit_management/role_bot.py new file mode 100644 index 0000000..954e6fa --- /dev/null +++ b/credit_management/role_bot.py @@ -0,0 +1,38 @@ +import nextcord +import aiohttp.web +from dotenv import load_dotenv +import os + +load_dotenv() + +TOKEN = os.getenv("BOT_TOKEN") +GUILD_ID = os.getenv("GUILD_ID") + +client = nextcord.Client(intents=nextcord.Intents.all()) + +@client.event +async def on_ready(): + # start webserver using aiohttp + app = aiohttp.web.Application() + + async def get_roles(request): + return aiohttp.web.json_response(await get_userinfo()) + + app.router.add_get('/get_roles', get_roles) + + runner = aiohttp.web.AppRunner(app) + await runner.setup() + site = aiohttp.web.TCPSite(runner, 'localhost', 50000) + await site.start() + + print('Bot is ready') + + +async def get_userinfo(): + guild = client.get_guild(int(GUILD_ID)) + members = guild.members + + user_roles = {member.id: [role.name for role in member.roles] for member in members} + return user_roles + +client.run(TOKEN) \ No newline at end of file diff --git a/credit_management/settings.py b/credit_management/settings.py new file mode 100644 index 0000000..be73b94 --- /dev/null +++ b/credit_management/settings.py @@ -0,0 +1,13 @@ +roles = { + '': '250', + 'lvl10': '280', + 'lvl20': '310', + 'lvl30': '340', + 'lvl40': '370', + 'lvl50': '400', + 'lvl60': '430', + 'lvl70': '460', + 'lvl80': '490', + 'lvl90': '520', + 'lvl100': '550', +} \ No newline at end of file