mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
#coding: utf-8
|
|
import pwd
|
|
|
|
import public,re
|
|
|
|
class projectBase:
|
|
|
|
def check_port(self, port):
|
|
'''
|
|
@name 检查端口是否被占用
|
|
@args port:端口号
|
|
@return: 被占用返回True，否则返回False
|
|
@author: lkq 2021-08-28
|
|
'''
|
|
a = public.ExecShell("netstat -nltp|awk '{print $4}'")
|
|
if a[0]:
|
|
if re.search(':' + port + '\n', a[0]):
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
def is_domain(self, domain):
|
|
'''
|
|
@name 验证域名合法性
|
|
@args domain:域名
|
|
@return: 合法返回True，否则返回False
|
|
@author: lkq 2021-08-28
|
|
'''
|
|
import re
|
|
domain_regex = re.compile(r'(?:[A-Z0-9_](?:[A-Z0-9-_]{0,247}[A-Z0-9])?\.)+(?:[A-Z]{2,6}|[A-Z0-9-]{2,}(?<!-))\Z', re.IGNORECASE)
|
|
return True if domain_regex.match(domain) else False
|
|
|
|
|
|
def generate_random_port(self):
|
|
'''
|
|
@name 生成随机端口
|
|
@args
|
|
@return: 端口号
|
|
@author: lkq 2021-08-28
|
|
'''
|
|
import random
|
|
port = str(random.randint(5000, 10000))
|
|
while True:
|
|
if not self.check_port(port): break
|
|
port = str(random.randint(5000, 10000))
|
|
return port
|
|
|
|
def IsOpen(self, port):
|
|
'''
|
|
@name 检查端口是否被占用
|
|
@args port:端口号
|
|
@return: 被占用返回True，否则返回False
|
|
@author: lkq 2021-08-28
|
|
'''
|
|
ip = '0.0.0.0'
|
|
import socket
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
try:
|
|
s.connect((ip, int(port)))
|
|
s.shutdown(2)
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
@staticmethod
|
|
def get_system_user_list(get=None):
|
|
"""
|
|
默认只返回uid>= 1000 的用户 和 root
|
|
get中包含 sys_user 返回 uid>= 100 的用户 和 root
|
|
get中包含 all_user 返回所有的用户
|
|
"""
|
|
sys_user = False
|
|
all_user = False
|
|
if get is not None:
|
|
if hasattr(get, "sys_user"):
|
|
sys_user = True
|
|
if hasattr(get, "all_user"):
|
|
all_user = True
|
|
|
|
user_set = set()
|
|
try:
|
|
for tmp_uer in pwd.getpwall():
|
|
if tmp_uer.pw_uid == 0:
|
|
user_set.add(tmp_uer.pw_name)
|
|
elif tmp_uer.pw_uid >= 1000:
|
|
user_set.add(tmp_uer.pw_name)
|
|
elif sys_user and tmp_uer.pw_uid >= 100:
|
|
user_set.add(tmp_uer.pw_name)
|
|
elif all_user:
|
|
user_set.add(tmp_uer.pw_name)
|
|
except Exception:
|
|
pass
|
|
return list(user_set) |