142 lines
4.0 KiB
Python
142 lines
4.0 KiB
Python
""" PagerMaid initialization. """
|
||
|
||
import posthog
|
||
from os import getcwd, makedirs
|
||
from os.path import exists
|
||
from sys import version_info, platform
|
||
from yaml import load, FullLoader
|
||
from shutil import copyfile
|
||
from redis import StrictRedis
|
||
from logging import getLogger, INFO, DEBUG, StreamHandler
|
||
from distutils2.util import strtobool
|
||
from coloredlogs import ColoredFormatter
|
||
from telethon import TelegramClient
|
||
|
||
persistent_vars = {}
|
||
module_dir = __path__[0]
|
||
working_dir = getcwd()
|
||
logging_format = "%(levelname)s [%(asctime)s] [%(name)s] %(message)s"
|
||
config = None
|
||
help_messages = {}
|
||
posthog.api_key = '1WepU-o7JwNKYqPNymWr_mrCu3RVPD-p28PUikPDfsI'
|
||
logs = getLogger(__name__)
|
||
logging_handler = StreamHandler()
|
||
logging_handler.setFormatter(ColoredFormatter(logging_format))
|
||
logs.addHandler(logging_handler)
|
||
logs.setLevel(INFO)
|
||
|
||
try:
|
||
config = load(open(r"config.yml"), Loader=FullLoader)
|
||
except FileNotFoundError:
|
||
logs.fatal("出错了呜呜呜 ~ 配置文件不存在，正在生成新的配置文件。")
|
||
copyfile(f"{module_dir}/assets/config.gen.yml", "config.yml")
|
||
exit(1)
|
||
|
||
if strtobool(config['debug']):
|
||
logs.setLevel(DEBUG)
|
||
else:
|
||
logs.setLevel(INFO)
|
||
|
||
if platform == "linux" or platform == "linux2" or platform == "darwin" or platform == "freebsd7" \
|
||
or platform == "freebsd8" or platform == "freebsdN" or platform == "openbsd6":
|
||
logs.info(
|
||
"将平台检测为“ " + platform + "，进入PagerMaid的早期加载过程。"
|
||
)
|
||
else:
|
||
logs.error(
|
||
"出错了呜呜呜 ~ 你的平台 " + platform + " 不支持运行 PagerMaid，请在Linux或 *BSD 上启动 PagerMaid。"
|
||
)
|
||
exit(1)
|
||
|
||
if version_info[0] < 3 or version_info[1] < 6:
|
||
logs.error(
|
||
"出错了呜呜呜 ~ 请将您的 python 升级到至少3.6版。"
|
||
)
|
||
exit(1)
|
||
|
||
if not exists(f"{getcwd()}/data"):
|
||
makedirs(f"{getcwd()}/data")
|
||
|
||
api_key = config['api_key']
|
||
api_hash = config['api_hash']
|
||
try:
|
||
proxy_addr = config['proxy_addr'].strip()
|
||
proxy_port = config['proxy_port'].strip()
|
||
mtp_addr = config['mtp_addr'].strip()
|
||
mtp_port = config['mtp_port'].strip()
|
||
mtp_secret = config['mtp_secret'].strip()
|
||
except:
|
||
proxy_addr = ''
|
||
proxy_port = ''
|
||
mtp_addr = ''
|
||
mtp_port = ''
|
||
mtp_secret = ''
|
||
try:
|
||
redis_host = config['redis']['host']
|
||
except KeyError:
|
||
redis_host = 'localhost'
|
||
try:
|
||
redis_port = config['redis']['port']
|
||
except KeyError:
|
||
redis_port = 6379
|
||
try:
|
||
redis_db = config['redis']['db']
|
||
except KeyError:
|
||
redis_db = 14
|
||
if api_key is None or api_hash is None:
|
||
logs.info(
|
||
"出错了呜呜呜 ~ 请在工作目录中放置一个有效的配置文件。"
|
||
)
|
||
exit(1)
|
||
|
||
if not proxy_addr == '' and not proxy_port == '':
|
||
try:
|
||
import socks
|
||
except:
|
||
pass
|
||
bot = TelegramClient("pagermaid", api_key, api_hash, auto_reconnect=True, proxy=(socks.SOCKS5, proxy_addr, int(proxy_port)))
|
||
elif not mtp_addr == '' and not mtp_port == '' and not mtp_secret == '':
|
||
from telethon import connection
|
||
bot = TelegramClient("pagermaid", api_key, api_hash, auto_reconnect=True,
|
||
connection=connection.ConnectionTcpMTProxyRandomizedIntermediate,
|
||
proxy=(mtp_addr, int(mtp_port), mtp_secret))
|
||
else:
|
||
bot = TelegramClient("pagermaid", api_key, api_hash, auto_reconnect=True)
|
||
redis = StrictRedis(host=redis_host, port=redis_port, db=redis_db)
|
||
|
||
async def upload_name():
|
||
me = await bot.get_me()
|
||
try:
|
||
posthog.identify(str(me.id), {
|
||
'name': str(me.first_name)
|
||
})
|
||
logs.info(
|
||
"上报用户名称成功。"
|
||
)
|
||
except:
|
||
logs.info(
|
||
"上报用户名称出错了呜呜呜 ~"
|
||
)
|
||
|
||
with bot:
|
||
bot.loop.run_until_complete(upload_name())
|
||
|
||
def redis_status():
|
||
try:
|
||
redis.ping()
|
||
return True
|
||
except BaseException:
|
||
return False
|
||
|
||
|
||
async def log(message):
|
||
logs.info(
|
||
message.replace('`', '\"')
|
||
)
|
||
if not strtobool(config['log']):
|
||
return
|
||
await bot.send_message(
|
||
int(config['log_chatid']),
|
||
message
|
||
)
|