rahix/openflexure-microscope-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Formatting and linting

Browse source
This commit is contained in:
Joel Collins 2020-11-06 17:11:17 +00:00
parent 5319beb735
commit e41656355f
2 changed files with 56 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

53
openflexure_microscope/rescue/check_system.py
View file

@ -4,17 +4,18 @@ import psutil
from .error_sources import WarningSource
def internet_on():
test_urls = (
'https://build.openflexure.org', # Bath server
'https://openflexure.org' # GitHub pages
"https://build.openflexure.org", # Bath server
"https://openflexure.org", # GitHub pages
)
for url in test_urls:
try:
urlopen(url, timeout=10)
# If any test passes, return True
return True
except URLError:
except URLError:
pass
# If no test passed, return False
return False
@ -22,32 +23,46 @@ def internet_on():
def main():
error_sources = []
# Check internet
if not internet_on():
error_sources.append(WarningSource(
"No internet connection detected. Updates may not be available."
))
error_sources.append(
WarningSource(
"No internet connection detected. Updates may not be available."
)
)
# Check memory
mem = psutil.virtual_memory()
total_gb = mem.total / 1E9
total_gb = mem.total / 1e9
if total_gb <= 1:
error_sources.append(WarningSource(
("Less than 1GB total memory available.",
"For small scans, or control from another device, this is usually fine.",
"\n More complex usage may require additional resources.")
))
error_sources.append(
WarningSource(
(
"Less than 1GB total memory available.",
"For small scans, or control from another device, this is usually fine.",
"\n More complex usage may require additional resources.",
)
)
)
# Check disks
data_partitions = [disk for disk in psutil.disk_partitions() if "rw" in disk.opts and disk.mountpoint != "/boot"]
data_partitions = [
disk
for disk in psutil.disk_partitions()
if "rw" in disk.opts and disk.mountpoint != "/boot"
]
for part in data_partitions:
usage = psutil.disk_usage(part.mountpoint)
if int(usage.percent) >= 90:
error_sources.append(WarningSource(
(f"Disk {part.device}, at {part.mountpoint} is {int(usage.percent)}% full.",
"Captures may fail to save soon.")
))
error_sources.append(
WarningSource(
(
f"Disk {part.device}, at {part.mountpoint} is {int(usage.percent)}% full.",
"Captures may fail to save soon.",
)
)
)
return error_sources