administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 02:35:22 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

black format util.py

Browse Source
This commit is contained in:
Sergey Krashevich 2023-04-28 02:00:45 +03:00
parent 4dc3f7fa9c
commit 2494d46eb6
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
frigate/util.py
View File

@ -744,22 +744,25 @@ def get_cgroups_version() -> str:
"""Determine what version of cgroups is enabled.""" """Determine what version of cgroups is enabled."""
cgroup_path = "/sys/fs/cgroup" cgroup_path = "/sys/fs/cgroup"
try: try:
stat_info = os.stat(cgroup_path) stat_info = os.stat(cgroup_path)
value = os.statvfs(cgroup_path).f_type value = os.statvfs(cgroup_path).f_type
if value == os.fsencode('cgroup2fs'): if value == os.fsencode("cgroup2fs"):
return "cgroup2" return "cgroup2"
elif value == os.fsencode('tmpfs'): elif value == os.fsencode("tmpfs"):
return "cgroup" return "cgroup"
else: else:
logger.debug(f"Could not determine cgroups version: unhandled filesystem {value}") logger.debug(
f"Could not determine cgroups version: unhandled filesystem {value}"
)
except Exception as e: except Exception as e:
logger.debug(f"Could not determine cgroups version: {e}") logger.debug(f"Could not determine cgroups version: {e}")
return "unknown" return "unknown"
def get_docker_memlimit_bytes() -> int: def get_docker_memlimit_bytes() -> int:
"""Get mem limit in bytes set in docker if present. Returns -1 if no limit detected.""" """Get mem limit in bytes set in docker if present. Returns -1 if no limit detected."""
@ -768,7 +771,7 @@ def get_docker_memlimit_bytes() -> int:
memlimit_path = "/sys/fs/cgroup/memory.max" memlimit_path = "/sys/fs/cgroup/memory.max"
try: try:
with open(memlimit_path, 'r') as f: with open(memlimit_path, "r") as f:
value = f.read().strip() value = f.read().strip()
if value.isnumeric(): if value.isnumeric():
@ -780,18 +783,19 @@ def get_docker_memlimit_bytes() -> int:
return -1 return -1
def get_cpu_stats() -> dict[str, dict]: def get_cpu_stats() -> dict[str, dict]:
"""Get cpu usages for each process id.""" """Get cpu usages for each process id."""
usages = {} usages = {}
docker_memlimit = get_docker_memlimit_bytes() / 1024 docker_memlimit = get_docker_memlimit_bytes() / 1024
for pid in os.listdir('/proc'): for pid in os.listdir("/proc"):
if pid.isdigit(): if pid.isdigit():
try: try:
with open(f'/proc/{pid}/stat', 'r') as f: with open(f"/proc/{pid}/stat", "r") as f:
stat_info = f.read().split() stat_info = f.read().split()
with open('/proc/meminfo', 'r') as f: with open("/proc/meminfo", "r") as f:
mem_info = f.readlines() mem_info = f.readlines()
total_mem = int(mem_info[0].split()[1]) total_mem = int(mem_info[0].split()[1])