mirror of https://github.com/jumpserver/jumpserver
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from collections import Counter
|
|
|
|
__all__ = ['FormatAssetInfo']
|
|
|
|
|
|
class FormatAssetInfo:
|
|
|
|
def __init__(self, tp):
|
|
self.tp = tp
|
|
|
|
@staticmethod
|
|
def get_cpu_model_count(cpus):
|
|
try:
|
|
if len(cpus) % 3 == 0:
|
|
step = 3
|
|
models = [cpus[i + 2] for i in range(0, len(cpus), step)]
|
|
elif len(cpus) % 2 == 0:
|
|
step = 2
|
|
models = [cpus[i + 1] for i in range(0, len(cpus), step)]
|
|
else:
|
|
raise ValueError("CPU list format not recognized")
|
|
|
|
model_counts = Counter(models)
|
|
result = ', '.join([f"{model} x{count}" for model, count in model_counts.items()])
|
|
except Exception as e:
|
|
print(f"Error processing CPU model list: {e}")
|
|
result = ''
|
|
return result
|
|
|
|
def posix_format(self, info):
|
|
cpus = self.get_cpu_model_count(info.get('cpu_model', []))
|
|
|
|
info['cpu_model'] = cpus
|
|
info['cpu_count'] = info.get('cpu_count', 0)
|
|
return info
|
|
|
|
def run(self, method_id_meta_mapper, info):
|
|
for k, v in info.items():
|
|
info[k] = v.strip() if isinstance(v, str) else v
|
|
|
|
run_method_name = None
|
|
for k, v in method_id_meta_mapper.items():
|
|
if self.tp not in v['type']:
|
|
continue
|
|
run_method_name = k.replace(f'{v["method"]}_', '')
|
|
|
|
if not run_method_name:
|
|
return info
|
|
|
|
if hasattr(self, f'{run_method_name}_format'):
|
|
return getattr(self, f'{run_method_name}_format')(info)
|
|
return info
|