54 lines
2.7 KiB
Python
54 lines
2.7 KiB
Python
from bot import db
|
|
from schedule import every, clear, run_pending
|
|
import asyncio
|
|
from datetime import datetime, timedelta
|
|
from loguru import logger
|
|
|
|
daysoftheweek = ("monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday")
|
|
|
|
async def save_activity():
|
|
day = (datetime.now()-timedelta(1)).weekday()
|
|
async for data in db.history.find({"type": "server"}):
|
|
await db.history.update_one(
|
|
{"type": "server", "id": data['id']},
|
|
{
|
|
"$push":
|
|
{f"avarage.{i}": data['current'][str(i)] for i in range(24)} | {f"history.{daysoftheweek[day]}.{i}": data['current'][str(i)] for i in range(24)},
|
|
"$set":
|
|
{f'yesterday.{i}': data['current'][str(i)] for i in range(24)} | {f"current.{i}": 0 for i in range(24)}
|
|
}
|
|
)
|
|
|
|
async def db_sync_hour():
|
|
async for member in db.members.find():
|
|
if member['history']['hour'][list(member['history']['hour'].keys())[-1]] != member['exp']:
|
|
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"history.hour.{int(datetime.now().timestamp())}": member["exp"]}})
|
|
if 'guild_stat' in member.keys():
|
|
for guild in member["guild_stat"].keys():
|
|
if member['guild_stat'][guild]['history']['hour'][list(member['guild_stat'][guild]['history']['hour'].keys())[-1]] != member['guild_stat'][guild]['exp']:
|
|
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"guild_stat.{guild}.history.hour.{int(datetime.now().timestamp())}": member["guild_stat"][guild]["exp"]}})
|
|
|
|
async def db_sync_day():
|
|
logger.info("Регистрация за день")
|
|
async for member in db.members.find():
|
|
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"history.day.{int(datetime.now().timestamp())}": member["exp"]}})
|
|
for guild in member["guild_stat"].keys():
|
|
await db.members.update_one({"_id": member["_id"]}, {"$set": {f"guild_stat.{guild}.history.day.{int(datetime.now().timestamp())}": member["guild_stat"][guild]["exp"]}})
|
|
logger.info("Регистрация завершена")
|
|
# await self.bot.get_user(self.bot.owner_id).send("Завершено за день????")
|
|
|
|
|
|
async def main(bot):
|
|
clear()
|
|
|
|
every().day.at('00:00:00').do(lambda: asyncio.run_coroutine_threadsafe(save_activity(), bot.loop))
|
|
every().hours.at("00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_hour(), bot.loop))
|
|
# every().day.at("00:00:00").do(lambda: asyncio.run_coroutine_threadsafe(db_sync_day(), bot.loop))
|
|
|
|
while True:
|
|
run_pending()
|
|
await asyncio.sleep(1)
|
|
|
|
async def setup(bot):
|
|
asyncio.create_task(main(bot))
|