mirror of
https://github.com/Xtao-Labs/sticker-captcha-bot.git
synced 2024-12-04 05:12:00 +00:00
93 lines
2.3 KiB
Python
93 lines
2.3 KiB
Python
import contextlib
|
|
import sys
|
|
from enum import Enum
|
|
|
|
from cashews import cache
|
|
from datetime import datetime, timezone
|
|
from logging import getLogger, StreamHandler, CRITICAL, INFO, basicConfig, DEBUG
|
|
|
|
from coloredlogs import ColoredFormatter
|
|
|
|
from sticker.bot import bot
|
|
from sticker.config import Config
|
|
from sticker.scheduler import scheduler
|
|
|
|
cache.setup("mem://")
|
|
logs = getLogger(__name__)
|
|
logging_format = "%(levelname)s [%(asctime)s] [%(name)s] %(message)s"
|
|
logging_handler = StreamHandler()
|
|
logging_handler.setFormatter(ColoredFormatter(logging_format))
|
|
root_logger = getLogger()
|
|
root_logger.setLevel(DEBUG if Config.DEBUG else CRITICAL)
|
|
root_logger.addHandler(logging_handler)
|
|
pyro_logger = getLogger("pyrogram")
|
|
pyro_logger.setLevel(INFO if Config.DEBUG else CRITICAL)
|
|
pyro_logger.addHandler(logging_handler)
|
|
basicConfig(level=DEBUG if Config.DEBUG else INFO)
|
|
logs.setLevel(DEBUG if Config.DEBUG else INFO)
|
|
|
|
# easy check
|
|
if not Config.API_ID:
|
|
logs.error("Api-ID Not Found!")
|
|
sys.exit(1)
|
|
elif not Config.API_HASH:
|
|
logs.error("Api-Hash Not Found!")
|
|
sys.exit(1)
|
|
|
|
start_time = datetime.now(timezone.utc)
|
|
|
|
with contextlib.suppress(ImportError):
|
|
import uvloop # noqa
|
|
|
|
uvloop.install()
|
|
|
|
if not scheduler.running:
|
|
scheduler.start()
|
|
|
|
|
|
class LogAction(str, Enum):
|
|
FAIL_ERROR = "FAIL_ERROR"
|
|
FAIL_TIMEOUT = "FAIL_TIMEOUT"
|
|
ACCEPT = "ACCEPT"
|
|
NEW_GROUP = "NEW_GROUP"
|
|
REQUEST = "REQUEST"
|
|
|
|
|
|
async def log(chat, user, action: LogAction):
|
|
scheduler.add_job(
|
|
_log,
|
|
args=[chat, user, action],
|
|
replace_existing=False,
|
|
)
|
|
|
|
|
|
async def _log(chat, user, action: LogAction):
|
|
if not Config.LOG_CHANNEL:
|
|
return
|
|
me = await bot.get_me()
|
|
event = {
|
|
LogAction.FAIL_ERROR: "回答错误",
|
|
LogAction.FAIL_TIMEOUT: "回答超时",
|
|
LogAction.ACCEPT: "通过验证",
|
|
LogAction.NEW_GROUP: "加入群组",
|
|
LogAction.REQUEST: "发起验证",
|
|
}
|
|
msg = """#%s
|
|
群组: %s
|
|
群组id: <code>%s</code>
|
|
用户: #id%s
|
|
昵称：<code>%s</code>
|
|
OPBot: #bot%s
|
|
事件: %s"""
|
|
user_name = user.full_name if user else ""
|
|
msg %= (
|
|
action.value,
|
|
chat.title,
|
|
chat.id,
|
|
user.id if user else "",
|
|
user_name,
|
|
me.id,
|
|
event[action],
|
|
)
|
|
await bot.send_message(Config.LOG_CHANNEL, msg)
|