mdserver-web/plugins/tgclient/startup/tgclient.py

# coding:utf-8

import sys
import io
import os
import time
import re
import json
import base64
import threading
import asyncio
import logging

# python /Users/midoks/Desktop/mwdev/server/tgclient/tgclient.py
# cd /www/server/mdserver-web && python3 /www/server/tgclient/tgclient.py

from telethon import TelegramClient


sys.path.append(os.getcwd() + "/class/core")
import mw

logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)


def getPluginName():
    return 'tgclient'


def getPluginDir():
    return mw.getPluginDir() + '/' + getPluginName()


def getServerDir():
    return mw.getServerDir() + '/' + getPluginName()


sys.path.append(getServerDir() + "/extend")


def getConfigData():
    cfg_path = getServerDir() + "/data.cfg"
    if not os.path.exists(cfg_path):
        mw.writeFile(cfg_path, '{}')
    t = mw.readFile(cfg_path)
    return json.loads(t)


def writeConf(data):
    cfg_path = getServerDir() + "/data.cfg"
    mw.writeFile(cfg_path, json.dumps(data))
    return True


def getExtCfg():
    cfg_path = getServerDir() + "/extend.cfg"
    if not os.path.exists(cfg_path):
        mw.writeFile(cfg_path, '{}')
    t = mw.readFile(cfg_path)
    return json.loads(t)


def getStartExtCfgByTag(tag='push'):
    # 获取开启的扩展
    elist = getExtCfg()
    rlist = []
    for x in elist:
        if x['tag'] == tag and x['status'] == 'start':
            rlist.append(x)
    return rlist


def writeLog(log_str):
    if __name__ == "__main__":
        print(log_str)

    now = mw.getDateFromNow()
    log_file = getServerDir() + '/task.log'
    mw.writeFileLog(now + ':' + log_str, log_file, limit_size=5 * 1024)
    return True


# start tgbot
cfg = getConfigData()
while True:
    cfg = getConfigData()
    if 'bot' in cfg and 'api_id' in cfg['bot']:
        if cfg['bot']['api_id'] != '' and cfg['bot']['api_id'] != 'api_id':
            break
        if cfg['bot']['api_hash'] != '' and cfg['bot']['api_hash'] != 'api_hash':
            break
    writeLog('等待输入配置,api_id')
    time.sleep(3)

client = TelegramClient('mdioks', cfg['bot']['api_id'], cfg['bot']['api_hash'])

async def plugins_run():
    plist = getStartExtCfgByTag('client')
    for p in plist:
        try:
            script = p['name'].split('.')[0]
            await __import__(script).run(client)
        except Exception as e:
            writeLog('-----client error start -------')
            writeLog(mw.getTracebackInfo())
            writeLog('-----client error end -------')

async def main(loop):
    await client.start()

    # create new task
    print('creating plugins_run task.')
    task = loop.create_task(plugins_run())
    await task

    print('It works.')
    await client.run_until_disconnected()
    task.cancel()

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main(loop))