mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
Jack
a24f2e23c6
1. Added Mail Marketing Automation trigger tasks 2. Added Mail Marketing Groups Import, Export, Merge 3. Added Mail Marketing Subscribers to support paste import 4. Added Mail Marketing Template Import Export Duplicate 5. Added Mail server add timed automatic reply 6. Added HTTPS Protection function for Website (disables automatic HTTP to HTTPS redirection when enabled) 7. Optimize Website Interface Button Integration 8. Optimize the response speed of WP Toolkit interface
356 lines
14 KiB
Python
356 lines
14 KiB
Python
# coding: utf-8
|
|
# +-------------------------------------------------------------------
|
|
# | version :1.0
|
|
# +-------------------------------------------------------------------
|
|
# | Author: 梁凯强 <1249648969@aapanel.com>
|
|
# +-------------------------------------------------------------------
|
|
# | 快速检索
|
|
# +--------------------------------------------------------------------
|
|
import os,public,re
|
|
import zipfile,time,json
|
|
import db
|
|
class panelSearch:
|
|
__backup_path = '/www/server/panel/backup/panel_search/'
|
|
|
|
def __init__(self):
|
|
if not os.path.exists(self.__backup_path):
|
|
os.makedirs(self.__backup_path)
|
|
if not public.M('sqlite_master').where('type=? AND name=?', ('table', 'panel_search_log')).count():
|
|
csql = '''CREATE TABLE `panel_search_log` (`id` INTEGER PRIMARY KEY AUTOINCREMENT, `rtext` TEXT,`exts` TEXT,`path` TEXT,`mode` TEXT,`isword` TEXT,`iscase` TEXT,`noword` TEXT,`backup_path` TEXT,`time` TEXT)'''
|
|
public.M('sqlite_master').execute(csql,())
|
|
|
|
def dtchg(self, x):
|
|
try:
|
|
time_local = time.localtime(float(x))
|
|
dt = time.strftime("%Y-%m-%d %H:%M:%S", time_local)
|
|
return dt
|
|
except:
|
|
return False
|
|
|
|
def insert_settings(self, rtext, exts, path, mode, isword,iscase,noword,backup_path):
|
|
inser_time = self.dtchg(int(time.time()))
|
|
data = {"rtext": rtext, "exts": json.dumps(exts), "path": path, "mode": mode,
|
|
"isword": isword, "iscase": iscase,"noword":noword,"backup_path":backup_path,"time":inser_time}
|
|
return public.M('panel_search_log').insert(data)
|
|
|
|
'''目录下所有的文件'''
|
|
def get_dir(self, path,exts,text,mode=0,isword=0,iscase=0,noword=0,is_backup=0,rtext=False,get=None):
|
|
if rtext or noword:
|
|
result=[]
|
|
else:
|
|
result={}
|
|
if is_backup:
|
|
t = time.strftime('%Y%m%d%H%M%S')
|
|
back_zip = os.path.join(self.__backup_path, "%s.zip" % t)
|
|
zfile = zipfile.ZipFile(back_zip, "w", compression=zipfile.ZIP_DEFLATED)
|
|
else:
|
|
zfile=False
|
|
back_zip=False
|
|
return_data = []
|
|
[[return_data.append(os.path.join(root, file)) for file in files]
|
|
for root, dirs, files in os.walk(path)]
|
|
num = 0
|
|
total_num = len(return_data)
|
|
if return_data: public.writeSpeed('files_search', num, total_num)
|
|
for i in return_data:
|
|
is_send = False
|
|
for i2 in exts:
|
|
i3 = i.split('.')
|
|
if i3[-1] == i2:
|
|
is_send = True
|
|
temp = self.get_files_lin(i, text, mode, isword, iscase, noword,is_backup,rtext,zfile)
|
|
if temp:
|
|
if rtext or noword:
|
|
if isinstance(result,list):
|
|
result.append(temp)
|
|
else:
|
|
result[i] = temp
|
|
num += 1
|
|
public.writeSpeed('files_search', num, total_num)
|
|
progress = int(public.getSpeed()['progress'])
|
|
if '_ws' in get:
|
|
get._ws.send(
|
|
public.getJson({
|
|
"end": False if progress < 100 else True,
|
|
"ws_callback": get.ws_callback,
|
|
"file": i if is_send else '',
|
|
"progress": progress,
|
|
"total": total_num,
|
|
"num": num,
|
|
"type": "files_search"
|
|
}))
|
|
if is_backup:
|
|
if zfile:
|
|
zfile.close()
|
|
self.insert_settings(rtext, exts, path, mode, isword, iscase, noword, back_zip)
|
|
return True
|
|
return result
|
|
|
|
'''获取单目录'''
|
|
|
|
def get_dir_files(self,
|
|
path,
|
|
exts,
|
|
text,
|
|
mode=0,
|
|
isword=0,
|
|
iscase=0,
|
|
noword=0,
|
|
is_backup=0,
|
|
rtext=False,
|
|
get=None):
|
|
is_list = rtext or noword
|
|
if is_list:
|
|
result=[]
|
|
else:
|
|
result={}
|
|
if is_backup:
|
|
t = time.strftime('%Y%m%d%H%M%S')
|
|
back_zip = os.path.join(self.__backup_path, "%s.zip" % t)
|
|
zfile = zipfile.ZipFile(back_zip, "w", compression=zipfile.ZIP_DEFLATED)
|
|
else:
|
|
zfile=False
|
|
back_zip=False
|
|
list_data=[]
|
|
for root, dirs, files in os.walk(path):
|
|
list_data=files
|
|
break
|
|
num = 0
|
|
total_num = len(list_data)
|
|
if list_data: public.writeSpeed('files_search', num, total_num)
|
|
for i2 in list_data:
|
|
is_send = False
|
|
for i in exts:
|
|
i3 = i2.split('.')
|
|
if i3[-1] == i:
|
|
is_send = True
|
|
temp = self.get_files_lin(path + '/' + i2, text, mode,
|
|
isword, iscase, noword,
|
|
is_backup, rtext, zfile)
|
|
if temp:
|
|
if isinstance(result,list):
|
|
result.append(path+'/'+i2)
|
|
else:
|
|
result[path + '/' + i2] = temp
|
|
num += 1
|
|
public.writeSpeed('files_search', num, total_num)
|
|
progress = int(public.getSpeed()['progress'])
|
|
if '_ws' in get:
|
|
get._ws.send(
|
|
public.getJson({
|
|
"end": False if progress < 100 else True,
|
|
"ws_callback": get.ws_callback,
|
|
"file": i2 if is_send else '',
|
|
"progress": progress,
|
|
"total": total_num,
|
|
"num": num,
|
|
"type": "files_search"
|
|
}))
|
|
if is_backup:
|
|
if zfile:
|
|
zfile.close()
|
|
self.insert_settings(rtext, exts, path, mode, isword, iscase, noword, back_zip)
|
|
return result
|
|
'''
|
|
获取目录下所有的后缀文件
|
|
'''
|
|
|
|
def get_exts_files(self,
|
|
path,
|
|
exts,
|
|
text,
|
|
mode=0,
|
|
isword=0,
|
|
iscase=0,
|
|
noword=0,
|
|
is_subdir=0,
|
|
is_backup=0,
|
|
rtext=False,
|
|
get=None):
|
|
if len(exts)==0:return []
|
|
if is_subdir==0:
|
|
return self.get_dir_files(path,
|
|
exts,
|
|
text,
|
|
mode,
|
|
isword,
|
|
iscase,
|
|
noword,
|
|
is_backup,
|
|
rtext,
|
|
get=get)
|
|
elif is_subdir==1:
|
|
return self.get_dir(path,
|
|
exts,
|
|
text,
|
|
mode,
|
|
isword,
|
|
iscase,
|
|
noword,
|
|
is_backup,
|
|
rtext,
|
|
get=get)
|
|
|
|
'''
|
|
获取文件内的关键词
|
|
'''
|
|
def get_files_lin(self, files, text,mode=0,isword=0,iscase=0,noword=0,is_backup=0,rtext=False,back_zip=False):
|
|
if not os.path.exists(files):return False
|
|
if os.path.getsize(files) > 1024 * 1024 * 20: return False
|
|
#文件替换部分
|
|
if rtext:
|
|
resutl=[]
|
|
try:
|
|
fp = open(files, 'r', encoding='UTF-8')
|
|
except:
|
|
fp = open(files, 'r')
|
|
content = fp.read()
|
|
fp.close()
|
|
if mode==2:
|
|
if iscase:
|
|
if not re.search(text, content, flags=re.I): return False
|
|
content = re.sub(text, rtext, content, flags=re.I)
|
|
else:
|
|
if not re.search(text, content): return False
|
|
content = re.sub(text, rtext, content)
|
|
else:
|
|
if iscase:
|
|
if not re.search(text, content, flags=re.I): return False
|
|
content = re.sub(text, rtext, content, flags=re.I)
|
|
else:
|
|
if content.find(text) == -1: return False
|
|
content = content.replace(text, rtext)
|
|
if is_backup and back_zip:
|
|
bf = files.strip('/')
|
|
back_zip.write(files, bf)
|
|
with open(files, 'w') as f:
|
|
f.write(content)
|
|
f.close()
|
|
return files
|
|
else:
|
|
#查找部分
|
|
if noword:
|
|
resutl = []
|
|
else:
|
|
resutl={}
|
|
try:
|
|
fp = open(files, 'r', encoding='UTF-8')
|
|
except:
|
|
fp = open(files, 'r')
|
|
i = 0
|
|
try:
|
|
for line in fp:
|
|
i += 1
|
|
if mode==1:
|
|
if iscase and not re.search(text, line, flags=re.I):
|
|
continue
|
|
elif not iscase and not re.search(text, line):
|
|
continue
|
|
else:
|
|
if iscase:
|
|
if not re.search(text, line, flags=re.I): continue
|
|
else:
|
|
if line.find(text) == -1: continue
|
|
if noword:
|
|
return files
|
|
resutl[i]=line
|
|
except:
|
|
pass
|
|
if resutl:
|
|
return resutl
|
|
return False
|
|
'''
|
|
text 搜索内容
|
|
exts 后缀名 参数例子 php,html
|
|
path 目录
|
|
is_subdir 0 不包含子目录 1 包含子目录
|
|
mode 0 为普通模式 1 为正则模式
|
|
isword 1 全词匹配 0 默认
|
|
iscase 1 不区分大小写 0 默认
|
|
noword 1 不输出行信息 0 默认
|
|
'''
|
|
def get_search(self, args):
|
|
if 'text' not in args or not args.text: return {'error': public.lang("Search content cannot be empty")}
|
|
if 'exts' not in args or not args.exts: return {'error': public.lang("The suffix cannot be empty; please enter [ *.* ] to search all files")}
|
|
if 'path' not in args or not args.path or args.path == '/': return {'error': public.lang("The directory cannot be empty or /")}
|
|
if not os.path.isdir(args.path): return {'error': public.lang("Directory does not exist")}
|
|
text=args.text
|
|
exts=args.exts
|
|
path=args.path
|
|
mode = int(args.mode) if 'mode' in args else 0
|
|
is_subdir = int(args.is_subdir) if 'is_subdir' in args else 0
|
|
iscase = int(args.iscase) if 'iscase' in args else 0
|
|
isword = int(args.isword) if 'isword' in args else 0
|
|
noword = int(args.noword) if 'noword' in args else 0
|
|
exts=exts.split(',')
|
|
is_tmpe_files = self.get_exts_files(path,
|
|
exts,
|
|
text,
|
|
mode,
|
|
isword,
|
|
iscase,
|
|
noword,
|
|
is_subdir,
|
|
get=args)
|
|
return is_tmpe_files
|
|
|
|
'''
|
|
text 搜索内容
|
|
rtext 替换成的内容
|
|
exts 后缀名 参数例子 php,html
|
|
path 目录
|
|
is_subdir 0 不包含子目录 1 包含子目录
|
|
mode 0 为普通模式 1 为正则模式
|
|
isword 1 全词匹配 0 默认
|
|
iscase 1 不区分大小写 0 默认
|
|
noword 1 不输出行信息 0 默认
|
|
'''
|
|
def get_replace(self, args):
|
|
if 'text' not in args or not args.text: return {'error': public.lang("Search content cannot be empty")}
|
|
if 'rtext' not in args or not args.text: return {'error': public.lang("The content to be replaced cannot be empty")}
|
|
if 'exts' not in args or not args.exts: return {'error': public.lang("The suffix cannot be empty; please enter [ *.* ] to search all files")}
|
|
if 'path' not in args or not args.path or args.path == '/': return {'error': public.lang("The directory cannot be empty or /")}
|
|
if not os.path.isdir(args.path): return {'error': public.lang("Directory does not exist")}
|
|
is_backup = int(args.isbackup) if 'isbackup' in args else 0
|
|
text = args.text
|
|
rtext = args.rtext
|
|
exts = args.exts
|
|
path = args.path
|
|
mode = int(args.mode) if 'mode' in args else 0
|
|
is_subdir = int(args.is_subdir) if 'is_subdir' in args else 0
|
|
iscase = int(args.iscase) if 'iscase' in args else 0
|
|
isword = int(args.isword) if 'isword' in args else 0
|
|
noword = int(args.noword) if 'noword' in args else 0
|
|
exts = exts.split(',')
|
|
is_tmpe_files = self.get_exts_files(path, exts, text, mode, isword, iscase, noword, is_subdir,is_backup,rtext)
|
|
return is_tmpe_files
|
|
|
|
#替換日志
|
|
def get_replace_logs(self,get):
|
|
import page
|
|
page = page.Page()
|
|
count = public.M('panel_search_log').order('id desc').count()
|
|
limit = 12
|
|
info = {}
|
|
info['count'] = count
|
|
info['row'] = limit
|
|
info['p'] = 1
|
|
if hasattr(get, 'p'):
|
|
info['p'] = int(get['p'])
|
|
info['uri'] = get
|
|
info['return_js'] = ''
|
|
if hasattr(get, 'tojs'):
|
|
info['return_js'] = get.tojs
|
|
data = {}
|
|
data['page'] = page.GetPage(info, '1,2,3,4,5,8')
|
|
data['data'] = public.M('panel_search_log').field('id,rtext,exts,path,mode,isword,iscase,noword,backup_path,time').order('id desc').limit(str(page.SHIFT) + ',' + str(page.ROW)).select()
|
|
if isinstance(data['data'],str): return public.returnMsg(False,[])
|
|
for i in data['data']:
|
|
if not isinstance(i,dict): continue
|
|
if 'backup_path' in i :
|
|
path=i['backup_path']
|
|
if os.path.exists(path):
|
|
i['is_path_status']=True
|
|
else:
|
|
i['is_path_status'] = False
|
|
return public.returnMsg(True, data) |