natsuko/cogs/timeble.py

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))