spalencsar/bearhub
1
0
Fork 0
mirror of https://github.com/spalencsar/bearhub.git synced 2026-07-07 00:04:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

[arch.cpu_manager] refactoring: String formatting method & PEP8

Browse Source
This commit is contained in:
Vinicius Moreira
2022-04-12 10:05:12 -03:00
parent f3514b91e1
commit 1dc7ce327a
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
bauh/gems/arch/cpu_manager.py
View File

@@ -8,14 +8,14 @@ from bauh.api.paths import TEMP_DIR
from bauh.commons.system import new_root_subprocess from bauh.commons.system import new_root_subprocess
def supports_performance_mode(): def supports_performance_mode() -> bool:
return os.path.exists('/sys/devices/system/cpu/cpu0/cpufreq/scaling_governor') return os.path.exists('/sys/devices/system/cpu/cpu0/cpufreq/scaling_governor')
def current_governors() -> Dict[str, Set[int]]: def current_governors() -> Dict[str, Set[int]]:
governors = {} governors = {}
for cpu in range(multiprocessing.cpu_count()): for cpu in range(multiprocessing.cpu_count()):
with open('/sys/devices/system/cpu/cpu{}/cpufreq/scaling_governor'.format(cpu)) as f: with open(f'/sys/devices/system/cpu/cpu{cpu}/cpufreq/scaling_governor') as f:
gov = f.read().strip() gov = f.read().strip()
cpus = governors.get(gov, set()) cpus = governors.get(gov, set())
cpus.add(cpu) cpus.add(cpu)
@@ -35,22 +35,21 @@ def set_governor(governor: str, root_password: Optional[str], cpu_idxs: Optional
if os.path.exists(new_gov_file): if os.path.exists(new_gov_file):
try: try:
os.remove(new_gov_file) os.remove(new_gov_file)
except: except OSError:
traceback.print_exc() traceback.print_exc()
def _change_governor(cpu_idx: int, new_gov_file_path: str, root_password: Optional[str]): def _change_governor(cpu_idx: int, new_gov_file_path: str, root_password: Optional[str]):
try: try:
gov_file = '/sys/devices/system/cpu/cpu{}/cpufreq/scaling_governor'.format(cpu_idx) gov_file = f'/sys/devices/system/cpu/cpu{cpu_idx}/cpufreq/scaling_governor'
replace = new_root_subprocess(['cp', new_gov_file_path, gov_file], root_password=root_password) replace = new_root_subprocess(['cp', new_gov_file_path, gov_file], root_password=root_password)
replace.wait() replace.wait()
except: except Exception:
traceback.print_exc() traceback.print_exc()
def set_all_cpus_to(governor: str, root_password: Optional[str], logger: Optional[Logger] = None) -> Tuple[bool, Optional[Dict[str, Set[int]]]]: def set_all_cpus_to(governor: str, root_password: Optional[str], logger: Optional[Logger] = None) \
""" -> Tuple[bool, Optional[Dict[str, Set[int]]]]:
"""
cpus_changed, cpu_governors = False, current_governors() cpus_changed, cpu_governors = False, current_governors()
if cpu_governors: if cpu_governors:
@@ -61,7 +60,7 @@ def set_all_cpus_to(governor: str, root_password: Optional[str], logger: Optiona
if not_in_performance: if not_in_performance:
if logger: if logger:
logger.info("Changing CPUs {} governors to '{}'".format(not_in_performance, governor)) logger.info(f"Changing CPUs {not_in_performance} governors to '{governor}'")
set_governor(governor, root_password, not_in_performance) set_governor(governor, root_password, not_in_performance)
cpus_changed = True cpus_changed = True
@@ -69,10 +68,12 @@ def set_all_cpus_to(governor: str, root_password: Optional[str], logger: Optiona
return cpus_changed, cpu_governors return cpus_changed, cpu_governors
def set_cpus(governors: Dict[str, Set[int]], root_password: Optional[str], ignore_governors: Optional[Set[str]] = None, logger: Optional[Logger] = None): def set_cpus(governors: Dict[str, Set[int]], root_password: Optional[str], ignore_governors: Optional[Set[str]] = None,
logger: Optional[Logger] = None):
for gov, cpus in governors.items(): for gov, cpus in governors.items():
if not ignore_governors or gov not in ignore_governors: if not ignore_governors or gov not in ignore_governors:
if logger: if logger:
logger.info("Changing CPUs {} governors to '{}'".format(cpus, gov)) logger.info(f"Changing CPUs {cpus} governors to '{gov}'")
set_governor(gov, root_password, cpus) set_governor(gov, root_password, cpus)