diff --git a/rewardsystem/autocredits.py b/rewardsystem/autocredits.py index 10afb91..7971fab 100644 --- a/rewardsystem/autocredits.py +++ b/rewardsystem/autocredits.py @@ -1,12 +1,15 @@ async def get_all_users(client): users = client['nova-core']['users'] return users - -async def update_credits(users, settings=None): + + +async def update_credits(pymongo_client, settings=None): + users = await get_all_users(pymongo_client) + if not settings: - users.update_many({}, {"$inc": {"credits": 250}}) + users.update_many({}, {"$inc": {"credits": 2500}}) else: for key, value in settings.items(): - users.update_many({'role': key}, {"$inc": {"credits": int(value)}}) - print(f'Updated {key} to {value}') + users.update_many( + {'level': key}, {"$inc": {"credits": int(value)}})