Initial YakPanel commit
This commit is contained in:
Niranjan
365
class_v2/panelSearchV2.py
Normal file
365
class_v2/panelSearchV2.py
Normal file
@@ -0,0 +1,365 @@
|
||||
# coding: utf-8
|
||||
# +-------------------------------------------------------------------
|
||||
# | version :1.0
|
||||
# +-------------------------------------------------------------------
|
||||
# | Author: liangkaiqiang <1249648969@yakpanel.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")}
|
||||
return public.return_message(-1, 0, public.lang("Search content cannot be empty"))
|
||||
if 'exts' not in args or not args.exts:
|
||||
return public.return_message(-1, 0, public.lang("The suffix cannot be empty; please enter [ *.* ] to search all files"))
|
||||
# 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 public.return_message(-1, 0, "Search content cannot be empty")
|
||||
# return {'error': public.lang("The directory cannot be empty or /")}
|
||||
if not os.path.isdir(args.path):
|
||||
return public.return_message(-1, 0, public.lang("Directory does not exist"))
|
||||
# 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
|
||||
return public.return_message(0, 0, 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)
|
||||
Reference in New Issue
Block a user