mirror of
https://github.com/holll/teledown.git
synced 2025-01-07 03:07:03 +08:00
127 lines
4.7 KiB
Python
127 lines
4.7 KiB
Python
import argparse
|
||
import json
|
||
import os
|
||
import sys
|
||
|
||
from telethon import TelegramClient
|
||
|
||
from tools.down_file import down_group
|
||
from tools.monit import StartMonit
|
||
from tools.tool import print_all_channel, Hook, print_group, initDb, md5
|
||
from tools.upload_file import upload_file
|
||
|
||
# 1.创建解释器
|
||
parser = argparse.ArgumentParser()
|
||
# 2.创建一个互斥参数组
|
||
mutex_group = parser.add_mutually_exclusive_group()
|
||
# 3.添加需要的参数
|
||
parser.add_argument('-c', '--config', default='config.json', help='配置文件')
|
||
parser.add_argument('-re', '--refresh', action='store_true', help='刷新缓存')
|
||
mutex_group.add_argument('-up', '--upload', action='store_true', help='上传文件')
|
||
mutex_group.add_argument('-down', '--download', action='store_true', help='下载文件')
|
||
mutex_group.add_argument('-print', action='store_true', help='打印消息')
|
||
mutex_group.add_argument('-m', '--monit', action='store_true', help='监控频道')
|
||
parser.add_argument('-id', help='频道ID')
|
||
parser.add_argument('-user', help='指定下载用户', default=None)
|
||
parser.add_argument('--range', default='>0', help='下载范围')
|
||
parser.add_argument('-path', help='上传路径')
|
||
parser.add_argument('-dau', default='N', choices=['y', 'Y', 'n', 'N'], help='上传完成删除原文件')
|
||
parser.add_argument('-at', '--addtag', help='增加tag')
|
||
# 3.进行参数解析
|
||
args = parser.parse_args()
|
||
config_path = args.config
|
||
# 配置处理开始
|
||
with open(config_path, 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
api_id = config.get('api_id')
|
||
api_hash = config.get('api_hash')
|
||
phone = config.get('phone')
|
||
bot_token = config.get('bot_token')
|
||
alias = config.get('alias')
|
||
if (phone is not None and bot_token is not None) or (phone is None and bot_token is None):
|
||
print('请确认使用机器人登录还是电话号码登录')
|
||
exit()
|
||
if phone:
|
||
md5Token = md5(phone)
|
||
else:
|
||
md5Token = md5(bot_token)
|
||
if alias:
|
||
for _id in alias:
|
||
os.environ[_id] = alias[_id]
|
||
initDb(md5Token)
|
||
os.environ['save_path'] = save_path = config.get('save_path')
|
||
proxy = config.get('proxy')
|
||
if proxy is not None:
|
||
import python_socks
|
||
username = None
|
||
password = None
|
||
if '@' in proxy:
|
||
username = proxy.split('@')[0].split(':')[0]
|
||
password = proxy.split('@')[0].split(':')[1]
|
||
addr = proxy.split('@')[1].split(':')[0]
|
||
port = proxy.split('@')[1].split(':')[1]
|
||
else:
|
||
addr = proxy.split(':')[0]
|
||
port = proxy.split(':')[1]
|
||
proxy = {
|
||
'proxy_type': python_socks.ProxyType.SOCKS5, # (mandatory) protocol to use (see above)
|
||
'addr': addr, # (mandatory) proxy IP address
|
||
'port': int(port), # (mandatory) proxy port number
|
||
'username': username, # (optional) username if the proxy requires auth
|
||
'password': password, # (optional) password if the proxy requires auth
|
||
'rdns': True # (optional) whether to use remote or local resolve, default remote
|
||
}
|
||
client = TelegramClient(md5Token, api_id, api_hash, proxy=proxy)
|
||
else:
|
||
client = TelegramClient(md5Token, api_id, api_hash)
|
||
|
||
|
||
# 配置处理结束
|
||
|
||
|
||
# 展示登陆的信息
|
||
def show_my_inf(me):
|
||
print("-----****************-----")
|
||
print("Name:", me.username)
|
||
print("ID:", me.id)
|
||
print("-----login successful-----")
|
||
|
||
|
||
async def client_main():
|
||
print("-client-main-")
|
||
me = await client.get_me()
|
||
show_my_inf(me)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 除了刷新缓存，都需要频道ID
|
||
if not args.refresh and args.id is None:
|
||
sys.exit(1)
|
||
if args.upload and args.path is None:
|
||
print('缺失上传路径')
|
||
sys.exit(1)
|
||
with client.start(phone=phone, bot_token=bot_token):
|
||
client.loop.run_until_complete(client_main())
|
||
client.loop.run_until_complete(Hook(client))
|
||
if args.refresh:
|
||
print_all_channel(client=client)
|
||
if args.download:
|
||
if 't.me' in args.id:
|
||
tmpList = args.id.split('/')
|
||
channel_id = tmpList[-2]
|
||
plus_func = '=' + tmpList[-1]
|
||
else:
|
||
channel_id = args.id
|
||
plus_func = args.range
|
||
for _id in channel_id.split('|'):
|
||
client.loop.run_until_complete(down_group(client, _id, plus_func, args.user))
|
||
elif args.upload:
|
||
del_after_upload = True if args.dau.upper() == 'Y' else False
|
||
client.loop.run_until_complete(upload_file(client, args.id, args.path, del_after_upload, args.addtag))
|
||
elif args.print:
|
||
client.loop.run_until_complete(print_group(client, args.id))
|
||
elif args.monit:
|
||
channel_ids = args.id.split(',')
|
||
client.loop.run_until_complete(StartMonit(client, channel_ids))
|
||
client.run_until_disconnected()
|