【已解答】宝塔跑分获取不到CPU型号出错的BUG！

1. #coding: utf-8
   
2. # +-------------------------------------------------------------------
   
3. # | 宝塔Linux面板 x3
   
4. # +-------------------------------------------------------------------
   
5. # | Copyright (c) 2015-2016 宝塔软件(http://bt.cn) All rights reserved.
   
6. # +-------------------------------------------------------------------
   
7. # | Author: 黄文良 <2879625666@qq.com>
   
8. # +-------------------------------------------------------------------
   
9. 
   
10. #+--------------------------------------------------------------------
    
11. #|   服务器测试
    
12. #+--------------------------------------------------------------------
    
13. 
    
14. import time,psutil,random,os,urllib,binascii,json,public,re,sys;
    
15. class score_main:
    
16.    
    
17.     __APIURL = 'http://www.bt.cn/api/Auth';
    
18.     __UPATH = 'data/userInfo.json';
    
19.     __userInfo = None;
    
20.     __PDATA = None;
    
21.    
    
22.     def CheckToken(self):
    
23.         pdata = {}
    
24.         data = {}
    
25.         if os.path.exists(self.__UPATH):
    
26.             self.__userInfo = json.loads(public.readFile(self.__UPATH));
    
27.             if self.__userInfo:
    
28.                 pdata['access_key'] = self.__userInfo['access_key'];
    
29.                 data['secret_key'] = self.__userInfo['secret_key'];
    
30.         else:
    
31.             pdata['access_key'] = 'test';
    
32.             data['secret_key'] = '123456';
    
33.         pdata['data'] = data;
    
34.         self.__PDATA = pdata;
    
35.             
    
36.     #修正信息
    
37.     def SetScore(self,get):
    
38.         if hasattr(get,'ips'):
    
39.             return self.SubmitSetScore('ips',get.ips);
    
40.         if hasattr(get,'virt'):
    
41.             return self.SubmitSetScore('virt',get.virt);
    
42.    
    
43.     #发送信息修正
    
44.     def SubmitSetScore(self,key,value):
    
45.         self.CheckToken();
    
46.         self.__PDATA['data'][key] = value
    
47.         self.__PDATA['data'] = self.De_Code(self.__PDATA['data']);
    
48.         result = json.loads(public.httpPost(self.__APIURL+'/SetSocre',self.__PDATA));
    
49.         result['data'] = self.En_Code(result['data']);
    
50.         return result;
    
51.    
    
52.     #获取得分列表
    
53.     def GetScore(self,get):
    
54.         self.CheckToken();
    
55.         self.__PDATA['data'] = self.De_Code(self.__PDATA['data']);
    
56.         result = json.loads(public.httpPost(self.__APIURL+'/GetSocre',self.__PDATA));
    
57.         result['data'] = self.En_Code(result['data']);
    
58.         result['last'] = self.GetConfig()
    
59.         return result;
    
60.    
    
61.     #获取配置信息
    
62.     def GetConfig(self,get=None):
    
63.         virt = '/usr/sbin/virt-what'
    
64.         if not os.path.exists(virt):
    
65.             if os.path.exists('/etc/yum.repos.d/epel.repo'):
    
66.                 os.system('mv /etc/yum.repos.d/epel.repo /etc/yum.repos.d/epel.repo_backup');
    
67.             os.system('yum install virt-what -y');
    
68.             if os.path.exists('/etc/yum.repos.d/epel.repo_backup'):
    
69.                 os.system('mv /etc/yum.repos.d/epel.repo_backup /etc/yum.repos.d/epel.repo');
    
70.         
    
71.         data = {}
    
72.         data['virt'] = public.ExecShell('virt-what')[0].strip();
    
73.         
    
74.         # 修复CPU信息提取
    
75.         try:
    
76.             cpuinfo = open('/proc/cpuinfo','r').read()
    
77.             
    
78.             # 方案1：优先尝试ARM架构字段
    
79.             arm_implementer = re.search(r"CPU implementer\s+:\s+(0x[\da-fA-F]+)", cpuinfo)
    
80.             arm_part = re.search(r"CPU part\s+:\s+(0x[\da-fA-F]+)", cpuinfo)
    
81.             
    
82.             if arm_implementer and arm_part:
    
83.                 implementer_code = arm_implementer.groups()[0]
    
84.                 part_code = arm_part.groups()[0]
    
85.                 data['cpu'] = f"ARM {self._decode_arm_cpu(implementer_code, part_code)}"
    
86.             else:
    
87.                 # 方案2：回退到x86的model name
    
88.                 tmp = re.search(r"model\s+name\s+:\s+(.+)", cpuinfo)
    
89.                 data['cpu'] = tmp.groups()[0].strip() if tmp else "Unknown CPU"
    
90.                
    
91.         except Exception as e:
    
92.             data['cpu'] = f"Error: {str(e)}"
    
93.          # 修复CPU信息结束
    
94.          
    
95.         data['core'] = psutil.cpu_count();
    
96.         data['memory'] = psutil.virtual_memory().total/1024/1024
    
97.         data['system'] = self.GetSystemVersion();
    
98.         
    
99.         scoreInfo = self.readScore();
    
100.         data['disk'] = str(scoreInfo['read'])+','+str(scoreInfo['write'])
     
101.         data['mem_score'] = scoreInfo['mem'];
     
102.         data['cpu_score'] = scoreInfo['cpu1'] + scoreInfo['cpu2'] + scoreInfo['cpu3'] + scoreInfo['cpu4'];
     
103.         data['disk_score'] = scoreInfo['disk_score'];
     
104.         data['total_score'] = scoreInfo['mem']+data['cpu_score']+scoreInfo['disk_score'];
     
105.         return data;
     
106. # 修复CPU信息提取
     
107. def _decode_arm_cpu(self, implementer, part):
     
108.         """将ARM CPU代码转换为可读名称"""
     
109.         arm_cpu_db = {
     
110.             ("0x41", "0xd03"): "Cortex-A53",
     
111.             ("0x41", "0xd04"): "Cortex-A35",
     
112.             ("0x41", "0xd05"): "Cortex-A55",  # 您的CPU
     
113.             ("0x41", "0xd0a"): "Cortex-A75",
     
114.         }
     
115.         return arm_cpu_db.get((implementer.lower(), part.lower()), f"Unrecognized ARM CPU ({implementer}/{part})")
     
116. # 修复CPU信息提取
     
117. #提交到云端
     
118.     def SubmitScore(self,get):
     
119.         try:
     
120.             self.CheckToken();
     
121.             pdata = self.GetConfig(get);
     
122.             if not pdata['total_score']: return public.returnMsg(False,'请先跑分!');
     
123.             pdata['secret_key'] = self.__userInfo['secret_key'];
     
124.             self.__PDATA['data'] = self.De_Code(pdata);
     
125.             result = json.loads(public.httpPost(self.__APIURL+'/SubmitScore',self.__PDATA));
     
126.             result['data'] = self.En_Code(result['data']);
     
127.             return result;
     
128.         except:
     
129.             return None;
     
130.         
     
131.     #取操作系统版本
     
132.     def GetSystemVersion(self):
     
133.         version = public.readFile('/etc/redhat-release')
     
134.         if not version:
     
135.             version = public.readFile('/etc/issue').replace('\\n \\l','').strip();
     
136.         else:
     
137.             version = version.replace('release ','').strip();
     
138.         return version
     
139.    
     
140.     #写当前得分
     
141.     def writeScore(self,type,value):
     
142.         scoreFile = 'plugin/score/score.json';
     
143.         tmp = public.readFile(scoreFile)
     
144.         if not tmp:
     
145.             data = {}
     
146.             data['cpu1'] = 0;
     
147.             data['cpu2'] = 0;
     
148.             data['cpu3'] = 0;
     
149.             data['cpu4'] = 0;
     
150.             data['mem'] = 0;
     
151.             data['disk_score'] = 0;
     
152.             data['read'] = 0;
     
153.             data['write'] = 0;
     
154.             public.writeFile(scoreFile,json.dumps(data));
     
155.             tmp = public.readFile(scoreFile)
     
156.         
     
157.         data = json.loads(tmp);
     
158.         data[type] = value;
     
159.         public.writeFile(scoreFile,json.dumps(data));
     
160.    
     
161.     #读当前得分
     
162.     def readScore(self):
     
163.         scoreFile = 'plugin/score/score.json';
     
164.         tmp = public.readFile(scoreFile)
     
165.         if not tmp:
     
166.             data = {}
     
167.             data['cpu1'] = 0;
     
168.             data['cpu2'] = 0;
     
169.             data['cpu3'] = 0;
     
170.             data['cpu4'] = 0;
     
171.             data['mem'] = 0;
     
172.             data['disk_score'] = 0;
     
173.             data['read'] = 0;
     
174.             data['write'] = 0;
     
175.             public.writeFile(scoreFile,json.dumps(data));
     
176.             tmp = public.readFile(scoreFile)
     
177.         data = json.loads(tmp);
     
178.         return data;
     
179.                         
     
180.    
     
181.     #测试CPU
     
182.     def testCpu(self,get,n = 1):
     
183.         data = {}
     
184.         data['cpuCount'] = psutil.cpu_count();
     
185.         if not hasattr(get,'type'): get.type = '0';
     
186.         import re;
     
187.         cpuinfo = open('/proc/cpuinfo','r').read();
     
188.         rep = "model\s+name\s+:\s+(.+)"
     
189.         tmp = re.search(rep,cpuinfo);
     
190.         data['cpuType'] = ""
     
191.         if tmp:
     
192.             data['cpuType'] = tmp.groups()[0];
     
193.         
     
194.         import system
     
195.         data['system'] = system.system().GetSystemVersion();
     
196.         path = 'plugin/score/testcpu';
     
197.         if not os.path.exists(path): os.system('gcc '+path+'.c -o ' +path + ' -lpthread');
     
198.         start = time.time();
     
199.         os.system(path + ' 32 ' + get.type);
     
200.         end = time.time();
     
201.         data['score'] = int(400 * 10 / (end - start));
     
202.         if not os.path.exists(path): data['score'] = 0;
     
203.         self.writeScore('cpu'+get.type, data['score'])
     
204.         return data;
     
205.    
     
206.     #测试整数运算
     
207.     def testInt(self):
     
208.         return self.testIntOrFloat(1);
     
209.    
     
210.     #测试浮点运行
     
211.     def testFloat(self):
     
212.         return self.testIntOrFloat(1.01);
     
213.         
     
214.     #CPU测试体
     
215.     def testIntOrFloat(self,n=1):
     
216.         start = time.time();
     
217.         num  = 10000 * 100;
     
218.         for i in range(num):
     
219.             if i == 0: continue;
     
220.             a = n + i;
     
221.             b = n - i;
     
222.             c = n * i;
     
223.             d = n / i;
     
224.             n = n + 1;
     
225.             
     
226.         end = time.time();
     
227.         usetime = end - start;
     
228.         return num / 100 / usetime;
     
229.    
     
230.     #冒泡算法测试
     
231.     def testBubble(self):
     
232.         start = time.time();
     
233.         num  = 10000 * 5;
     
234.         xx = 'qwertyuiopasdfghjklzxcvbnm1234567890'
     
235.         for c in xrange(num):
     
236.             lst = []
     
237.             for k in range(10):
     
238.                 lst.append(xx[random.randint(0,len(xx)-1)])
     
239.             lst = self.bubbleSort(lst)
     
240.         end = time.time();
     
241.         usetime = end - start;
     
242.         return num / 5 / usetime;
     
243.    
     
244.     #冒泡排序
     
245.     def bubbleSort(self,lst):
     
246.         length = len(lst)
     
247.         for i in xrange(0, length, 1):
     
248.             for j in xrange(0, length-1-i, 1):
     
249.                 if lst[j] < lst[j+1]:
     
250.                     temp = lst[j]
     
251.                     lst[j] = lst[j+1]
     
252.                     lst[j+1] = temp
     
253.         return lst
     
254.    
     
255.     #二叉树算法测试
     
256.     def testTree(self):
     
257.         import testTree
     
258.         
     
259.         start = time.time();
     
260.         elems = range(3000)              #生成树节点
     
261.         tree = testTree.Tree()           #新建一个树对象
     
262.         for elem in elems:                  
     
263.             tree.add(elem)               #逐个加入树的节点
     
264.    
     
265.         tree.level_queue(tree.root)
     
266.         tree.front_digui(tree.root)
     
267.         tree.middle_digui(tree.root)
     
268.         tree.later_digui(tree.root)
     
269.         tree.front_stack(tree.root)
     
270.         tree.middle_stack(tree.root)
     
271.         tree.later_stack(tree.root)
     
272.         
     
273.         end = time.time();
     
274.         usetime = end - start;
     
275.         return 3000 / usetime;
     
276.    
     
277.    
     
278.    
     
279.     #测试内存
     
280.     def testMem(self,get):
     
281.         mem = psutil.virtual_memory()
     
282.         self.writeScore('mem', mem.total/1024/1024)
     
283.         #提交数据
     
284.         self.SubmitScore(get)
     
285.         return int(mem.total/1024/1024);
     
286.    
     
287.     #测试磁盘
     
288.     def testDisk(self,get):
     
289.         import os
     
290.         data = {}
     
291.         os.system('rm -f testDisk_*');
     
292.         filename = "testDisk_" + time.strftime('%Y%m%d%H%M%S',time.localtime());
     
293.         data['write'] =  self.testDiskWrite(filename);
     
294.         import shutil
     
295.         filename2 = "testDisk_" + time.strftime('%Y%m%d%H%M%S',time.localtime());
     
296.         shutil.move(filename,filename2);
     
297.         data['read'] =  self.testDiskRead(filename2);
     
298.         diskIo = psutil.disk_partitions()
     
299.         diskInfo = []
     
300.         for disk in diskIo:
     
301.             tmp = {}
     
302.             tmp['path'] = disk[1]
     
303.             tmp['size'] = psutil.disk_usage(disk[1])[0]
     
304.             diskInfo.append(tmp)
     
305.         data['diskInfo'] = diskInfo;
     
306.         writeDisk = data['write'];
     
307.         if data['write'] > 1000: writeDisk = 1000;
     
308.         readDisk = data['read'];
     
309.         if data['read'] > 1000: readDisk = 1000;
     
310.         
     
311.         data['score'] = (writeDisk * 6) + (readDisk * 6)
     
312.         os.remove(filename2);
     
313.         
     
314.         self.writeScore('disk_score', data['score'])
     
315.         self.writeScore('write', data['write'])
     
316.         self.writeScore('read', data['read'])
     
317.         return data;
     
318.         pass
     
319.    
     
320.     #测试磁盘写入速度
     
321.     def testDiskWrite(self,filename):
     
322.         import random
     
323.         start = time.time();
     
324.         fp = open(filename,'w+');
     
325.         strTest = "";
     
326.         strTmp = "";
     
327.         for n in range(4):
     
328.             strTmp += chr(random.randint(97, 122))
     
329.         for n in range(1024):
     
330.             strTest += strTmp;
     
331.    
     
332.         for i in range(1024 * 256):
     
333.             fp.write(strTest);
     
334.             
     
335.         del(strTest);
     
336.         del(strTmp);
     
337.         fp.close()
     
338.         end = time.time();
     
339.         usetime = end - start;
     
340.         return int(1024/usetime);
     
341.         
     
342.     #测试磁盘读取速度   
     
343.     def testDiskRead(self,filename):
     
344.         os.system('echo 3 > /proc/sys/vm/drop_caches');
     
345.         import random
     
346.         start = time.time();
     
347.         fp = open(filename,'r');
     
348.         size = 4096;
     
349.         while True:
     
350.             tmp = fp.read(size);
     
351.             if not tmp: break;
     
352.             del(tmp);
     
353.         fp.close()
     
354.         end = time.time();
     
355.         usetime = end - start;
     
356.         return int(1024/usetime);
     
357.    
     
358.     def testWorkNet(self):
     
359.         pass
     
360.    
     
361.    
     
362.     #加密数据
     
363.     def De_Code(self,data):
     
364.         if sys.version_info[0] == 2:
     
365.             pdata = urllib.urlencode(data);
     
366.         else:
     
367.             pdata = urllib.parse.urlencode(data);
     
368.             if type(pdata) == str: pdata = pdata.encode('utf-8')
     
369.         return binascii.hexlify(pdata);
     
370.    
     
371.     #解密数据
     
372.     def En_Code(self,data):
     
373.         if sys.version_info[0] == 2:
     
374.             result = urllib.unquote(binascii.unhexlify(data));
     
375.         else:
     
376.             if type(data) == str: data = data.encode('utf-8')
     
377.             tmp = binascii.unhexlify(data)
     
378.             if type(tmp) != str: tmp = tmp.decode('utf-8')
     
379.             result = urllib.parse.unquote(tmp)
     
380.         if type(result) != str: result = result.decode('utf-8')
     
381.         try:
     
382.             return json.loads(result);
     
383.         except: return result
     
384.    

复制代码

我是小盒子刷的linux。安装完宝塔，获取不到CPU型号，就会报错，。

图片传不了。。。提示帖子长度不符合要求

感谢反馈，这个问题已经记录过了
另外，面板本身不针对aarch这类机构兼容测试哦