from data import constants from data.remind import Reminders from discord.ext import commands, tasks from enums.numbers import Numbers from functions import timeFormatters from functions.database import currency, poke, prison, birthdays, stats import json import random import requests import time class Tasks(commands.Cog): def __init__(self, client): self.client = client self.bankInterest.start() self.resetPrison.start() self.resetLost.start() # self.resetPoke.start() self.checkBirthdays.start() self.updateMessageCounts.start() self.sendReminders.start() @tasks.loop(hours=1.0) async def bankInterest(self): """ Task that gives daily interest """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 0 and int(time.time()) - int(lastTasks["interest"]) > 10000: users = currency.getAllRows() bitcoinPrice = self.getCurrentBitcoinPrice() for user in users: # People in prison don't get interest if len(prison.getUser(int(user[0]))) != 0: continue if float(user[3]) != 0.0: currency.update(user[0], "investeddays", int(user[4]) + 1) profit = ((float(user[3]) + float(user[5])) * (1 + (float(user[2]) * 0.01))) - float(user[3]) # Can't exceed 1 quadrillion # Check BC as well so they can't put everything into BC to cheat the system if float(user[1]) + float(user[3]) + float(user[5]) + profit + (float(user[8]) * bitcoinPrice) > Numbers.q.value: # In case adding profit would exceed 1q, only add the difference profit = Numbers.q.value - float(user[1]) - float(user[3]) - float(user[5]) - (float(user[8]) * bitcoinPrice) # Don't reduce the current profit if Dinks were gained some other way (rob, bc, ...) if profit > 0: currency.update(user[0], "profit", float(user[5]) + profit) await self.client.get_user(int(user[0])).send("Je hebt de invest-limiet van 1Q Didier Dinks bereikt.\nIndien je nog meer Didier Dinks wil sparen, kan je 1q Didier Dinks omruilen voor een Platinum Dink in de shop.") else: currency.update(user[0], "profit", float(user[5]) + profit) lastTasks["interest"] = int(round(time.time())) with open("files/lastTasks.json", "w") as fp: json.dump(lastTasks, fp) @bankInterest.before_loop async def beforeBankInterest(self): await self.client.wait_until_ready() @tasks.loop(hours=1.0) async def resetLost(self): """ Task that resets Lost Today """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 0 and int(time.time()) - int(lastTasks["lost"]) > 10000: with open("files/lost.json", "r") as fp: fc = json.load(fp) fc["today"] = 0 with open("files/lost.json", "w") as fp: json.dump(fc, fp) lastTasks["lost"] = round(time.time()) with open("files/lastTasks.json", "w") as fp: json.dump(lastTasks, fp) @resetLost.before_loop async def beforeResetLost(self): await self.client.wait_until_ready() @tasks.loop(hours=6.0) async def resetPoke(self): """ Task that resets Poke """ if int(time.time()) - int(poke.get()[1]) > 259200: await self.client.get_guild(int(self.client.constants.CallOfCode))\ .get_channel(int(self.client.constants.DidierPosting))\ .send("Poke is gereset door inactiviteit. <@!{}> is hem!".format(int(poke.reset()))) @resetPoke.before_loop async def beforeResetPoke(self): await self.client.wait_until_ready() @tasks.loop(hours=1.0) async def resetPrison(self): """ Task that lowers prison time daily """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 0 and int(time.time()) - int(lastTasks["prison"]) > 10000: prison.dailyLowers() with open("files/lastTasks.json", "w") as fp: lastTasks["prison"] = round(time.time()) json.dump(lastTasks, fp) @resetPrison.before_loop async def beforeResetPrison(self): await self.client.wait_until_ready() @tasks.loop(hours=1.0) async def checkBirthdays(self): """ Task that wishes people a happy birthday """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 6 and int(time.time()) - int(lastTasks["birthdays"]) > 10000: dt = timeFormatters.dateTimeNow() res = birthdays.get_users_on_date(dt.day, dt.month) COC = self.client.get_guild(int(constants.CallOfCode)) people = [COC.get_member(int(user[0])) for user in res] general = COC.get_channel(int(constants.CoCGeneral)) lastTasks["birthdays"] = round(time.time()) with open("files/lastTasks.json", "w") as fp: json.dump(lastTasks, fp) if not people: return if len(people) == 1: return await general.send("Gelukkige verjaardag {}!".format(people[0].mention)) return await general.send("Gelukkige verjaardag {} en {}!".format( ", ".join(user.mention for user in people[:-1]), people[-1].mention )) @checkBirthdays.before_loop async def beforecheckBirthdays(self): await self.client.wait_until_ready() @tasks.loop(hours=1.0) async def updateMessageCounts(self): """ Task that updates the activity counter for channels """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 0 and int(time.time()) - int(lastTasks["channels"]) > 10000: channels = stats.channel_activity() for channel in channels: stats.lower_channel(int(channel[0]), 0.95 * float(channel[1])) with open("files/lastTasks.json", "w") as fp: lastTasks["channels"] = round(time.time()) json.dump(lastTasks, fp) @updateMessageCounts.before_loop async def beforeupdateMessageCounts(self): await self.client.wait_until_ready() @tasks.loop(hours=1.0) async def sendReminders(self): """ Task that sends people daily reminders """ # Don't do it multiple times a day if bot dc's, ... with open("files/lastTasks.json", "r") as fp: lastTasks = json.load(fp) if int(self.getCurrentHour()) == 7 and int(time.time()) - int(lastTasks["remind"]) > 10000: reminders = Reminders() weekday = self.getCurrentWeekday() for category in reminders.categories: # Check if this reminder is temporarily disabled if category["disabled"]: continue # Checks if this reminder can be sent on weekdays if (not category["weekends"]) and weekday > 4: continue for user in category["users"]: userInstance = self.client.get_user(user) # User can't be fetched for whatever reason, ignore instead of crashing if userInstance is None: continue # Check if a special embed has to be attached for this reminder if "embed" not in category: await userInstance.send(random.choice(category["messages"])) else: await userInstance.send(random.choice(category["messages"]), embed=category["embed"]) with open("files/lastTasks.json", "w") as fp: lastTasks["remind"] = round(time.time()) json.dump(lastTasks, fp) @sendReminders.before_loop async def beforeSendReminders(self): await self.client.wait_until_ready() def getCurrentHour(self): return timeFormatters.dateTimeNow().hour def getCurrentWeekday(self): return timeFormatters.dateTimeNow().weekday() def getCurrentBitcoinPrice(self): result = requests.get("https://api.coindesk.com/v1/bpi/currentprice.json").json() currentPrice = result["bpi"]["EUR"]["rate_float"] return float(currentPrice) def setup(client): client.add_cog(Tasks(client))