main.py

import os
import sys
import string
import random
import time
import subprocess
import traceback
from time import sleep


if not os.path.exists("config.py"):
    print("Error: 配置文件不存在，请检查是否正确配置！")
    exit(0)
import config

if not os.path.exists("data.py"):
    print("Error: 数据文件不存在，请检查是否正确配置！")
    exit(0)
from data import *


try:  # 向下兼容
    if config.USE_SENTRY:  # Bug Tracker
        import sentry_sdk

        sentry_sdk.init(config.SENTRY_LINK)
except Exception as e:
    ret = str(e)


xx = ""


def getReportID():
    seed = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
    sa = []
    for i in range(8):
        sa.append(random.choice(seed))
    salt = "".join(sa)
    return salt


totalcount = 0
chatcount = 0
inlinecount = 0


def getanother():

    xx = " "
    xx += "\r\n"
    xx += "    "
    return xx


samepoint = 2


def shuffle(List):
    global samepoint
    pool = list(List) * samepoint
    while True:
        random.shuffle(pool)
        for element in pool:
            yield element


nextsays = shuffle(text)
nextquote = shuffle(quotes)


def getquotes():
    global nextquote
    xx = next(nextquote)
    xx = xx.replace("a", random.choice(frontsays))
    xx = xx.replace("b", random.choice(backsays))
    return xx


def Process(msg, maxlength):
    global nextsays
    xx = msg
    tmp = str()
    while len(tmp) < maxlength:
        branch = random.randint(0, 100)
        if branch < 5:
            tmp += getanother()
            while tmp[-1] == "，":
                tmp += next(nextsays)
            while tmp[-1] == "：":
                tmp += next(nextsays)

        elif branch < 20:
            tmp += getquotes()
        else:
            tmp += next(nextsays)
    while tmp[-1] == "，":
        tmp += next(nextsays)
    while tmp[-1] == "：":
        tmp += next(nextsays)
    tmp = "    " + tmp
    tmp = tmp.replace("x", xx)
    # print(tmp)
    return tmp


def xiaobMode(qtext, fr):
    tmp = (
        xiaobtext_front.replace("%ss", qtext) + fr + xiaobtext_end.replace("%ss", qtext)
    )
    return tmp


import logging
import telebot

bot = telebot.TeleBot(config.TOKEN)
from telebot import apihelper
from telebot import types
from telebot import util

if config.USE_PROXY:
    apihelper.proxy = config.HTTP_PROXY
# telebot.logger.setLevel(logging.DEBUG)

logging.basicConfig(level=logging.INFO)


try:
    ghash = str(
        subprocess.check_output(["git", "rev-parse", "--short", "HEAD"]).decode("utf-8")
    )
    gtime = str(
        subprocess.check_output(["git", "log", '--pretty=format:"%cd"', "-1"]).decode(
            "utf-8"
        )
    )
except Exception as e:
    rid = getReportID()
    traceback.print_exc()
    logging.error(str(rid + "::" + str(repr(e)) + "\n" + traceback.format_exc()))


def get_githash():
    global ghash
    return ghash


def get_gittime():
    global gtime
    return gtime


def getstat(mode):
    global totalcount
    global inlinecount
    global chatcount
    ret = '欢迎使用"狗屁不通生成器"的 Telegram 移植版本！\n'
    ret += "源码：https://github.com/abc1763613206/TGBullshitGeneratorBot\n"
    ret += "当前 commit 版本 {}最后更新于 {}\n".format(get_githash(), get_gittime())
    ret += data_stat()
    ret += (
        "\n自上一次重启以来，总调用{}次，其中 inline 模式{}次，私聊模式{}次\n".format(
            str(totalcount), str(inlinecount), str(chatcount)
        )
    )
    if mode == 0:
        ret += "您正在使用 inline 模式，如有疑问请 @abc1763613206"
    else:
        ret += "您正在使用 私聊 模式，如有疑问请 @abc1763613206"
    return ret


@bot.message_handler(commands=["start", "help"])
def send_welcome(message):
    bot.reply_to(
        message,
        "狗屁不通生成器(https://github.com/menzi11/BullshitGenerator) 的 Telegram 移植版本，如有疑问请 @abc1763613206\n源码： https://github.com/abc1763613206/TGBullshitGeneratorBot",
    )


@bot.message_handler(commands=["stat"])
def send_stat(message):
    try:
        global totalcount
        totalcount += 1
        global chatcount
        chatcount += 1
        bot.reply_to(message, getstat(1))
    except Exception as e:
        traceback.print_exc()
        logging.error(e)


@bot.message_handler(func=lambda m: True)
def echo_all(message):
    try:
        global totalcount
        totalcount += 1
        global chatcount
        chatcount += 1
        if message.content_type != "text":
            bot.reply_to(message, "抱歉，我还只支持文字！")
        logging.info(
            str(
                "Chat::@{} ({} {}): {}".format(
                    message.from_user.username,
                    message.from_user.first_name,
                    message.from_user.last_name,
                    message.text,
                )
            )
        )
        # print(message.type)

        # print(type(message.from_user))
        # if len(message.text) > 15:
        #    bot.reply_to(message,'你说的内容太长了！何不切分一下试试？')
        #    return
        ret = Process(message.text, 1000)
        if len(ret) >= 4500:
            bot.reply_to(message, "您要说的内容实在太长了，将为您分段处理！")
            splitted_ret = util.split_string(ret, 3000)
            logging.info(
                str("Too long.Split into " + str(len(splitted_ret)) + " parts")
            )
            for sret in splitted_ret:
                bot.reply_to(message, sret)
                # sleep(0.3)
        else:
            bot.reply_to(message, ret)

    except Exception as e:
        rid = getReportID()
        traceback.print_exc()
        logging.error(str(rid + "::" + str(repr(e)) + "\n" + traceback.format_exc()))
        bot.send_message(
            message.from_user.id,
            "抱歉，我出现错误，识别ID为 "
            + rid
            + "\n\n信息如下：\n"
            + str(repr(e))
            + "\n\n请将这条信息 Forward 给 @abc1763613206 中所列的用户 进行处理！",
        )


# @bot.inline_handler(lambda query: SetQText(query.query) )
@bot.inline_handler(lambda query: query.query != "")
def query_text(inline_query):
    global totalcount
    totalcount += 1
    global inlinecount
    inlinecount += 1
    qtext = inline_query.query
    if qtext == "":
        pass
    logging.info(
        str("Inline:: @{} ({} {}): {}").format(
            inline_query.from_user.username,
            inline_query.from_user.first_name,
            inline_query.from_user.last_name,
            inline_query.query,
        )
    )

    try:
        if len(qtext) > 15:
            r5 = types.InlineQueryResultArticle(
                "1",
                "你要说的内容太长了！将直接发送原消息(小提示：私聊模式可以发送长消息哦！)",
                types.InputTextMessageContent(str(qtext)),
            )
            r6 = types.InlineQueryResultArticle(
                "2", "查看当前统计(Beta)", types.InputTextMessageContent(getstat(0))
            )
            bot.answer_inline_query(inline_query.id, [r5, r6], cache_time=1)
        else:
            r1 = types.InlineQueryResultArticle(
                "1", "小作文(200字)", types.InputTextMessageContent(Process(qtext, 200))
            )
            r2 = types.InlineQueryResultArticle(
                "2",
                "普通玩法(600字)",
                types.InputTextMessageContent(Process(qtext, 600)),
            )
            r3 = types.InlineQueryResultArticle(
                "3",
                "加强玩法(1000字)",
                types.InputTextMessageContent(Process(qtext, 1000)),
            )
            r4 = types.InlineQueryResultArticle(
                "4",
                "再多来些(1500字)",
                types.InputTextMessageContent(Process(qtext, 1500)),
            )
            r5 = types.InlineQueryResultArticle(
                "5",
                "与小编联动(@xiaobbot)(400字)",
                types.InputTextMessageContent(xiaobMode(qtext, Process(qtext, 200))),
            )
            r6 = types.InlineQueryResultArticle(
                "6",
                "小编(@xiaobbot)想多说点(800字)",
                types.InputTextMessageContent(xiaobMode(qtext, Process(qtext, 600))),
            )
            r7 = types.InlineQueryResultArticle(
                "7", "查看当前统计(Beta)", types.InputTextMessageContent(getstat(0))
            )
            bot.answer_inline_query(
                inline_query.id, [r1, r2, r3, r4, r5, r6, r7], cache_time=1
            )
    except Exception as e:
        rid = getReportID()
        traceback.print_exc()
        logging.error(str(rid + "::" + str(repr(e)) + "\n" + traceback.format_exc()))


@bot.inline_handler(lambda query: query.query == "")
def query_text(inline_query):
    try:
        r1 = types.InlineQueryResultArticle(
            "1",
            "查看当前统计(Beta)(不会即时更新)",
            types.InputTextMessageContent(getstat(0)),
        )
        r2 = types.InlineQueryResultArticle(
            "2",
            "您正在使用 inline 模式，请输入内容以便生成",
            types.InputTextMessageContent("？"),
        )
        bot.answer_inline_query(inline_query.id, [r1, r2], cache_time=360)
    except Exception as e:
        rid = getReportID()
        traceback.print_exc()
        logging.error(str(rid + "::" + str(repr(e)) + "\n" + traceback.format_exc()))


# @bot.inline_handler(lambda query: True)
# def query_text(inline_query):
#    try:
#        r = types.InlineQueryResultArticle('1', 'Result', types.InputTextMessageContent('Result message.'))
#        r2 = types.InlineQueryResultArticle('2', 'Result2', types.InputTextMessageContent('Result message2.'))
#        bot.answer_inline_query(inline_query.id, [r, r2])
#    except Exception as e:
#        print(e)


def main_loop():
    bot.polling(True)
    while 1:
        time.sleep(3)


if __name__ == "__main__":
    try:
        logging.info(data_stat())
        logging.info(
            "当前 commit 版本 {}最后更新于 {}\n".format(get_githash(), get_gittime())
        )  # git supported
        logging.info("准备处理信息")
        main_loop()
    except KeyboardInterrupt:
        print("\nExiting by user request.\n")
        exit(0)
    except Exception as e:
        rid = getReportID()
        traceback.print_exc()
        logging.error(str(rid + "::" + str(repr(e)) + "\n" + traceback.format_exc()))