524 lines
20 KiB
Python
524 lines
20 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, re, json, shutil, psutil, time
|
|||
|
|
import uuid
|
|||
|
|
from mod.project.push import taskMod
|
|||
|
|
|
|||
|
|
from sslModel.base import sslBase
|
|||
|
|
import public
|
|||
|
|
|
|||
|
|
|
|||
|
|
class main(sslBase):
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__()
|
|||
|
|
# self.__init_data()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def run_fun(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 执行指定函数
|
|||
|
|
@param get.def_name 函数名
|
|||
|
|
@param get.dns_type dns类型
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
if 'fun_name' not in get:
|
|||
|
|
return public.returnMsg(False, public.lang('undefined fun_name'))
|
|||
|
|
|
|||
|
|
if not get.fun_name in ['delete_dns_record','create_dns_record','get_dns_record','update_dns_record','set_dns_record_status', 'get_domain_list']:
|
|||
|
|
return public.returnMsg(False,public.lang('wrong params，undefined func name'))
|
|||
|
|
|
|||
|
|
dns_type = False
|
|||
|
|
try:
|
|||
|
|
dns_data = self.get_dns_data(get)
|
|||
|
|
dns_type = dns_data[get.dns_id]['dns_type']
|
|||
|
|
except: pass
|
|||
|
|
if not dns_type:
|
|||
|
|
return public.returnMsg(False,public.lang('dns_id unknown'))
|
|||
|
|
get.dns_type = dns_type
|
|||
|
|
try:
|
|||
|
|
res = self.func_models(get, get.fun_name)
|
|||
|
|
return res
|
|||
|
|
except Exception as e:
|
|||
|
|
return public.returnMsg(False,public.lang('Execution failed:{}'.format(str(e))))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_domain_list(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 获取域名列表
|
|||
|
|
"""
|
|||
|
|
self.__init_data()
|
|||
|
|
|
|||
|
|
filter_sql = ''
|
|||
|
|
if get.get('search'):
|
|||
|
|
filter_sql += ' and domain like {}'.format("'%"+get.search+"%'")
|
|||
|
|
public.set_search_history('ssl', 'get_domain_list', get.search)
|
|||
|
|
if get.get('type_id'):
|
|||
|
|
filter_sql += " and type_id = '{}'".format(get.type_id)
|
|||
|
|
|
|||
|
|
p = 1
|
|||
|
|
if 'p' in get:
|
|||
|
|
p = int(get.p)
|
|||
|
|
collback = ''
|
|||
|
|
if 'collback' in get:
|
|||
|
|
collback = get.collback
|
|||
|
|
limit = 20
|
|||
|
|
if 'limit' in get:
|
|||
|
|
limit = int(get.limit)
|
|||
|
|
|
|||
|
|
count = public.M('ssl_domains').where('1=1 {}'.format(filter_sql), ()).count()
|
|||
|
|
page_data = public.get_page(count, p, limit, collback)
|
|||
|
|
|
|||
|
|
dns_data = self.get_dns_data(get)
|
|||
|
|
data = public.M('ssl_domains').field('id,domain,type_id,dns_id,endtime,ps').where('1=1 {}'.format(filter_sql), ()).limit(page_data['shift'] + ',' + page_data['row']).select()
|
|||
|
|
|
|||
|
|
record_data = self.get_record_data()
|
|||
|
|
|
|||
|
|
report_data = taskMod.main().get_task_list().get('data')
|
|||
|
|
report_data_dic = {}
|
|||
|
|
if report_data:
|
|||
|
|
report_data_dic = {i["keyword"]: i["id"] for i in report_data if i["source"] == "domain_endtime" and i["status"]}
|
|||
|
|
|
|||
|
|
domain_list = []
|
|||
|
|
_data = []
|
|||
|
|
remove_ids = []
|
|||
|
|
for val in data:
|
|||
|
|
if val['domain'] in domain_list:
|
|||
|
|
remove_ids.append(str(val['id']))
|
|||
|
|
continue
|
|||
|
|
domain_list.append(val['domain'])
|
|||
|
|
val["report_id"] = report_data_dic.get(val['domain'], "") or ""
|
|||
|
|
|
|||
|
|
if val['dns_id'] == '0': val['dns_id'] = ''
|
|||
|
|
|
|||
|
|
val['dns'] = dns_data.get(val['dns_id'],{})
|
|||
|
|
# 获取子域名数量
|
|||
|
|
val['sbd_count'] = record_data.get(val["domain"], {}).get('info', {}).get('record_total', '-')
|
|||
|
|
_data.append(val)
|
|||
|
|
try:
|
|||
|
|
public.M('ssl_domains').where("id in ({})".format(",".join(remove_ids)), ()).delete()
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e)
|
|||
|
|
|
|||
|
|
search_history = public.get_search_history('ssl', 'get_domain_list')
|
|||
|
|
page_data.update({'data': _data, 'search_history': search_history})
|
|||
|
|
public.set_module_logs('ssl', 'get_domain_list', 1)
|
|||
|
|
return page_data
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_domain_ps(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 设置域名备注
|
|||
|
|
@param get.id 域名id
|
|||
|
|
@param get.ps 备注
|
|||
|
|
"""
|
|||
|
|
id = get.id
|
|||
|
|
ps = get.ps
|
|||
|
|
res = public.M('ssl_domains').where('id=?',(id,)).setField('ps',ps)
|
|||
|
|
if not res:
|
|||
|
|
return public.returnMsg(False,public.lang('Setting fail!'))
|
|||
|
|
return public.returnMsg(True,public.lang('setting succeed!'))
|
|||
|
|
|
|||
|
|
def set_domain_dns(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 给域名设置dns
|
|||
|
|
@param get.ids 域名id [1,2,3]
|
|||
|
|
@param get.dns_id dns_id
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
ids = json.loads(get.ids)
|
|||
|
|
except: pass
|
|||
|
|
|
|||
|
|
if not ids:
|
|||
|
|
return public.returnMsg(False,public.lang('Please select the domain name you want to set!'))
|
|||
|
|
|
|||
|
|
dns_id = get.dns_id
|
|||
|
|
dns_data = self.get_dns_data(get)
|
|||
|
|
if dns_id not in dns_data:
|
|||
|
|
return public.returnMsg(False,public.lang('The specified DNS does not exist!'))
|
|||
|
|
|
|||
|
|
params = []
|
|||
|
|
for id in ids:
|
|||
|
|
params.append((dns_id,id))
|
|||
|
|
|
|||
|
|
res = public.M('ssl_domains').executemany('update ssl_domains set dns_id=? where id=?',params)
|
|||
|
|
if type(res) != int:
|
|||
|
|
return public.returnMsg(False,public.lang('setting fail!'))
|
|||
|
|
|
|||
|
|
return public.returnMsg(True,public.lang('The setting was successful, and {} records have been updated!'.format(res)))
|
|||
|
|
|
|||
|
|
def set_domain_endtime(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 设置域名到期时间
|
|||
|
|
@param get.ids 域名id [1,2,3]
|
|||
|
|
@param get.endtime 到期时间
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
ids = json.loads(get.ids)
|
|||
|
|
except: pass
|
|||
|
|
|
|||
|
|
if not ids:
|
|||
|
|
return public.returnMsg(False,public.lang('Please select the domain name you want to set!'))
|
|||
|
|
|
|||
|
|
endtime = get.endtime
|
|||
|
|
params = []
|
|||
|
|
for id in ids:
|
|||
|
|
params.append((endtime,id))
|
|||
|
|
|
|||
|
|
res = public.M('ssl_domains').executemany('update ssl_domains set endtime=? where id=?',params)
|
|||
|
|
if type(res) != int:
|
|||
|
|
return public.returnMsg(False,public.lang('setting fail!'))
|
|||
|
|
|
|||
|
|
return public.returnMsg(True,public.lang('The setting was successful, and {} records have been updated!'.format(res)))
|
|||
|
|
|
|||
|
|
def set_domain_type(self,get):
|
|||
|
|
"""
|
|||
|
|
@设置域名到期时间
|
|||
|
|
@param get.ids 域名id [1,2,3]
|
|||
|
|
@param get.type_id 类型id
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
ids = json.loads(get.ids)
|
|||
|
|
except: pass
|
|||
|
|
|
|||
|
|
if not ids:
|
|||
|
|
return public.returnMsg(False,public.lang('Please select the domain name you want to set!'))
|
|||
|
|
|
|||
|
|
type_id = get.type_id
|
|||
|
|
params = []
|
|||
|
|
for id in ids:
|
|||
|
|
params.append((type_id,id))
|
|||
|
|
|
|||
|
|
res = public.M('ssl_domains').executemany('update ssl_domains set type_id=? where id=?',params)
|
|||
|
|
if type(res) != int:
|
|||
|
|
return public.returnMsg(False,'setting fail!')
|
|||
|
|
|
|||
|
|
return public.returnMsg(True,public.lang('The setting was successful, and {} records have been updated!'.format(res)))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_domain_type(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 获取域名类型
|
|||
|
|
"""
|
|||
|
|
data = [{'type_id': 0, 'name': 'default'}]
|
|||
|
|
try:
|
|||
|
|
sfile = '{}/data/domains_type.json'.format(public.get_panel_path())
|
|||
|
|
data.extend(json.loads(public.readFile(sfile)))
|
|||
|
|
except:pass
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def add_domain_type(self,get):
|
|||
|
|
"""
|
|||
|
|
@name 添加域名分类
|
|||
|
|
@param get.name 分类名称
|
|||
|
|
@param get.type_id 分类类型
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
sfile = '{}/data/domains_type.json'.format(public.get_panel_path())
|
|||
|
|
try:
|
|||
|
|
data = json.loads(public.readFile(sfile))
|
|||
|
|
except:
|
|||
|
|
data = []
|
|||
|
|
type_id = str(uuid.uuid4().hex)
|
|||
|
|
if 'type_id' in get:
|
|||
|
|
type_id = get.type_id
|
|||
|
|
for i in data:
|
|||
|
|
if get.name == i["name"] or get.name == 'default':
|
|||
|
|
return public.returnMsg(False, public.lang('This type already exists.'))
|
|||
|
|
|
|||
|
|
data.append({'name':get.name,'type_id':type_id})
|
|||
|
|
public.writeFile(sfile,json.dumps(data))
|
|||
|
|
return public.returnMsg(True,public.lang('add successfully!'))
|
|||
|
|
|
|||
|
|
def del_domain_type(self, get):
|
|||
|
|
sfile = '{}/data/domains_type.json'.format(public.get_panel_path())
|
|||
|
|
try:
|
|||
|
|
data = json.loads(public.readFile(sfile))
|
|||
|
|
i = 0
|
|||
|
|
while i < len(data):
|
|||
|
|
if data[i]["type_id"] == get.type_id:
|
|||
|
|
del data[i]
|
|||
|
|
break
|
|||
|
|
i += 1
|
|||
|
|
public.writeFile(sfile, json.dumps(data))
|
|||
|
|
return public.returnMsg(True, public.lang('del successful'))
|
|||
|
|
except:
|
|||
|
|
return public.returnMsg(True, public.lang('del fail'))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __init_data(self):
|
|||
|
|
"""
|
|||
|
|
@name 同步数据
|
|||
|
|
"""
|
|||
|
|
# 检查表是否存在
|
|||
|
|
self.__create_table()
|
|||
|
|
|
|||
|
|
# 根据文件修改时间判断是否要同步
|
|||
|
|
check_path = "/www/server/panel/data/sync_domains.pl"
|
|||
|
|
m_time = str(os.stat("/www/server/panel/data/db/site.db").st_mtime)
|
|||
|
|
check_path_docker = "/www/server/panel/data/sync_domains_docker.pl"
|
|||
|
|
m_time_docker = str(os.stat("/www/server/panel/data/db/docker.db").st_mtime)
|
|||
|
|
if os.path.exists(check_path) and os.path.exists(check_path_docker):
|
|||
|
|
if m_time == public.readFile(check_path) and m_time_docker == public.readFile(check_path_docker):
|
|||
|
|
return
|
|||
|
|
public.writeFile(check_path, m_time)
|
|||
|
|
public.writeFile(check_path_docker, m_time)
|
|||
|
|
|
|||
|
|
# 获取手动删除的域名
|
|||
|
|
del_path = '{}/config/del_domains.pl'.format(public.get_panel_path())
|
|||
|
|
try:
|
|||
|
|
skip_domains = json.loads(public.readFile(del_path))
|
|||
|
|
except:
|
|||
|
|
skip_domains = []
|
|||
|
|
|
|||
|
|
root_domains = {i['domain'] for i in public.M('ssl_domains').field('domain').select()}
|
|||
|
|
site_domains = {self.extract_zone(i['name'])[0] for i in public.M('domain').field('name').select()}
|
|||
|
|
docker_site_domains = {self.extract_zone(i['name'])[0] for i in public.M('docker_domain').field('name').select()}
|
|||
|
|
add_domains = (site_domains | docker_site_domains) - root_domains - set(skip_domains)
|
|||
|
|
# del_domains = root_domains - (docker_site_domains | site_domains)
|
|||
|
|
|
|||
|
|
# 添加域名
|
|||
|
|
for domain in add_domains:
|
|||
|
|
public.M('ssl_domains').add('domain,dns_id,type_id,endtime,ps', (domain, 0, 0, 0, ''))
|
|||
|
|
|
|||
|
|
# 删除域名
|
|||
|
|
# for domain in del_domains:
|
|||
|
|
# public.M('ssl_domains').where("domain=?", (domain,)).delete()
|
|||
|
|
|
|||
|
|
def del_domains(self, get):
|
|||
|
|
"""
|
|||
|
|
@name 删除域名
|
|||
|
|
@param get.domains 域名列表，以逗号分隔
|
|||
|
|
"""
|
|||
|
|
domains = get.domains.split(',')
|
|||
|
|
try:
|
|||
|
|
public.M('ssl_domains').where("domain in ('{}')".format("','".join(domains)), ()).delete()
|
|||
|
|
path = '{}/config/del_domains.pl'.format(public.get_panel_path())
|
|||
|
|
try:
|
|||
|
|
del_domains = json.loads(public.readFile(path))
|
|||
|
|
del_domains.extend(domains)
|
|||
|
|
public.writeFile(path, json.dumps(del_domains))
|
|||
|
|
except:
|
|||
|
|
public.writeFile(path, json.dumps(domains))
|
|||
|
|
return public.returnMsg(True, public.lang('del successfully'))
|
|||
|
|
except:
|
|||
|
|
return public.returnMsg(False, public.lang('del fail'))
|
|||
|
|
|
|||
|
|
# 手动同步域名
|
|||
|
|
def sync_domains(self, get):
|
|||
|
|
"""
|
|||
|
|
@name 手动同步域名
|
|||
|
|
"""
|
|||
|
|
root_domains = {i['domain'] for i in public.M('ssl_domains').field('domain').select()}
|
|||
|
|
site_domains = {self.extract_zone(i['name'])[0] for i in public.M('domain').field('name').select()}
|
|||
|
|
docker_site_domains = {self.extract_zone(i['name'])[0] for i in public.M('docker_domain').field('name').select()}
|
|||
|
|
add_domains = (site_domains | docker_site_domains) - root_domains
|
|||
|
|
del_domains = root_domains - (docker_site_domains | site_domains)
|
|||
|
|
|
|||
|
|
# 添加域名
|
|||
|
|
for domain in add_domains:
|
|||
|
|
public.M('ssl_domains').add('domain,dns_id,type_id,endtime,ps', (domain, 0, 0, 0, ''))
|
|||
|
|
# 删除域名
|
|||
|
|
for domain in del_domains:
|
|||
|
|
public.M('ssl_domains').where("domain=?", (domain,)).delete()
|
|||
|
|
# 删除文件
|
|||
|
|
try:
|
|||
|
|
os.remove('{}/config/del_domains.pl'.format(public.get_panel_path()))
|
|||
|
|
except: pass
|
|||
|
|
return public.returnMsg(True, public.lang('Sync successful.'))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __create_table(self):
|
|||
|
|
"""
|
|||
|
|
@name 检查表是否存在
|
|||
|
|
"""
|
|||
|
|
public.check_table('ssl_domains',"""CREATE TABLE IF NOT EXISTS `ssl_domains` (
|
|||
|
|
`id` INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
`domain` TEXT,
|
|||
|
|
`dns_id` TEXT,
|
|||
|
|
`type_id` INTEGER,
|
|||
|
|
`endtime` INTEGER,
|
|||
|
|
`ps` TEXT
|
|||
|
|
)
|
|||
|
|
""")
|
|||
|
|
|
|||
|
|
def get_objectModel(self):
|
|||
|
|
'''
|
|||
|
|
获取模型对象
|
|||
|
|
'''
|
|||
|
|
from panelController import Controller
|
|||
|
|
project_obj = Controller()
|
|||
|
|
|
|||
|
|
return project_obj
|
|||
|
|
|
|||
|
|
|
|||
|
|
def func_models(self,get,def_name):
|
|||
|
|
'''
|
|||
|
|
获取模型对象
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
if 'dns_type' not in get:
|
|||
|
|
raise Exception(public.lang('undefined dns_type'))
|
|||
|
|
|
|||
|
|
sfile = '{}/class/sslModel/{}Model.py'.format(public.get_panel_path(),get.dns_type)
|
|||
|
|
if not os.path.exists(sfile):
|
|||
|
|
raise Exception(public.lang('The module file {} does not exist.'.format(sfile)))
|
|||
|
|
obj_main = self.get_objectModel()
|
|||
|
|
|
|||
|
|
args = public.dict_obj()
|
|||
|
|
args['data'] = get
|
|||
|
|
args['mod_name'] = get.dns_type
|
|||
|
|
args['def_name'] = def_name
|
|||
|
|
|
|||
|
|
return obj_main.model(args)
|
|||
|
|
|
|||
|
|
def get_domain_dns_config(self, get):
|
|||
|
|
self.__init_data()
|
|||
|
|
|
|||
|
|
dns_data = self.get_dns_data(get)
|
|||
|
|
root_data = public.M('ssl_domains').field('id,domain,dns_id').select()
|
|||
|
|
where_sql = '1=1'
|
|||
|
|
param = []
|
|||
|
|
if 'site_id' in get:
|
|||
|
|
where_sql += " AND pid in ({})".format(get.site_id)
|
|||
|
|
# param.append("({})".format(get.site_id))
|
|||
|
|
if 'domain_name' in get:
|
|||
|
|
where_sql += " AND name like ?"
|
|||
|
|
param.append('%{}%'.format(get.domain_name))
|
|||
|
|
data = public.M('domain').where(where_sql, param).field('name,pid as site_id').select() + public.M('docker_domain').where(where_sql, param).field('name,pid as site_id').select()
|
|||
|
|
for i in data:
|
|||
|
|
root, sub_domain, _ = self.extract_zone(i['name'])
|
|||
|
|
i["status"] = 0
|
|||
|
|
i["domain_id"] = -1
|
|||
|
|
for j in root_data:
|
|||
|
|
if j['domain'] == root:
|
|||
|
|
i["domain_id"] = j["id"]
|
|||
|
|
i["status"] = 1 if j["dns_id"] in dns_data.keys() else 0
|
|||
|
|
break
|
|||
|
|
if 'auto_wildcard' in get and get.auto_wildcard == '1':
|
|||
|
|
new_data = []
|
|||
|
|
for i in data:
|
|||
|
|
root, _, _ = self.extract_zone(i['name'])
|
|||
|
|
# 创建新对象以避免修改原数据
|
|||
|
|
new_item = i.copy()
|
|||
|
|
new_item['name'] = root
|
|||
|
|
if new_item not in new_data:
|
|||
|
|
new_data.append(new_item)
|
|||
|
|
|
|||
|
|
# 处理通配符域名
|
|||
|
|
wildcard_item = new_item.copy()
|
|||
|
|
wildcard_item['name'] = "*." + root
|
|||
|
|
if wildcard_item not in new_data:
|
|||
|
|
new_data.append(wildcard_item)
|
|||
|
|
data = new_data
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def get_site_list(self, get):
|
|||
|
|
return public.M('sites').field('name,id').select() + public.M('docker_sites').field('name,id').select()
|
|||
|
|
|
|||
|
|
def create_report_task(self, get):
|
|||
|
|
from mod.base.push_mod import manager
|
|||
|
|
|
|||
|
|
sender_lsit = get.sender.split(",")
|
|||
|
|
task_data = {"task_data":{"tid":"70","type":"domain_endtime","title":"Domain name expiration","status":True,"count":0,"interval":600,"project":get.domain,"cycle":int(get.cycle)},"sender":sender_lsit,"number_rule":{"day_num":0,"total":int(get.total)},"time_rule":{"send_interval":0,"time_range":[0,86399]}}
|
|||
|
|
get.template_id = "70"
|
|||
|
|
get.task_data = json.dumps(task_data)
|
|||
|
|
|
|||
|
|
return manager.PushManager().set_task_conf(get)
|
|||
|
|
|
|||
|
|
def remove_report_task(self, get):
|
|||
|
|
from mod.base.push_mod import manager
|
|||
|
|
return manager.PushManager().remove_task_conf(get)
|
|||
|
|
|
|||
|
|
def add_dns_value_by_domain(self, domain, dns_value, record_type="TXT", is_let_txt=False):
|
|||
|
|
root, _, subd = self.extract_zone(domain, is_let_txt)
|
|||
|
|
domain_name = subd+'.'+root if is_let_txt else domain
|
|||
|
|
try:
|
|||
|
|
data = public.M('ssl_domains').field('dns_id').where("domain=?", (root,)).select()
|
|||
|
|
dns_id = data[0]['dns_id']
|
|||
|
|
except:
|
|||
|
|
dns_id = ''
|
|||
|
|
for dns_config in self.get_dns_data(public.dict_obj()).values():
|
|||
|
|
if root in dns_config.get("domains", []) or str(dns_id) == dns_config["id"]:
|
|||
|
|
args = {
|
|||
|
|
"fun_name": "create_dns_record",
|
|||
|
|
"dns_id": dns_id,
|
|||
|
|
"domain_dns_value": dns_value,
|
|||
|
|
"record_type": record_type,
|
|||
|
|
"domain_name": domain_name,
|
|||
|
|
}
|
|||
|
|
_return = self.run_fun(public.to_dict_obj(args))
|
|||
|
|
if not _return["status"] and '记录已存在' not in _return["msg"]:
|
|||
|
|
raise Exception(public.lang("Failed to set the DNS record：{}".format(_return["msg"])))
|
|||
|
|
else:
|
|||
|
|
return _return
|
|||
|
|
raise Exception(public.lang("No valid DNS API key information was found for the domain name {}.".format(domain)))
|
|||
|
|
|
|||
|
|
def get_dns_value_by_domain(self, domain, record_type="TXT", is_let_txt=False):
|
|||
|
|
root, _, subd = self.extract_zone(domain, is_let_txt)
|
|||
|
|
domain_name = subd + '.' + root if is_let_txt else domain
|
|||
|
|
try:
|
|||
|
|
data = public.M('ssl_domains').field('dns_id').where("domain=?", (root,)).select()
|
|||
|
|
dns_id = data[0]['dns_id']
|
|||
|
|
except:
|
|||
|
|
dns_id = ''
|
|||
|
|
for dns_config in self.get_dns_data(public.dict_obj()).values():
|
|||
|
|
if root in dns_config.get("domains", []) or str(dns_id) == dns_config["id"]:
|
|||
|
|
args = {
|
|||
|
|
"fun_name": "get_dns_record",
|
|||
|
|
"dns_id": dns_id,
|
|||
|
|
"domain_name": domain,
|
|||
|
|
}
|
|||
|
|
record_data = self.run_fun(public.to_dict_obj(args))
|
|||
|
|
_return = []
|
|||
|
|
for record in record_data.get("list", []):
|
|||
|
|
if record["type"] == record_type and record["name"] == domain_name:
|
|||
|
|
_return.append(record["RecordId"])
|
|||
|
|
return _return, dns_id
|
|||
|
|
return [], ""
|
|||
|
|
|
|||
|
|
def del_dns_value_by_domain(self, domain, record_type="TXT", is_let_txt=False):
|
|||
|
|
ids, dns_id = self.get_dns_value_by_domain(domain, record_type, is_let_txt)
|
|||
|
|
for record_id in ids:
|
|||
|
|
args = {
|
|||
|
|
"fun_name": "delete_dns_record",
|
|||
|
|
"dns_id": dns_id,
|
|||
|
|
"RecordId": record_id,
|
|||
|
|
"domain_name": domain,
|
|||
|
|
}
|
|||
|
|
_return = self.run_fun(public.to_dict_obj(args))
|
|||
|
|
if not _return["status"]:
|
|||
|
|
raise Exception(public.lang("Failed to delete the DNS record：{}".format(_return["msg"])))
|
|||
|
|
|
|||
|
|
def get_sub_domains(self, get):
|
|||
|
|
import socket
|
|||
|
|
root_domain = get.root_domain
|
|||
|
|
domains = public.M('domain').where("name like ?", ('%'+root_domain,)).field('name,pid').select()
|
|||
|
|
for domain in domains:
|
|||
|
|
site_data = public.M('sites').where("id = ?", (domain['pid'],)).find()
|
|||
|
|
domain.update({"site_name": site_data['name']})
|
|||
|
|
domain.update({"record_a": socket.gethostbyname(domain['name'])})
|
|||
|
|
return domains
|
|||
|
|
|
|||
|
|
def add_domain(self, get):
|
|||
|
|
domain_name = get.domain_name
|
|||
|
|
dns_id = get.dns_id
|
|||
|
|
data = public.M('ssl_domains').field('id').where("domain=?", (domain_name,)).select()
|
|||
|
|
if data:
|
|||
|
|
return public.returnMsg(False, public.lang('The domain name already exists.'))
|
|||
|
|
public.M('ssl_domains').add('domain,dns_id,type_id,endtime,ps', (domain_name, dns_id, 0, 0, ''))
|
|||
|
|
return public.returnMsg(True, public.lang('add success'))
|
|||
|
|
|
|||
|
|
|