177 lines
6.6 KiB
Python
177 lines
6.6 KiB
Python
|
#coding: utf-8
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# YakPanel
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# Copyright (c) 2015-2020 YakPanel(www.yakpanel.com) All rights reserved.
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
# Author: zhwen <zhw@yakpanel.com>
|
||
|
|
#-------------------------------------------------------------------
|
||
|
|
|
||
|
|
#------------------------------
|
||
|
|
# 允许网站自动列出目录
|
||
|
|
#------------------------------
|
||
|
|
|
||
|
|
import public
|
||
|
|
import re
|
||
|
|
|
||
|
|
class website_auto_index:
|
||
|
|
|
||
|
|
def _init_conf(self, website):
|
||
|
|
self.ng_website_conf = '/www/server/panel/vhost/nginx/{}.conf'.format(website)
|
||
|
|
self.ap_website_conf = '/www/server/panel/vhost/apache/{}.conf'.format(website)
|
||
|
|
self.webserver = public.get_webserver()
|
||
|
|
|
||
|
|
# 获取某个网站自动索引目录
|
||
|
|
def get_auto_index(self, args):
|
||
|
|
'''
|
||
|
|
# 添加某个网站禁止运行PHP
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
self._init_conf(args.website)
|
||
|
|
if self.webserver == 'nginx':
|
||
|
|
data = self._get_nginx_auto_index()
|
||
|
|
elif self.webserver == 'apache':
|
||
|
|
data = self._get_apache_auto_index()
|
||
|
|
else:
|
||
|
|
data = public.returnMsg(False,'Sorry, OLS does not currently support this feature')
|
||
|
|
return data
|
||
|
|
|
||
|
|
def _get_nginx_auto_index(self):
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_AUTOINDEX_.*', conf)
|
||
|
|
deny_name = [i.split('_')[-1] for i in data]
|
||
|
|
result = []
|
||
|
|
for i in deny_name:
|
||
|
|
reg = '#BEGIN_AUTOINDEX_{}\n\\s*location\\s*\\~\\*\\s*\\^(.*)\\.\\*.*\\((.*)\\)\\$'.format(i)
|
||
|
|
deny_directory = re.search(reg, conf).groups()[0]
|
||
|
|
deny_suffix = re.search(reg, conf).groups()[1]
|
||
|
|
result.append({'name': i, 'dir': deny_directory, 'suffix': deny_suffix})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def _get_apache_auto_index(self):
|
||
|
|
conf = public.readFile(self.ap_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_AUTOINDEX_.*', conf)
|
||
|
|
deny_name = [i.split('_')[-1] for i in data]
|
||
|
|
result = []
|
||
|
|
for i in deny_name:
|
||
|
|
reg = '#BEGIN_AUTOINDEX_{}\n\\s*<Directory\\s*\\~\\s*"(.*)\\.\\*.*\\((.*)\\)\\$'.format(i)
|
||
|
|
deny_directory = re.search(reg, conf).groups()[0]
|
||
|
|
deny_suffix = re.search(reg, conf).groups()[1]
|
||
|
|
result.append({'name': i, 'dir': deny_directory, 'suffix': deny_suffix})
|
||
|
|
return result
|
||
|
|
|
||
|
|
def set_auto_index(self, args):
|
||
|
|
'''
|
||
|
|
# 添加自动索引目录
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:param args: index_name 规则名称 str
|
||
|
|
:param args: dir 自动索引目录 str
|
||
|
|
:param args: act 操作方法
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
if self.webserver == 'openlitespeed':
|
||
|
|
return public.returnMsg(False, public.lang("Sorry, OLS does not currently support this feature"))
|
||
|
|
|
||
|
|
tmp = self._check_args(args)
|
||
|
|
if tmp:
|
||
|
|
return tmp
|
||
|
|
deny_name = args.index_name
|
||
|
|
dir = args.dir
|
||
|
|
website = args.website
|
||
|
|
self._init_conf(website)
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
data = re.findall('BEGIN_AUTOINDEX_.*', conf)
|
||
|
|
exist_index_name = [i.split('_')[-1] for i in data]
|
||
|
|
if args.act == 'edit':
|
||
|
|
if deny_name not in exist_index_name:
|
||
|
|
return public.returnMsg(False, public.lang("The specify rule name is not exists! [ {} ]", deny_name))
|
||
|
|
self.del_auto_index(args)
|
||
|
|
else:
|
||
|
|
if deny_name in exist_index_name:
|
||
|
|
return public.returnMsg(False, public.lang("The specify rule name is already exists! [ {} ]", deny_name))
|
||
|
|
self._set_nginx_auto_index(deny_name, dir)
|
||
|
|
self._set_apache_auto_index(deny_name, dir)
|
||
|
|
public.serviceReload()
|
||
|
|
return public.returnMsg(True, public.lang("Add Successfully"))
|
||
|
|
|
||
|
|
def _set_nginx_auto_index(self, name, dir=None):
|
||
|
|
conf = public.readFile(self.ng_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
if not dir:
|
||
|
|
reg = '\\s*#BEGIN_AUTOINDEX_{n}\n(.|\n)*#END_AUTOINDEX_{n}\n'.format(n=name)
|
||
|
|
conf = re.sub(reg, '', conf)
|
||
|
|
else:
|
||
|
|
new = '''
|
||
|
|
#BEGIN_AUTOINDEX_%s
|
||
|
|
location %s
|
||
|
|
{
|
||
|
|
autoindex on;
|
||
|
|
autoindex_format html;
|
||
|
|
autoindex_exact_size off;
|
||
|
|
autoindex_localtime on;
|
||
|
|
}
|
||
|
|
#END_AUTOINDEX_%s
|
||
|
|
''' % (name, dir, name)
|
||
|
|
if '#BEGIN_AUTOINDEX_{}\n'.format(name) in conf:
|
||
|
|
return True
|
||
|
|
conf = re.sub('#ERROR-PAGE-END', '#ERROR-PAGE-END' + new, conf)
|
||
|
|
public.writeFile(self.ng_website_conf, conf)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def _set_apache_auto_index(self, name, dir=None):
|
||
|
|
conf = public.readFile(self.ap_website_conf)
|
||
|
|
if not conf:
|
||
|
|
return False
|
||
|
|
if not dir:
|
||
|
|
reg = '\\s*#BEGIN_AUTOINDEX_{n}\n(.|\n)*#END_AUTOINDEX_{n}'.format(n=name)
|
||
|
|
conf = re.sub(reg, '', conf)
|
||
|
|
else:
|
||
|
|
new = '''
|
||
|
|
#BEGIN_AUTOINDEX_{n}
|
||
|
|
<Location "{d}">
|
||
|
|
Options Indexes FollowSymLinks
|
||
|
|
AllowOverride None
|
||
|
|
Options +Indexes
|
||
|
|
</Directory>
|
||
|
|
#END_AUTOINDEX_{n}
|
||
|
|
'''.format(n=name, d=dir)
|
||
|
|
if '#BEGIN_AUTOINDEX_{}'.format(name) in conf:
|
||
|
|
return True
|
||
|
|
conf = re.sub(r'#DENY\s*FILES', new + '\n #DENY FILES', conf)
|
||
|
|
public.writeFile(self.ap_website_conf, conf)
|
||
|
|
return True
|
||
|
|
|
||
|
|
# 删除某个网站禁止运行PHP
|
||
|
|
def del_auto_index(self, args):
|
||
|
|
'''
|
||
|
|
# 添加某个网站禁止运行PHP
|
||
|
|
author: zhwen<zhw@yakpanel.com>
|
||
|
|
:param args: website 网站名 str
|
||
|
|
:param args: deny_name 规则名称 str
|
||
|
|
:return:
|
||
|
|
'''
|
||
|
|
self._init_conf(args.website)
|
||
|
|
deny_name = args.deny_name
|
||
|
|
self._set_nginx_auto_index(deny_name)
|
||
|
|
self._set_apache_auto_index(deny_name)
|
||
|
|
public.serviceReload()
|
||
|
|
return public.returnMsg(True, public.lang("Delete Successfully"))
|
||
|
|
|
||
|
|
# 检查传入参数
|
||
|
|
def _check_args(self, args):
|
||
|
|
if hasattr(args, 'deny_name'):
|
||
|
|
if len(args.deny_name) < 3:
|
||
|
|
return public.returnMsg(False, public.lang("Rule name needs to be greater than 3 bytes"))
|
||
|
|
if hasattr(args, 'dir'):
|
||
|
|
if not args.dir:
|
||
|
|
return public.returnMsg(False, public.lang("Directory cannot be empty"))
|