267 lines
10 KiB
Python
267 lines
10 KiB
Python
#!/usr/bin/python3
|
||
import json
|
||
import queue
|
||
import re
|
||
import sys
|
||
import socket
|
||
import asyncio
|
||
import threading
|
||
import time
|
||
from typing import Dict, List
|
||
from datetime import datetime
|
||
from utils import *
|
||
import logger
|
||
log = logger.get_logger()
|
||
|
||
def _get_ip_address(domain):
|
||
try:
|
||
ip_address = socket.gethostbyname(domain)
|
||
# log.debug(f"get_ip_address {domain} {ip_address}")
|
||
return ip_address
|
||
except socket.gaierror as e:
|
||
# print(f"Error retrieving IP address: {e}")
|
||
return None
|
||
|
||
def _get_valid_licenses():
|
||
lic_id = 0
|
||
while True:
|
||
lic_id += 1
|
||
domain = f"szmaslic{lic_id:02d}"
|
||
if _get_ip_address(domain) is None:
|
||
break
|
||
yield "27003@" + domain
|
||
|
||
class UsedInfo:
|
||
def __init__(self):
|
||
self.user = ''
|
||
self.host = ''
|
||
self.lic = ''
|
||
self.start_time:datetime = None
|
||
def __str__(self):
|
||
time = datetime.now() - self.start_time
|
||
login_time = (time.days * 24 + time.seconds / 3600.0)
|
||
obj = {
|
||
"user":self.user,
|
||
"host":self.host,
|
||
'lic':self.lic,
|
||
'login':f'{login_time:.2f}h'
|
||
}
|
||
return str(obj)
|
||
def __repr__(self):
|
||
return self.__str__()
|
||
|
||
class LicenseInfo:
|
||
def __init__(self, name='', total=0, used=0):
|
||
self.name = name
|
||
self.total = total
|
||
self.used = used
|
||
self.used_info:List[UsedInfo] = []
|
||
|
||
def available(self):
|
||
return self.total - self.used
|
||
|
||
def __str__(self):
|
||
obj = {
|
||
"name":self.name,
|
||
"total":self.total,
|
||
"used":self.used,
|
||
"used_info":self.used_info
|
||
}
|
||
return str(obj)
|
||
def __repr__(self):
|
||
return self.__str__()
|
||
|
||
|
||
class LicenseManager:
|
||
def __init__(self):
|
||
self.wait_queue = queue.Queue()
|
||
self.ack_queue = queue.Queue()
|
||
self.licenses_stat = LicenseStatistic()
|
||
self.licenses:Dict[str,LicenseInfo] = {}
|
||
#启动定时获取
|
||
self.licenses_stat.lic_info_updater()
|
||
|
||
def lic_info_updater(self):
|
||
self.update()
|
||
if self.wait_queue.qsize() > 0:
|
||
for lic, available_cnt in self.licenses_stat.available():
|
||
for i in range(available_cnt):
|
||
req = self.wait_queue.get()
|
||
if req == None:
|
||
return
|
||
req["lic"] = lic
|
||
self.ack_queue.put(req)
|
||
threading.Timer(10, self.lic_info_updater).start() #每间隔一段时间更新一次lic的使用情况
|
||
|
||
def _in_wait_queue(self, user):
|
||
return any(user == _.user for _ in list(self.wait_queue))
|
||
|
||
def request(self, user:str, host:str):
|
||
#如果已经是在线的状态就不允许申请
|
||
if any(user == _.user for _ in self.licenses_stat.online_users()):
|
||
log.warning(f"{user} is online, please try again later")
|
||
return False
|
||
#如果是已经在队列中则返回等待
|
||
if self._in_wait_queue(user):
|
||
return None
|
||
|
||
data = {
|
||
"user":user,
|
||
"host":host,
|
||
"start_time":time.now()
|
||
}
|
||
self.wait_queue.put(data)
|
||
return False
|
||
|
||
def cancel(self):
|
||
pass
|
||
|
||
def release(self):
|
||
pass
|
||
|
||
def get_wait_queue(self):
|
||
return [_ for _ in self.wait_queue.queue]
|
||
|
||
class LicenseStatistic:
|
||
def __init__(self):
|
||
self.licenses:Dict[str,LicenseInfo] = {}
|
||
|
||
def update(self):
|
||
# 更新license信息，使用异步获取
|
||
tasks = []
|
||
for lic in _get_valid_licenses():
|
||
log.trace(f'find valid lic: {lic}')
|
||
tasks.append(get_license_used_info(lic))
|
||
loop = asyncio.get_event_loop()
|
||
infos = loop.run_until_complete(asyncio.gather(*tasks, return_exceptions=True))
|
||
for info in infos:
|
||
if info is not None:
|
||
self.licenses[info.name] = info
|
||
if not self.licenses:
|
||
log.warning("not vcast lic found")
|
||
def get_lic_list(self):
|
||
# 获取所有的license名称
|
||
return self.licenses.keys()
|
||
|
||
def available(self):
|
||
# 获取所有的可用数量
|
||
for _,lic in self.licenses.items():
|
||
value = lic.available()
|
||
if value > 0:
|
||
yield (lic.name, value) #lic的name和个数
|
||
|
||
def online_users(self):
|
||
# 获取在线用户
|
||
for used_info in self.used():
|
||
yield used_info.user #返回的是user name
|
||
|
||
def used(self):
|
||
# 获取所有的已使用信息
|
||
for _,info in self.licenses.items():
|
||
for used in info.used_info:
|
||
yield used #返回的是UsedInfo对象
|
||
|
||
def sorted(self):
|
||
info = {}
|
||
for used_info in self.used():
|
||
time = datetime.now() - used_info.start_time
|
||
login_time = (time.days * 24 + time.seconds / 3600.0)
|
||
info[used_info.user] = login_time
|
||
sorted_info = dict(sorted(info.items(), key=lambda item: item[1], reverse=True))
|
||
return sorted_info #返回的是字典，key为username value为登录时间
|
||
|
||
def __str__(self):
|
||
string = str(self.licenses).replace("'", '"')
|
||
log.info(string)
|
||
obj = json.loads(string)
|
||
string = json.dumps(obj, indent = 4)
|
||
return string
|
||
def __repr__(self):
|
||
return self.__str__()
|
||
|
||
"""
|
||
{"27003@szmaslic03": {"name": "27003@szmaslic03", "total": 2, "used": 1, "used_info": "[{"name": "haoxiang.ran", "host": "szl3bc12808", "lic": "szmaslic03", "login": "0.41"}]"}, "27003@szmaslic04": {"name": "27003@szmaslic04", "total": 2, "used": 1, "used_info": "[{"name": "macro.yang", "host": "szl3bc12804", "lic": "szmaslic04", "login": "52.28"}]"}, "27003@szmaslic06": {"name": "27003@szmaslic06", "total": 2, "used": 2, "used_info": "[{"name": "zhenhao.he", "host": "szl3bc12804", "lic": "szmaslic06", "login": "0.20"}, {"name": "phillip.chan", "host": "szl3bc12810", "lic": "szmaslic06", "login": "0.16"}]"}, "27003@szmaslic07": {"name": "27003@szmaslic07", "total": 2, "used": 1, "used_info": "[{"name": "zhiyi.li", "host": "szl3bc12806", "lic": "szmaslic07", "login": "2.96"}]"}, "27003@szmaslic08": {"name": "27003@szmaslic08", "total": 2, "used": 2, "used_info": "[{"name": "soo.liu", "host": "szl3bc12809", "lic": "szmaslic08", "login": "6.55"}, {"name": "haichao.ou", "host": "szl3bc12806", "lic": "szmaslic08", "login": "5.93"}]"}, "27003@szmaslic09": {"name": "27003@szmaslic09", "total": 2, "used": 2, "used_info": "[{"name": "louie.liang", "host": "szl3bc12810", "lic": "szmaslic09", "login": "8.31"}, {"name": "zabbix", "host": "szl3bc06409", "lic": "szmaslic09", "login": "0.90"}]"}, "27003@szmaslic10": {"name": "27003@szmaslic10", "total": 2, "used": 2, "used_info": "[{"name": "harvey.li", "host": "szl3bc12810", "lic": "szmaslic10", "login": "57.38"}, {"name": "kw.hu", "host": "szl3bc12809", "lic": "szmaslic10", "login": "1.41"}]"}}
|
||
"""
|
||
async def get_license_used_info(lic):
|
||
server_used_info = f"timeout 1 /tools/software/vcast/flexlm/lmutil lmstat -a -c {lic} -a"
|
||
ret,out,err = shell(server_used_info, checkret=False)
|
||
# log.info(f"get_license_used_info {lic} {out} {ret} {err}")
|
||
if ret != 0:
|
||
return None
|
||
""" data like:
|
||
Users of VCAST_C_ENT_0800: (Total of 2 licenses issued; Total of 2 licenses in use)
|
||
|
||
"VCAST_C_ENT_0800" v23, vendor: vector, expiry: permanent(no expiration date)
|
||
vendor_string: CUST:64925:
|
||
floating license
|
||
|
||
zhenhao.he szl3bc12804 /dev/tty (v23) (szmaslic06/27003 337), start Thu 1/2 8:52
|
||
phillip.chan szl3bc12810 /dev/tty (v23) (szmaslic06/27003 233), start Thu 1/2 14:59
|
||
|
||
Users of CCAST_0801: (Total of 2 licenses issued; Total of 0 licenses in use)
|
||
|
||
"""
|
||
status = 'idle'
|
||
info = LicenseInfo(lic)
|
||
for line in out.splitlines():
|
||
line = line.strip()
|
||
if status == 'idle':
|
||
#Error getting status: Cannot find license file. (-1,359:2 "No such file or directory")
|
||
if line.startswith("Error getting status"):
|
||
log.warning(f"{lic}")
|
||
return None
|
||
if not line.startswith("Users of VCAST_C_ENT_0800:"):
|
||
continue
|
||
status = 'start'
|
||
#[Users of VCAST_C_ENT_0800: (Total of 2 licenses issued; Total of 1 license in use)
|
||
pattern = r"\(Total of (\d+).*Total of (\d+).*\)"
|
||
ret = re.findall(pattern, line)
|
||
if not ret:
|
||
log.warning(f"{lic} parser [{line}] fail")
|
||
return None
|
||
(total, used) = ret[0]
|
||
info.total = int(total)
|
||
info.used = int(used)
|
||
elif status == 'start':
|
||
#zhenhao.he szl3bc12804 /dev/tty (v23) (szmaslic06/27003 337), start Thu 1/2 8:52
|
||
pattern = '(\S+) (\S+).*\((\S+)/.*\), start (.*)'
|
||
ret = re.findall(pattern, line)
|
||
if not ret:
|
||
continue
|
||
used = UsedInfo()
|
||
(used.user,used.host,used.lic,time_string) = ret[0]
|
||
#处理跨年的情况
|
||
year = datetime.now().year
|
||
start_time = datetime.strptime(f"{year} {time_string}", "%Y %a %m/%d %H:%M")
|
||
if start_time > datetime.now():
|
||
year -= 1
|
||
start_time = datetime.strptime(f"{year} {time_string}", "%Y %a %m/%d %H:%M")
|
||
used.start_time = start_time
|
||
# log.info(f"{lic} => {used}")
|
||
info.used_info.append(used)
|
||
# Next section
|
||
elif line.startswith('Users of '):
|
||
break
|
||
return info
|
||
|
||
def draw_time_grpah(license_manager):
|
||
info = license_manager.sorted()
|
||
max_time = max(info.values())
|
||
name_len = max([len(_) for _ in info.keys()])
|
||
top = 100
|
||
for user, time in info.items():
|
||
curr = '█' * max(int(top * time / max_time), 1)
|
||
name_remain = ' ' * (name_len - len(user))
|
||
log.info(f"{user}{name_remain} :{time:5.2f}h {curr}")
|
||
|
||
if __name__ == '__main__':
|
||
log.info("License status query...")
|
||
lm = LicenseStatistic()
|
||
log.info(lm)
|
||
available_lic = lm.available()
|
||
if not available_lic:
|
||
log.info("No license available")
|
||
log.info(f"All licenses is: {[str(_) for _ in lm.get_lic_list()]}")
|
||
for lic,number in lm.available():
|
||
log.info(f"{lic}: {number} lic available")
|
||
draw_time_grpah(lm) |