diff --git a/README.md b/README.md new file mode 100644 index 0000000..00f6fa2 --- /dev/null +++ b/README.md @@ -0,0 +1,2 @@ +# 介绍 +电报读秀查询机器人 diff --git a/bot.py b/bot.py deleted file mode 100644 index 4fd3c6b..0000000 --- a/bot.py +++ /dev/null @@ -1,116 +0,0 @@ -import telebot -from typing import NoReturn -from time import sleep -from asyncio import sleep -import requests -import re -import pymysql,telegram -from telegram import Update -from telegram.ext import Updater, CommandHandler, CallbackContext -import requests -#tg机器人id -bot_api = "58023:AAH2DlZY-eRswzx_niNVeQ" -import pymysql #连接数据库 -connect = pymysql.connect(host='127.0.0.1', # 本地数据库 - user='duxiu', - password='KNFP5Zf7bM', - db='duxiu', - charset='utf8') #服务器名,账户,密码,数据库名称 -db = connect.cursor() - - - -def start(update: Update, context: CallbackContext): - update.message.reply_text('请输入:读秀SS号,输入格式为:/ss 14061086') -def ss(update: Update, context: CallbackContext) -> None: - text = update.message.text - names = text.split(' ') - name = names[1] - content = '^[0-9]{8}$' - result = re.findall(content, str(name)) - if len(result) != 0: - sql = "SELECT * FROM FileObject WHERE name LIKE '%" + name + "%';" - db.execute(sql) - result = db.fetchall() - if len(result) !=0: - fs_id = result[0][1] - ss = duxiu(fs_id) - update.message.reply_text(ss) - else: - update.message.reply_text("数据库没有该书集") - - else: - update.message.reply_text("输入的读秀SS格式错误") - - - print(name) - print(ss) - -#百度网盘连接生成功能 -def duxiu(ss): - session = requests.session() - # BDUSS和STOKEN的值从cookies种获取 - session.cookies["BDUSS"] = 'G2hIMmNTTHRVM283WTNmajFmMHAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAPVuj2P1bo9jY' - session.cookies["STOKEN"] = '12da7949ba2a7EI1QjBFQkQ0MkE4Qzg6U0w9MDpOUj0xMDpGRz0x&clienttype=0' - - headerss = { - 'Host': 'pan.baidu.com', - 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/77.0.3865.120 Safari/537.36', - } - response = session.get('https://pan.baidu.com/', headers=headerss) - if(not False): - pwd = 1123 - form_data = { - 'schannel': 4, - 'channel_list': '[]', - 'period': 7, - 'pwd': pwd, - 'fid_list': str([int(ss)]), - } - print(str([ss])) - # 补充cookie的值 - headers = { - "Accept":"*/*" , - "Accept-Encoding":"gzip, deflate, br" , - "Accept-Language":"zh-CN,zh;q=0.9" , - "Cache-Control":"no-cache" , - "Connection":"keep-alive" , - "Content-Length":"16930" , - "Content-Type":"application/x-www-form-urlencoded; charset=UTF-8" , - "Cookie":"newlogin=1; BAIDUID=80070C95EF42:FG=1; BDUSS=GhHLlyj2O5co9jR; BDUSS_BFESS=GhHLVYwemRLlyj2O5co9jR; ab_sr=1.0.1_MTU2ODJmYmQzYzcwODI4MmM3ZTc5YTE2OTNlNjBjZGRkZDY2ODE3ZWEyMmQyY2Q3MDhkNDg2ZjQxODZkMTJkNjU1ZWYxZjM2NGI0MmUyOTE3YTcxM2EwYTgyNWEwOWRjYTMyZTEzZTc1NTMxZTA4ZjgzNjNlMTliNWEzMjRkZTdmMzEzYmE2MmQ5Mjk4YWRlNmNjOTEwNjg4Y2Q3NGNiMQ==" , - "Host":"pan.baidu.com" , - "Origin":"https://pan.baidu.com" , - "Pragma":"no-cache" , - "Referer":"https://pan.baidu.com/disk/home?" , - "Sec-Fetch-Dest":"empty" , - "Sec-Fetch-Mode":"cors" , - "Sec-Fetch-Site":"same-origin" , - "User-Agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.190 Safari/537.36" , - "X-Requested-With":"XMLHttpRequest" - } - responsess = session.post(url, headers=headers, data=form_data) - print(responsess.headers) - if(responsess.json()['errno'] == 0): - print({'errno': 0, 'err_msg': '创建分享链接成功!', 'info': {'link': responsess.json()['link'], 'pwd': pwd}}) - return {'link': responsess.json()['link'], 'pwd': pwd} - else: - print( {'errno': 1, 'err_msg': '创建分享链接失败!', 'info': responsess.json()}) - return "创建分享链接失败!" - - - -def main() -> None: - updater = Updater(bot_api) - - dispatcher = updater.dispatcher - - # 机器人命令 - dispatcher.add_handler(CommandHandler("start", start)) - dispatcher.add_handler(CommandHandler("ss", ss)) - # 启动机器人,勿删 - updater.start_polling() - updater.idle() - - -if __name__ == '__main__': - main()