-
Notifications
You must be signed in to change notification settings - Fork 0
/
id.py
285 lines (242 loc) · 8.43 KB
/
id.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
# -*- coding: utf-8 -*-
"""
Created on Wed Mar 25 06:17:18 2020
@author: GFI
requires Python 3.6 or later
"""
import sys
import socket
import time
from platform import uname
from os import popen
import psutil
__version__ = '0.1.6'
# Raspi 4 shows BCM2835 instead of BCM2711
RASPIS = [
{'Raspberry Pi 4 Model B 1GB': ['BCM2835', '1500', '1024']},
{'Raspberry Pi 4 Model B 2GB': ['BCM2835', '1500', '2048']},
{'Raspberry Pi 4 Model B 4GB': ['BCM2835', '1500', '4096']},
{'Raspberry Pi 3 Model B+': ['1400', '1024']},
{'Raspberry Pi 3 Model A+': ['1400', '512']},
{'Raspberry Pi 3 Model B': ['1200', '1024']},
{'Raspberry Pi 2 Model B': ['900', '1024']},
{'Raspberry Pi Zero W': ['BCM2835', '1000', '1024']},
{'Raspberry Pi Zero': ['BCM2835', '1000', '1024']},
{'Raspberry Pi 1 Model B+': ['BCM2835', '700', ]},
{'Raspberry Pi 1 Model A+': ['BCM2835', '700', ]},
]
def raspi_model():
'''Return Raspi Model'''
global RASPIS
cpu_dict = get_cpudict()
if cpu_dict:
tot_mem = str(_get_mem())
proc = cpu_dict.get('Hardware', 'unkonwn')
cpu_freq = f'{psutil.cpu_freq().current:.0f}'
criterias = [cpu_freq, proc, tot_mem]
for raspi in RASPIS:
values = list(raspi.values())[0]
res = [v in criterias for v in values]
if all(res):
return list(raspi.keys())[0] + '\n'
return 'kein Raspberry PI\n'
def pi_camera():
'''Detect Pi Camera Modul'''
try:
with open('/boot/config.txt', 'r') as file:
config = file.readlines()
assert 'start_x' in ''.join(config)
except (AssertionError, FileNotFoundError):
return ''
else:
camtypes = {'ov5647': 'v1 Omnivision OV5647-5MP 2592 x 1944 Px',
'imx219': 'v2 Sony IMX219 8MP 3280 x 2464 Px'}
try:
import picamera
except ModuleNotFoundError:
return ''
try:
with picamera.PiCamera() as cam:
camtype = cam.revision
except picamera.exc.PiCameraError:
return ''
camtype = camtypes.get(camtype, '')
if camtype:
return f'Pi Camera Modul : {camtype}\n'
return ''
def pretty_size(nr_of_bytes) -> str:
'''return string in KB, MB or GB'''
extensions = ['GB', 'MB', 'KB', 'B']
nrbytes = int(nr_of_bytes) / 1024 ** 3
for extension in extensions:
if nrbytes >= 1:
return f'{nrbytes:.0f}{extension}'
nrbytes = nrbytes * 1024
return ''
def system_info():
'''get system info'''
pname = uname()
infotxt = (f'System : {pname.system}{pname.version}\n'
f'Machine : {pname.machine}\n'
f'Processor : {getattr(pname, "processor", "n/a")}\n'
)
return infotxt
def welcome() -> str:
'''Welcome header'''
hostname = socket.gethostname()
return f'Welcome to {hostname!r}'
def python_version() -> str:
'''Return python version information'''
version = ' '.join(sys.version.split('\n'))
version = 'Python version : %s\n' % version
vinfo = sys.version_info
if vinfo.major * 10 + vinfo.minor >= 36:
return version
raise 'Invalid python version'
def debian_version() -> str:
'''Returns Debian Version if running on linux'''
with popen('cat /etc/debian_version') as file:
line = file.readline()
if line:
return f'Platform : {sys.platform} {line.strip()}\n'
return f'Platform : {sys.platform} {sys.winver}\n'
def disk_space() -> str:
'''Return available disk space'''
dsk = psutil.disk_usage('.')
dskt = pretty_size(dsk.total)
dskf = pretty_size(dsk.free)
dskp = f'{dsk.percent:3.0f}%'
return f'Disk Space : total={dskt} used={dskp} free={dskf}\n'
def _get_mem():
'''Return RAM in megabytes'''
megabytes = 1048576
gigabytes = megabytes * 1024
with popen('vcgencmd get_mem arm') as file:
line1 = file.readline()
with popen('vcgencmd get_mem gpu') as file:
line2 = file.readline()
try:
arm_mem = int(line1[4:-2]) * megabytes
gpu_mem = int(line2[4:-2]) * megabytes
virt_mem = psutil.virtual_memory().total
if virt_mem > gigabytes:
tot_mem = int(round((virt_mem + gpu_mem) / gigabytes,
0) * gigabytes)
else:
tot_mem = arm_mem + gpu_mem
except ValueError:
tot_mem = psutil.virtual_memory().total
return int(tot_mem / megabytes)
def memory_info() -> str:
'''Return formatted total memory'''
megabytes = 1048576
return f'Total memory : {pretty_size(_get_mem() * megabytes)}\n'
def cpu_info() -> str:
'''REturn CPU information'''
cpu_log = psutil.cpu_count(logical=True)
cpu_phy = psutil.cpu_count(logical=False)
cpu_phy = '-' if cpu_phy is None else cpu_phy
cpu_freq = psutil.cpu_freq().current
infotxt = (f'# of CPUs : {cpu_phy} [{cpu_log}]\n'
f'CPU frequency : {cpu_freq:,.0f}Mhz\n'
)
return infotxt
def os_release():
'''Return os information if running on linux'''
if 'linux' not in sys.platform:
return ''
info = dict()
with popen('cat /etc/*-release') as file:
for line in file:
if '=' in line:
key, value = line.split('=')
info[key] = value.strip('"')
infotxt = (f'Pretty Name : {info["PRETTY_NAME"]}'
f'Home URL : {info["HOME_URL"]}'
f'Support URL : {info["SUPPORT_URL"]}'
)
return infotxt
def get_cpudict():
'''Return Dict with CPU Information'''
if 'linux' not in sys.platform:
return ''
infos = dict()
with popen('cat /proc/cpuinfo') as file:
for line in file:
if ':' in line:
key, value = [val.strip().strip('\t')
for val in line.split(':')]
infos[key.strip('\t')] = value
with popen('cat /etc/os-release') as file:
for line in file:
if '=' in line:
key, value = line.split('=')
infos[key] = value.strip('"')
return infos
def all_cpuinfo():
'''Return all key cpu info'''
if 'linux' not in sys.platform:
return f'{cpu_info()}'
infos = get_cpudict()
cpu_arch = infos.get("CPU architecture")
cpu_rev = infos.get("CPU revision")
info = (f'Model : {infos.get("Model")}\n'
f'Processorname : {infos.get("model name")}'
f' {cpu_arch}.{cpu_rev}\n'
f'Hardware - Revision: {infos.get("Hardware")}'
f' - {infos.get("Revision")}\n'
f'{cpu_info()}'
f'BogoMIPs/CPU : {infos.get("BogoMIPS")}\n'
f'CPU serial : {infos.get("Serial")}\n'
)
return info
def net_info():
'''return active connections information'''
conns = psutil.net_if_stats()
ifs = psutil.net_if_addrs()
info_txts = ['Active Connections', ]
for conn, snicstats in conns.items():
if snicstats.isup:
connifs = ifs[conn]
for connif in connifs:
if str(connif.family.name) == 'AF_INET':
ftext = f'- {conn[:17]:17s}: {connif.address}'
info_txts.append(ftext)
return '\n'.join(info_txts) + '\n'
def net_active():
'''True if internet available'''
try:
socket.gethostbyname('www.google.com')
return True
except Exception:
return False
def get_allinfo():
'''Return string with all computer info'''
_ = python_version()
num = 80
id_version = f' id.py {__version__} -- {time.ctime()} '.center(num, '-')
txt = (f' {welcome()} '.center(num, '-') + '\n'
f'Computer Model : {raspi_model()}'
f'{memory_info()}'
f'{disk_space()}'
f'{all_cpuinfo()}'
f'{system_info()}'
f'{os_release()}'
f'{debian_version()}'
f'{python_version()}'
f'{net_info()}'
f'{pi_camera()}'
f'{ id_version }\n'
)
return txt
def main():
'''Show data only if internet connection established'''
while not net_active():
time.sleep(1)
txt = get_allinfo()
print(txt)
with open('id.txt', 'w+') as file:
for line in txt:
file.write(line)
if __name__ == '__main__':
main()