from data import constants 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 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() @tasks.loop(hours=1.0) async def bankInterest(self): # 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): # 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): 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): # 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): # 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): # 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() def getCurrentHour(self): return timeFormatters.dateTimeNow().hour 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))