64 lines
2.5 KiB
Python
64 lines
2.5 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")
|
|
|
|
def save_activity():
|
|
day = (datetime.now()-timedelta(1)).weekday()
|
|
for data in db.history.find({"type": "server"}):
|
|
db.history.update_one(
|
|
{"type": "server", "id": data['id']},
|
|
{
|
|
"$push":
|
|
{f"avarage.{i}": data['current'][str(i)] for i in range(24)},
|
|
"$set":
|
|
{f'yesterday.{i}': data['current'][str(i)] for i in range(24)}
|
|
}
|
|
)
|
|
db.history.update_one(
|
|
{"type": "server", "id": data['id']},
|
|
{
|
|
"$push":
|
|
{f"history.{daysoftheweek[day]}.{i}": data['current'][str(i)] for i in range(24)},
|
|
"$set":
|
|
{f"current.{i}": 0 for i in range(24)}
|
|
}
|
|
)
|
|
|
|
def db_sync_hour():
|
|
logger.info("Регистрация за час")
|
|
for member in list(db.members.find()):
|
|
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():
|
|
db.members.update_one({"_id": member["_id"]}, {"$set": {f"guild_stat.{guild}.history.hour.{int(datetime.now().timestamp())}": member["guild_stat"][guild]["exp"]}})
|
|
logger.info("Регистрация завершена")
|
|
|
|
|
|
def db_sync_day():
|
|
logger.info("Регистрация за день")
|
|
for member in list(db.members.find()):
|
|
db.members.update_one({"_id": member["_id"]}, {"$set": {f"history.day.{int(datetime.now().timestamp())}": member["exp"]}})
|
|
for guild in member["guild_stat"].keys():
|
|
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():
|
|
clear()
|
|
|
|
every().day.at('00:00:00').do(save_activity)
|
|
every().hours.at("00:00").do(db_sync_hour)
|
|
every().day.at("00:00:00").do(db_sync_day)
|
|
|
|
while True:
|
|
run_pending()
|
|
await asyncio.sleep(1)
|
|
|
|
async def setup(_):
|
|
asyncio.create_task(main())
|