123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210 |
- #!/usr/bin/env python3
- # -*- coding: utf-8 -*-
- # File : rules.py.py
- # Author: DaShenHan&道长-----先苦后甜,任凭晚风拂柳颜------
- # Date : 2022/8/25
- import json
- import os
- from time import time
- import js2py
- from utils.log import logger
- # from utils.web import get_interval,UA
- from utils.ua import UA,get_interval
- from flask import render_template_string
- def getRuleLists():
- base_path = os.path.dirname(os.path.abspath(__file__)) # 当前文件所在目录
- # print(base_path)
- file_name = os.listdir(base_path)
- file_name = list(filter(lambda x:str(x).endswith('.js') and str(x).find('模板') < 0,file_name))
- # print(file_name)
- rule_list = [file.replace('.js','') for file in file_name]
- # print(rule_list)
- return rule_list
- def getCacheCount():
- base_path = 'cache/' # 当前缓存js所在目录
- os.makedirs(base_path, exist_ok=True)
- file_name = os.listdir(base_path)
- file_name = list(filter(lambda x: str(x).endswith('.js') and str(x).find('模板') < 0, file_name))
- return len(file_name)
- def getRules(path='cache',js_mode=0):
- t1 = time()
- base_path = path+'/' # 当前文件所在目录
- # print(base_path)
- os.makedirs(base_path,exist_ok=True)
- file_name = os.listdir(base_path)
- file_name = list(filter(lambda x: str(x).endswith('.js') and str(x).find('模板') < 0, file_name))
- # print(file_name)
- rule_list = [file.replace('.js', '') for file in file_name]
- js_path = [f'{path}/{rule}.js' for rule in rule_list]
- with open('js/模板.js', encoding='utf-8') as f:
- before = f.read().split('export')[0]
- rule_codes = []
- # for js in js_path:
- # with open(js,encoding='utf-8') as f:
- # code = f.read()
- # rule_codes.append(js2py.eval_js(before+code))
- ctx = js2py.EvalJs()
- codes = []
- for i in range(len(js_path)):
- js = js_path[i]
- with open(js,encoding='utf-8') as f:
- code = f.read()
- new_code = 'var muban = JSON.parse(JSON.stringify(mubanDict));\n'+code.replace('rule',f'rule{i}',1)
- # new_code = ''+code.replace('rule',f'rule{i}',1)
- codes.append(new_code)
- newCodes = before + '\n'+ '\n'.join(codes)
- # print(newCodes)
- try:
- ctx.execute(newCodes)
- for i in range(len(js_path)):
- rule_codes.append(ctx.eval(f'rule{i}'))
- # print(rule_codes)
- # print(type(rule_codes[0]),rule_codes[0])
- # print(rule_codes[0].title)
- # print(rule_codes[0].searchable)
- # print(rule_codes[0].quickSearch)
- new_rule_list = []
- for i in range(len(rule_list)):
- if js_mode == 1 and rule_list[i] == 'drpy':
- continue
- sable = rule_codes[i].searchable or 0
- tmpObj = {
- 'name':rule_list[i],
- # 'searchable':1 if (js_mode==1 and sable==2) else sable, # 对js模式1开放软件聚搜(还是算了,服务器遭不住)
- 'searchable':sable,
- 'quickSearch':rule_codes[i].quickSearch or 0,
- 'filterable':rule_codes[i].filterable or 0,
- }
- if rule_codes[i].multi:
- tmpObj['multi'] = 1
- new_rule_list.append(tmpObj)
- # print(new_rule_list)
- rules = {'list': new_rule_list, 'count': len(rule_list)}
- except Exception as e:
- logger.info(f'装载js内置源列表失败,置空内置源')
- rules = []
- logger.info(f'自动配置装载耗时:{get_interval(t1)}毫秒')
- return rules
- def jxTxt2Json(text:str,host:str):
- try:
- data = render_template_string(text,host=host).strip().split('\n')
- except Exception as e:
- logger.info(f'jxTxt2Json发生错误:{e}')
- data = []
- jxs = []
- for i in data:
- i = i.strip()
- dt = i.split(',')
- if not i.startswith('#') and len(i) > 10:
- try:
- jxs.append({
- 'name':dt[0],
- 'url':dt[1],
- 'type':dt[2] if len(dt) > 2 and dt[2] else 0,
- 'ua':dt[3] if len(dt) > 3 and dt[3] else UA,
- })
- except Exception as e:
- logger.info(f'解析行有错误:{e}')
- return jxs
- def getJxs(path='js',host=None):
- custom_jx = 'base/解析.conf'
- if not os.path.exists(custom_jx):
- with open(custom_jx,'w+',encoding='utf-8') as f1:
- msg = """# 这是用户自定义解析列表,不会被系统升级覆盖
- # 0123,对应,普通解析,json解析,并发多json解析,聚合解析,参数3不填默认0
- # flags是线路名称标识,会自动拦截并走以下的解析
- # 名称,链接,类型,ua (ua不填默认 Mozilla/5.0) 可以手动填 Dart/2.14 (dart:io)
- 虾米,https://dm.xmflv.com:4433/?url=
- """
- f1.write(msg)
- base_path = 'jiexi' # 自建解析目录
- os.makedirs(base_path, exist_ok=True)
- file_name = os.listdir(base_path)
- file_name = list(filter(lambda x: str(x).endswith('.js') and str(x).find('模板') < 0, file_name))
- # print(file_name)
- jx_list = [file.replace('.js', '') for file in file_name]
- # print(file_name)
- # print(jx_list)
- jx_str = '\n'.join([jx+',{{host}}'+f'/parse/api/{jx}.js?url=,1' for jx in jx_list])
- # print(jx_str)
- with open(f'{path}/解析.conf',encoding='utf-8') as f:
- text = f.read()
- text = jx_str + '\n' + text
- jxs = jxTxt2Json(text,host)
- with open(custom_jx,encoding='utf-8') as f2:
- text = f2.read()
- jxs2 = jxTxt2Json(text,host)
- jxs.extend(jxs2)
- print(f'共计{len(jxs)}条解析')
- return jxs
- def getPys(path='txt/py'):
- t1 = time()
- base_path = os.path.dirname(os.path.abspath(os.path.dirname(__file__))) # 上级目录
- py_path = os.path.join(base_path, path)
- os.makedirs(py_path, exist_ok=True)
- file_name = os.listdir(py_path)
- file_name = list(filter(lambda x: str(x).endswith('.py'), file_name))
- # print(file_name)
- rule_list = [file.replace('.py', '') for file in file_name]
- py_path = [f'{path}/{rule}.py' for rule in rule_list]
- new_rule_list = []
- for i in range(len(rule_list)):
- new_rule_list.append({
- 'name': rule_list[i],
- 'searchable': 1,
- 'quickSearch': 1,
- 'filterable': 0,
- })
- logger.info(f'自动加载Pyramid耗时:{get_interval(t1)}毫秒')
- return new_rule_list
- def gen_cache(path='txt/js/tg'):
- t1 = time()
- base_path = os.path.dirname(os.path.abspath(os.path.dirname(__file__))) # 上级目录
- py_path = os.path.join(base_path, path)
- os.makedirs(py_path, exist_ok=True)
- file_name = os.listdir(py_path)
- file_name = list(filter(lambda x: str(x).endswith('.js'), file_name))
- # print(file_name)
- rule_list = [file.replace('.js', '') for file in file_name]
- js_path = [f'{path}/{rule}.js' for rule in rule_list]
- new_rule_list = []
- for i in range(len(rule_list)):
- # print(js_path[i])
- rname = rule_list[i]
- new_rule_list.append(
- {
- "key": f"dr_{rname}",
- "name": f"{rname}(道长)",
- "type": 1,
- # "api": "{{host}}"+f"/vod?rule={rname}&ext="+"{{host}}/"+js_path[i],
- "api": "{{host}}"+f"/vod?rule={rname}&ext="+js_path[i],
- "searchable": 2,
- "quickSearch": 0,
- "filterable": 0
- })
- logger.info(f'自动加载{len(new_rule_list)}个缓存JS耗时:{get_interval(t1)}毫秒')
- new_rules_texts = [json.dumps(new_rule,ensure_ascii=False) for new_rule in new_rule_list]
- # new_rules_text = json.dumps(new_rule_list,ensure_ascii=False)
- new_rules_text = ',\n'.join(new_rules_texts)+','
- return new_rules_text
- if __name__ == '__main__':
- print(getRuleLists())
- # print(gen_cache())
- # print(gen_cache('txt/js/18'))
- custom_file = gen_cache() + '\n'+gen_cache('txt/js/18')
- print(custom_file)
- with open('custom.conf','w+',encoding='utf-8') as f:
- f.write(custom_file)
|