147 lines
5.0 KiB
Python
147 lines
5.0 KiB
Python
# coding: utf-8
|
||
# -------------------------------------------------------------------
|
||
# YakPanel
|
||
# -------------------------------------------------------------------
|
||
# Copyright (c) 2015-2099 YakPanel(https://www.yakpanel.com) All rights reserved.
|
||
# -------------------------------------------------------------------
|
||
# Author: cjxin <cjxin@yakpanel.com>
|
||
# -------------------------------------------------------------------
|
||
#
|
||
# ------------------------------
|
||
|
||
import os, sys, time, json, re
|
||
import traceback
|
||
|
||
if '/www/server/panel/class/' not in sys.path:
|
||
sys.path.insert(0, '/www/server/panel/class/')
|
||
import public, db
|
||
import panelSite
|
||
from datalistModel.base import dataBase
|
||
|
||
|
||
class main(dataBase):
|
||
web_server = None
|
||
site_obj = None
|
||
|
||
def __init__(self):
|
||
self.site_obj = panelSite.panelSite()
|
||
|
||
"""
|
||
@name 获取公共数据后，格式化为网站列表需要的数据
|
||
"""
|
||
|
||
def get_data_list(self, get):
|
||
try:
|
||
get = self._get_site_args(get)
|
||
data_list = get.data_list
|
||
# 获取前端需要的表头
|
||
from config import config
|
||
table_header = config().get_table_header(public.to_dict_obj({'table_name': 'ftpTableColumn'}))
|
||
if table_header['ftpTableColumn'] == '':
|
||
table_header = ['用户名', '密码', '状态', '复制快速连接信息', '根目录', '设置密码有效期', '容量', '备注', '操作']
|
||
else:
|
||
table_header["ftpTableColumn"] = json.loads(table_header["ftpTableColumn"])
|
||
table_header = [i['title'] if 'title' in i else i['label'] for i in table_header["ftpTableColumn"] if
|
||
(i.get('value', False) == True or i.get('isCustom', True) == True) and ('label' in i.keys() or 'title' in i.keys())]
|
||
# 按需加载
|
||
custom_list = {"容量": "quota", '设置密码有效期': 'end_time'}
|
||
custom_conf = {'quota': {}, "end_time": {}}
|
||
paths = []
|
||
ids = []
|
||
default_quota = {
|
||
"used": 0,
|
||
"size": 0,
|
||
"quota_push": {
|
||
"size": 0,
|
||
"used": 0,
|
||
},
|
||
"quota_storage": {
|
||
"size": 0,
|
||
"used": 0,
|
||
}}
|
||
|
||
# 用户权限处理
|
||
user_Data = self.get_user_power()
|
||
if user_Data != 'all':
|
||
data_list = [i for i in data_list if str(i['id']) in user_Data.get('ftps', [])]
|
||
# 用户权限处理结束
|
||
|
||
[paths.append(val['path']) or ids.append(val['id']) for val in data_list]
|
||
for i, j in custom_list.items():
|
||
if i in table_header:
|
||
if j == 'quota':
|
||
custom_conf[j] = self.get_all_quota(paths)
|
||
if j == 'end_time':
|
||
custom_conf[j] = self.get_all_ftp_end_time(ids)
|
||
for i in data_list:
|
||
i['quota'] = custom_conf['quota'].get(i['path'], default_quota)
|
||
i['end_time'] = custom_conf['end_time'].get(i['id'], '0')
|
||
return data_list
|
||
except:
|
||
pass
|
||
|
||
"""
|
||
@name 初始化参数
|
||
"""
|
||
|
||
def _get_site_args(self, get):
|
||
try:
|
||
if not 'type' in get:
|
||
get.type = 0
|
||
get.type = int(get.p)
|
||
except:
|
||
get.type = 0
|
||
|
||
# if not 'project_type' in get:
|
||
# get.project_type = 'PHP'
|
||
# get.search_key = get.project_type.lower()
|
||
get.search_key = get.table.lower()
|
||
if hasattr(get, 'search'):
|
||
public.set_search_history(get.table,get.search_key,get.search)
|
||
return get
|
||
|
||
"""
|
||
@name 追加 or条件
|
||
"""
|
||
|
||
def get_search_where(self, get):
|
||
|
||
where = ''
|
||
params = get.search
|
||
|
||
conditions = ''
|
||
if '_' in get.search:
|
||
conditions = " escape '/'"
|
||
if params:
|
||
where = "name LIKE ? OR ps LIKE ? {}".format(conditions)
|
||
params = ('%' + params + '%', '%' + params + '%')
|
||
return where, params
|
||
|
||
|
||
"""
|
||
@获取ftp查询条件，追加and查询条件
|
||
"""
|
||
|
||
def get_data_where(self, get):
|
||
wheres = []
|
||
get = self._get_site_args(get)
|
||
if "type_id" in get and get.type_id:
|
||
type_id = get.get("type_id")
|
||
wheres.append("type_id = {}".format(type_id))
|
||
return wheres
|
||
|
||
# 获取fto到期时间
|
||
def get_all_ftp_end_time(self, ids):
|
||
config_path = '/www/server/panel/data/ftp_push_config.json'
|
||
if not os.path.exists(config_path):
|
||
return {i: "0" for i in ids}
|
||
config = json.loads(public.readFile(config_path))
|
||
content = {}
|
||
for _, i in config.items():
|
||
if _ == 'channel':
|
||
continue
|
||
for j in i:
|
||
content[j['id']] = j['end_time']
|
||
return {i: content[i] if i in content.keys() else '0' for i in ids}
|
||
|