Compare commits
2 Commits
3684dca243
...
fdfbc23d0f
Author | SHA1 | Date | |
---|---|---|---|
|
fdfbc23d0f | ||
|
1d2c07ae63 |
@ -2,10 +2,10 @@ language: python
|
|||||||
dist: xenial
|
dist: xenial
|
||||||
|
|
||||||
python:
|
python:
|
||||||
|
- "3.5"
|
||||||
- "3.6"
|
- "3.6"
|
||||||
- "3.7"
|
- "3.7"
|
||||||
- "3.8"
|
- "3.8"
|
||||||
# TODO add 3.9
|
|
||||||
- "3.9-dev"
|
- "3.9-dev"
|
||||||
|
|
||||||
install: pip install tox-travis tox tox-venv
|
install: pip install tox-travis tox tox-venv
|
||||||
|
@ -1,43 +0,0 @@
|
|||||||
# Contributing to afancontrol
|
|
||||||
|
|
||||||
I started afancontrol in 2013 in an attempt to make my custom PC case quiet.
|
|
||||||
It's been working 24/7 ever since with no issues, and eventually I started using
|
|
||||||
it on my other machines as well.
|
|
||||||
|
|
||||||
I'm quite happy with how this package serves my needs, and I hope
|
|
||||||
it can be useful for someone else too.
|
|
||||||
|
|
||||||
Contributions are welcome, however, keep in mind, that:
|
|
||||||
* Complex features and large diffs would probably be rejected,
|
|
||||||
because it would make maintenance more complicated for me,
|
|
||||||
* I don't have any plans for active development and promotion
|
|
||||||
of the package.
|
|
||||||
|
|
||||||
|
|
||||||
## Dev workflow
|
|
||||||
|
|
||||||
Prepare a virtualenv:
|
|
||||||
|
|
||||||
mkvirtualenv afancontrol
|
|
||||||
make develop
|
|
||||||
|
|
||||||
I use [TDD](https://en.wikipedia.org/wiki/Test-driven_development) for development.
|
|
||||||
|
|
||||||
Run tests:
|
|
||||||
|
|
||||||
make test
|
|
||||||
|
|
||||||
Autoformat the code and imports:
|
|
||||||
|
|
||||||
make format
|
|
||||||
|
|
||||||
Run linters:
|
|
||||||
|
|
||||||
make lint
|
|
||||||
|
|
||||||
So essentially after writing a small part of code and tests I call these
|
|
||||||
three commands and fix the errors until they stop failing.
|
|
||||||
|
|
||||||
To build docs:
|
|
||||||
|
|
||||||
make docs
|
|
@ -14,14 +14,14 @@ RUN apt-get update \
|
|||||||
RUN mkdir ~/.gnupg && echo "disable-ipv6" >> ~/.gnupg/dirmngr.conf
|
RUN mkdir ~/.gnupg && echo "disable-ipv6" >> ~/.gnupg/dirmngr.conf
|
||||||
|
|
||||||
# Import the GPG key used to sign the PyPI releases of `afancontrol`:
|
# Import the GPG key used to sign the PyPI releases of `afancontrol`:
|
||||||
RUN gpg --recv-keys "AA7B5406547AF062"
|
RUN gpg --recv-keys "2D3B9C1712FF84F7"
|
||||||
|
|
||||||
COPY debian /build/afancontrol/debian
|
COPY debian /build/afancontrol/debian
|
||||||
WORKDIR /build/afancontrol/
|
WORKDIR /build/afancontrol/
|
||||||
|
|
||||||
RUN mkdir -p debian/upstream \
|
RUN mkdir -p debian/upstream \
|
||||||
&& gpg --export --export-options export-minimal --armor \
|
&& gpg --export --export-options export-minimal --armor \
|
||||||
'A18FE9F6F570D5B4E1E1853FAA7B5406547AF062' \
|
'BE3D633AB6792715ECF34D742D3B9C1712FF84F7' \
|
||||||
> debian/upstream/signing-key.asc
|
> debian/upstream/signing-key.asc
|
||||||
|
|
||||||
RUN apt-get -y build-dep .
|
RUN apt-get -y build-dep .
|
||||||
|
4
Makefile
4
Makefile
@ -1,11 +1,11 @@
|
|||||||
|
|
||||||
.PHONY: format
|
.PHONY: format
|
||||||
format:
|
format:
|
||||||
black src tests *.py && isort src tests *.py
|
black src tests *.py && isort -rc src tests *.py
|
||||||
|
|
||||||
.PHONY: lint
|
.PHONY: lint
|
||||||
lint:
|
lint:
|
||||||
flake8 src tests *.py && isort --check-only src tests *.py && black --check src tests *.py && mypy src tests
|
flake8 src tests *.py && isort --check-only -rc src tests *.py && black --check src tests *.py && mypy src tests
|
||||||
|
|
||||||
.PHONY: test
|
.PHONY: test
|
||||||
test:
|
test:
|
||||||
|
@ -17,7 +17,7 @@ afancontrol
|
|||||||
`fancontrol <https://github.com/lm-sensors/lm-sensors/blob/master/prog/pwm/fancontrol>`_
|
`fancontrol <https://github.com/lm-sensors/lm-sensors/blob/master/prog/pwm/fancontrol>`_
|
||||||
with more advanced configuration abilities.
|
with more advanced configuration abilities.
|
||||||
|
|
||||||
`afancontrol` measures temperatures from sensors, computes the required
|
`afancontrol` measures temperature from the sensors, computes the required
|
||||||
airflow and sets PWM fan speeds accordingly.
|
airflow and sets the PWM fan speeds accordingly.
|
||||||
|
|
||||||
The docs are available at `<https://afancontrol.readthedocs.io/>`_.
|
The docs are available at `<https://afancontrol.readthedocs.io/>`_.
|
||||||
|
11
debian/changelog
vendored
11
debian/changelog
vendored
@ -1,14 +1,3 @@
|
|||||||
afancontrol (3.0.0-1) unstable; urgency=medium
|
|
||||||
|
|
||||||
* Drop support for prometheus-client < 0.1.0 (debian stretch)
|
|
||||||
* Drop support for Python 3.5 (debian stretch)
|
|
||||||
* Add support for Python 3.9
|
|
||||||
* config: add `ipmi_sensors` location property
|
|
||||||
* Add dh-systemd (would automatically (re)start the systemd service upon
|
|
||||||
package (re)installation)
|
|
||||||
|
|
||||||
-- Kostya Esmukov <kostya@esmukov.ru> Sat, 10 Oct 2020 14:43:01 +0000
|
|
||||||
|
|
||||||
afancontrol (2.2.1-1) unstable; urgency=medium
|
afancontrol (2.2.1-1) unstable; urgency=medium
|
||||||
|
|
||||||
* Fix compatibility with py3.5
|
* Fix compatibility with py3.5
|
||||||
|
5
debian/control
vendored
5
debian/control
vendored
@ -4,13 +4,12 @@ Priority: optional
|
|||||||
Maintainer: Kostya Esmukov <kostya@esmukov.ru>
|
Maintainer: Kostya Esmukov <kostya@esmukov.ru>
|
||||||
Build-Depends: debhelper (>= 9),
|
Build-Depends: debhelper (>= 9),
|
||||||
dh-python,
|
dh-python,
|
||||||
debhelper (>= 9.20160709) | dh-systemd,
|
|
||||||
python3-all,
|
python3-all,
|
||||||
python3-setuptools
|
python3-setuptools
|
||||||
Build-Depends-Indep: python3-pytest,
|
Build-Depends-Indep: python3-pytest,
|
||||||
python3-requests,
|
python3-requests,
|
||||||
python3-click,
|
python3-click,
|
||||||
python3-prometheus-client (>= 0.1.0),
|
python3-prometheus-client,
|
||||||
python3-serial
|
python3-serial
|
||||||
Standards-Version: 3.9.8
|
Standards-Version: 3.9.8
|
||||||
Homepage: https://github.com/KostyaEsmukov/afancontrol
|
Homepage: https://github.com/KostyaEsmukov/afancontrol
|
||||||
@ -28,7 +27,7 @@ Depends: ${python3:Depends},
|
|||||||
lm-sensors,
|
lm-sensors,
|
||||||
python3-click,
|
python3-click,
|
||||||
python3-pkg-resources,
|
python3-pkg-resources,
|
||||||
python3-prometheus-client (>= 0.1.0),
|
python3-prometheus-client,
|
||||||
python3-serial
|
python3-serial
|
||||||
Suggests: freeipmi-tools,
|
Suggests: freeipmi-tools,
|
||||||
Description: Advanced Fan Control program (Python 3)
|
Description: Advanced Fan Control program (Python 3)
|
||||||
|
2
debian/rules
vendored
2
debian/rules
vendored
@ -9,4 +9,4 @@ export PYBUILD_TEST_PYTEST=1
|
|||||||
export PYBUILD_TEST_ARGS={dir}/tests/
|
export PYBUILD_TEST_ARGS={dir}/tests/
|
||||||
|
|
||||||
%:
|
%:
|
||||||
dh $@ --with systemd,python3 --buildsystem=pybuild
|
dh $@ --with python3 --buildsystem=pybuild
|
||||||
|
@ -220,7 +220,8 @@ There's a Dockerfile which can be used to build a Debian `.deb` package:
|
|||||||
make deb-from-pypi
|
make deb-from-pypi
|
||||||
|
|
||||||
# Install the package:
|
# Install the package:
|
||||||
sudo apt install ./dist/debian/*.deb
|
sudo dpkg -i dist/debian/*.deb
|
||||||
|
sudo apt install -f
|
||||||
|
|
||||||
Perhaps one day the package might get published to the Debian repos,
|
Perhaps one day the package might get published to the Debian repos,
|
||||||
so a simple ``apt install afancontrol`` would work. But for now, given
|
so a simple ``apt install afancontrol`` would work. But for now, given
|
||||||
|
@ -10,15 +10,10 @@ logfile = /var/log/afancontrol.log
|
|||||||
# Default: 5
|
# Default: 5
|
||||||
interval = 5
|
interval = 5
|
||||||
|
|
||||||
# Hddtemp location. Used by the `type = hdd` temperature sensors.
|
# Hddtemp location. Relevant only when there're `type = hdd` temperature sensors.
|
||||||
# Default: hddtemp
|
# Default: hddtemp
|
||||||
;hddtemp = /usr/local/bin/hddtemp
|
;hddtemp = /usr/local/bin/hddtemp
|
||||||
|
|
||||||
# `ipmi-sensors` location from the `freeipmi-tools` package.
|
|
||||||
# Used by the `type = freeipmi` fans.
|
|
||||||
# Default: ipmi-sensors
|
|
||||||
;ipmi_sensors = /usr/local/bin/ipmi-sensors
|
|
||||||
|
|
||||||
# Prometheus exporter listening hostname and TCP port.
|
# Prometheus exporter listening hostname and TCP port.
|
||||||
# Default: (empty value)
|
# Default: (empty value)
|
||||||
;exporter_listen_host = 127.0.0.1:8083
|
;exporter_listen_host = 127.0.0.1:8083
|
||||||
|
23
setup.cfg
23
setup.cfg
@ -32,12 +32,13 @@ include_trailing_comma = True
|
|||||||
force_grid_wrap = 0
|
force_grid_wrap = 0
|
||||||
combine_as_imports = True
|
combine_as_imports = True
|
||||||
line_length = 88
|
line_length = 88
|
||||||
|
not_skip = __init__.py
|
||||||
|
|
||||||
[metadata]
|
[metadata]
|
||||||
author = Kostya Esmukov
|
author = Kostya Esmukov
|
||||||
author_email = kostya@esmukov.ru
|
author_email = kostya@esmukov.ru
|
||||||
classifier =
|
classifier =
|
||||||
Development Status :: 5 - Production/Stable
|
Development Status :: 4 - Beta
|
||||||
Intended Audience :: System Administrators
|
Intended Audience :: System Administrators
|
||||||
License :: OSI Approved :: MIT License
|
License :: OSI Approved :: MIT License
|
||||||
Natural Language :: English
|
Natural Language :: English
|
||||||
@ -45,10 +46,10 @@ classifier =
|
|||||||
Programming Language :: Python
|
Programming Language :: Python
|
||||||
Programming Language :: Python :: 3
|
Programming Language :: Python :: 3
|
||||||
Programming Language :: Python :: 3 :: Only
|
Programming Language :: Python :: 3 :: Only
|
||||||
|
Programming Language :: Python :: 3.5
|
||||||
Programming Language :: Python :: 3.6
|
Programming Language :: Python :: 3.6
|
||||||
Programming Language :: Python :: 3.7
|
Programming Language :: Python :: 3.7
|
||||||
Programming Language :: Python :: 3.8
|
Programming Language :: Python :: 3.8
|
||||||
Programming Language :: Python :: 3.9
|
|
||||||
Topic :: System :: Hardware
|
Topic :: System :: Hardware
|
||||||
Topic :: System :: Monitoring
|
Topic :: System :: Monitoring
|
||||||
Topic :: System :: Systems Administration
|
Topic :: System :: Systems Administration
|
||||||
@ -76,7 +77,7 @@ install_requires =
|
|||||||
package_dir =
|
package_dir =
|
||||||
= src
|
= src
|
||||||
packages = find:
|
packages = find:
|
||||||
python_requires = >=3.6
|
python_requires = >=3.5
|
||||||
|
|
||||||
[options.entry_points]
|
[options.entry_points]
|
||||||
console_scripts =
|
console_scripts =
|
||||||
@ -86,16 +87,16 @@ console_scripts =
|
|||||||
arduino =
|
arduino =
|
||||||
pyserial>=3.0
|
pyserial>=3.0
|
||||||
metrics =
|
metrics =
|
||||||
prometheus-client>=0.1.0
|
prometheus-client
|
||||||
dev =
|
dev =
|
||||||
black==20.8b1
|
black==19.10b0; python_version>='3.6'
|
||||||
coverage==5.3
|
coverage==5.1
|
||||||
flake8==3.8.4
|
flake8==3.7.9
|
||||||
isort==5.5.4
|
isort==4.3.21
|
||||||
mypy==0.782
|
mypy==0.770
|
||||||
pytest==6.1.0
|
pytest==5.4.2
|
||||||
requests
|
requests
|
||||||
sphinx==3.2.1
|
sphinx==3.0.3
|
||||||
wheel
|
wheel
|
||||||
|
|
||||||
[options.packages.find]
|
[options.packages.find]
|
||||||
|
@ -1 +1 @@
|
|||||||
__version__ = "3.0.0"
|
__version__ = "2.2.1"
|
||||||
|
@ -5,7 +5,6 @@ import threading
|
|||||||
from timeit import default_timer
|
from timeit import default_timer
|
||||||
from typing import TYPE_CHECKING, Any, Dict, NewType, Optional
|
from typing import TYPE_CHECKING, Any, Dict, NewType, Optional
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
@ -51,22 +50,11 @@ class ArduinoConnection:
|
|||||||
lambda: _StatusProtocol(self), url=serial_url, baudrate=baudrate
|
lambda: _StatusProtocol(self), url=serial_url, baudrate=baudrate
|
||||||
)
|
)
|
||||||
self._context_manager_depth = 0
|
self._context_manager_depth = 0
|
||||||
self._status: Optional[Dict[str, Dict[str, int]]] = None
|
self._status = None # type: Optional[Dict[str, Dict[str, int]]]
|
||||||
self._status_clock: Optional[float] = None
|
self._status_clock = None # type: Optional[float]
|
||||||
self._status_lock = threading.Lock()
|
self._status_lock = threading.Lock()
|
||||||
self._status_event = threading.Event()
|
self._status_event = threading.Event()
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls, section: ConfigParserSection[ArduinoName]
|
|
||||||
) -> "ArduinoConnection":
|
|
||||||
return cls(
|
|
||||||
name=section.name,
|
|
||||||
serial_url=section["serial_url"],
|
|
||||||
baudrate=section.getint("baudrate", fallback=DEFAULT_BAUDRATE),
|
|
||||||
status_ttl=section.getint("status_ttl", fallback=DEFAULT_STATUS_TTL),
|
|
||||||
)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
if isinstance(other, type(self)):
|
if isinstance(other, type(self)):
|
||||||
return (
|
return (
|
||||||
@ -230,10 +218,10 @@ class _AutoRetriedReaderThread:
|
|||||||
def __init__(self, protocol_factory, **serial_for_url_kwargs) -> None:
|
def __init__(self, protocol_factory, **serial_for_url_kwargs) -> None:
|
||||||
self.protocol_factory = protocol_factory
|
self.protocol_factory = protocol_factory
|
||||||
self.serial_for_url_kwargs = serial_for_url_kwargs
|
self.serial_for_url_kwargs = serial_for_url_kwargs
|
||||||
self._reader_thread: Optional[ReaderThread] = None
|
self._reader_thread = None # type: Optional[ReaderThread]
|
||||||
self._transport: Optional[ReaderThread] = None
|
self._transport = None # type: Optional[ReaderThread]
|
||||||
self._watchdog_thread: Optional[threading.Thread] = None
|
self._watchdog_thread = None # type: Optional[threading.Thread]
|
||||||
self._watchdog_queue: queue.Queue[Any] = queue.Queue()
|
self._watchdog_queue = queue.Queue() # type: queue.Queue[Any]
|
||||||
|
|
||||||
def __enter__(self): # reusable
|
def __enter__(self): # reusable
|
||||||
# TODO ?? maybe clean the _watchdog_queue?
|
# TODO ?? maybe clean the _watchdog_queue?
|
||||||
|
@ -9,120 +9,171 @@ from typing import (
|
|||||||
Sequence,
|
Sequence,
|
||||||
Tuple,
|
Tuple,
|
||||||
TypeVar,
|
TypeVar,
|
||||||
|
Union,
|
||||||
)
|
)
|
||||||
|
|
||||||
import afancontrol.filters
|
from afancontrol.arduino import (
|
||||||
from afancontrol.arduino import ArduinoConnection, ArduinoName
|
DEFAULT_BAUDRATE,
|
||||||
from afancontrol.configparser import ConfigParserSection, iter_sections
|
DEFAULT_STATUS_TTL,
|
||||||
from afancontrol.exec import Programs
|
ArduinoConnection,
|
||||||
from afancontrol.filters import FilterName, TempFilter
|
ArduinoName,
|
||||||
|
ArduinoPin,
|
||||||
|
)
|
||||||
|
from afancontrol.filters import (
|
||||||
|
MovingMedianFilter,
|
||||||
|
MovingQuantileFilter,
|
||||||
|
NullFilter,
|
||||||
|
TempFilter,
|
||||||
|
)
|
||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
from afancontrol.pwmfan import FanName, ReadonlyFanName
|
from afancontrol.pwmfan import (
|
||||||
|
ArduinoFanPWMRead,
|
||||||
|
ArduinoFanPWMWrite,
|
||||||
|
ArduinoFanSpeed,
|
||||||
|
BaseFanPWMRead,
|
||||||
|
BaseFanPWMWrite,
|
||||||
|
BaseFanSpeed,
|
||||||
|
FanInputDevice,
|
||||||
|
FreeIPMIFanSpeed,
|
||||||
|
LinuxFanPWMRead,
|
||||||
|
LinuxFanPWMWrite,
|
||||||
|
LinuxFanSpeed,
|
||||||
|
PWMDevice,
|
||||||
|
PWMValue,
|
||||||
|
)
|
||||||
from afancontrol.pwmfannorm import PWMFanNorm, ReadonlyPWMFanNorm
|
from afancontrol.pwmfannorm import PWMFanNorm, ReadonlyPWMFanNorm
|
||||||
from afancontrol.temp import FilteredTemp, TempName
|
from afancontrol.temp import CommandTemp, FileTemp, HDDTemp, Temp, TempCelsius
|
||||||
|
|
||||||
DEFAULT_CONFIG = "/etc/afancontrol/afancontrol.conf"
|
DEFAULT_CONFIG = "/etc/afancontrol/afancontrol.conf"
|
||||||
DEFAULT_PIDFILE = "/run/afancontrol.pid"
|
DEFAULT_PIDFILE = "/run/afancontrol.pid"
|
||||||
|
DEFAULT_INTERVAL = 5
|
||||||
|
DEFAULT_FANS_SPEED_CHECK_INTERVAL = 3
|
||||||
|
DEFAULT_HDDTEMP = "hddtemp"
|
||||||
DEFAULT_REPORT_CMD = (
|
DEFAULT_REPORT_CMD = (
|
||||||
'printf "Subject: %s\nTo: %s\n\n%b"'
|
'printf "Subject: %s\nTo: %s\n\n%b"'
|
||||||
' "afancontrol daemon report: %REASON%" root "%MESSAGE%"'
|
' "afancontrol daemon report: %REASON%" root "%MESSAGE%"'
|
||||||
" | sendmail -t"
|
" | sendmail -t"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
DEFAULT_FAN_TYPE = "linux"
|
||||||
|
DEFAULT_PWM_LINE_START = 100
|
||||||
|
DEFAULT_PWM_LINE_END = 240
|
||||||
|
|
||||||
|
DEFAULT_NEVER_STOP = True
|
||||||
|
|
||||||
|
DEFAULT_WINDOW_SIZE = 3
|
||||||
|
|
||||||
|
FilterName = NewType("FilterName", str)
|
||||||
|
TempName = NewType("TempName", str)
|
||||||
|
FanName = NewType("FanName", str)
|
||||||
|
ReadonlyFanName = NewType("ReadonlyFanName", str)
|
||||||
|
AnyFanName = Union[FanName, ReadonlyFanName]
|
||||||
MappingName = NewType("MappingName", str)
|
MappingName = NewType("MappingName", str)
|
||||||
|
|
||||||
T = TypeVar("T")
|
T = TypeVar("T")
|
||||||
|
|
||||||
|
|
||||||
class FanSpeedModifier(NamedTuple):
|
FanSpeedModifier = NamedTuple(
|
||||||
fan: FanName
|
"FanSpeedModifier",
|
||||||
modifier: float # [0..1]
|
# fmt: off
|
||||||
|
[
|
||||||
|
("fan", FanName),
|
||||||
class FansTempsRelation(NamedTuple):
|
("modifier", float), # [0..1]
|
||||||
temps: Sequence[TempName]
|
]
|
||||||
fans: Sequence[FanSpeedModifier]
|
# fmt: on
|
||||||
|
|
||||||
|
|
||||||
class AlertCommands(NamedTuple):
|
|
||||||
enter_cmd: Optional[str]
|
|
||||||
leave_cmd: Optional[str]
|
|
||||||
|
|
||||||
|
|
||||||
class Actions(NamedTuple):
|
|
||||||
panic: AlertCommands
|
|
||||||
threshold: AlertCommands
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> "Actions":
|
|
||||||
panic = AlertCommands(
|
|
||||||
enter_cmd=section.get("panic_enter_cmd", fallback=None),
|
|
||||||
leave_cmd=section.get("panic_leave_cmd", fallback=None),
|
|
||||||
)
|
)
|
||||||
|
|
||||||
threshold = AlertCommands(
|
FansTempsRelation = NamedTuple(
|
||||||
enter_cmd=section.get("threshold_enter_cmd", fallback=None),
|
"FansTempsRelation",
|
||||||
leave_cmd=section.get("threshold_leave_cmd", fallback=None),
|
# fmt: off
|
||||||
|
[
|
||||||
|
("temps", Sequence[TempName]),
|
||||||
|
("fans", Sequence[FanSpeedModifier]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
)
|
)
|
||||||
|
|
||||||
return cls(panic=panic, threshold=threshold)
|
AlertCommands = NamedTuple(
|
||||||
|
"AlertCommands",
|
||||||
|
# fmt: off
|
||||||
class TriggerConfig(NamedTuple):
|
[
|
||||||
global_commands: Actions
|
("enter_cmd", Optional[str]),
|
||||||
temp_commands: Mapping[TempName, Actions]
|
("leave_cmd", Optional[str]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
class DaemonCLIConfig(NamedTuple):
|
|
||||||
pidfile: Optional[str]
|
|
||||||
logfile: Optional[str]
|
|
||||||
exporter_listen_host: Optional[str]
|
|
||||||
|
|
||||||
|
|
||||||
class DaemonConfig(NamedTuple):
|
|
||||||
pidfile: Optional[str]
|
|
||||||
logfile: Optional[str]
|
|
||||||
interval: int
|
|
||||||
exporter_listen_host: Optional[str]
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls, section: ConfigParserSection, daemon_cli_config: DaemonCLIConfig
|
|
||||||
) -> "DaemonConfig":
|
|
||||||
pidfile = first_not_none(
|
|
||||||
daemon_cli_config.pidfile, section.get("pidfile", fallback=DEFAULT_PIDFILE)
|
|
||||||
)
|
|
||||||
if pidfile is not None and not pidfile.strip():
|
|
||||||
pidfile = None
|
|
||||||
|
|
||||||
logfile = first_not_none(
|
|
||||||
daemon_cli_config.logfile, section.get("logfile", fallback=None)
|
|
||||||
)
|
|
||||||
|
|
||||||
interval = section.getint("interval", fallback=5)
|
|
||||||
|
|
||||||
exporter_listen_host = first_not_none(
|
|
||||||
daemon_cli_config.exporter_listen_host,
|
|
||||||
section.get("exporter_listen_host", fallback=None),
|
|
||||||
)
|
|
||||||
|
|
||||||
return cls(
|
|
||||||
pidfile=pidfile,
|
|
||||||
logfile=logfile,
|
|
||||||
interval=interval,
|
|
||||||
exporter_listen_host=exporter_listen_host,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class ParsedConfig(NamedTuple):
|
Actions = NamedTuple(
|
||||||
daemon: DaemonConfig
|
"Actions",
|
||||||
report_cmd: str
|
# fmt: off
|
||||||
triggers: TriggerConfig
|
[
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection]
|
("panic", AlertCommands),
|
||||||
fans: Mapping[FanName, PWMFanNorm]
|
("threshold", AlertCommands),
|
||||||
readonly_fans: Mapping[ReadonlyFanName, ReadonlyPWMFanNorm]
|
]
|
||||||
temps: Mapping[TempName, FilteredTemp]
|
# fmt: on
|
||||||
mappings: Mapping[MappingName, FansTempsRelation]
|
)
|
||||||
|
|
||||||
|
|
||||||
|
TriggerConfig = NamedTuple(
|
||||||
|
"TriggerConfig",
|
||||||
|
# fmt: off
|
||||||
|
[
|
||||||
|
("global_commands", Actions),
|
||||||
|
("temp_commands", Mapping[TempName, Actions]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
DaemonCLIConfig = NamedTuple(
|
||||||
|
"DaemonCLIConfig",
|
||||||
|
# fmt: off
|
||||||
|
[
|
||||||
|
("pidfile", Optional[str]),
|
||||||
|
("logfile", Optional[str]),
|
||||||
|
("exporter_listen_host", Optional[str]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
|
)
|
||||||
|
|
||||||
|
DaemonConfig = NamedTuple(
|
||||||
|
"DaemonConfig",
|
||||||
|
# fmt: off
|
||||||
|
[
|
||||||
|
("pidfile", Optional[str]),
|
||||||
|
("logfile", Optional[str]),
|
||||||
|
("interval", int),
|
||||||
|
("exporter_listen_host", Optional[str]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
|
)
|
||||||
|
|
||||||
|
FilteredTemp = NamedTuple(
|
||||||
|
"FilteredTemp",
|
||||||
|
# fmt: off
|
||||||
|
[
|
||||||
|
("temp", Temp),
|
||||||
|
("filter", TempFilter),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
|
)
|
||||||
|
|
||||||
|
ParsedConfig = NamedTuple(
|
||||||
|
"ParsedConfig",
|
||||||
|
# fmt: off
|
||||||
|
[
|
||||||
|
("daemon", DaemonConfig),
|
||||||
|
("report_cmd", str),
|
||||||
|
("triggers", TriggerConfig),
|
||||||
|
("arduino_connections", Mapping[ArduinoName, ArduinoConnection]),
|
||||||
|
("fans", Mapping[FanName, PWMFanNorm]),
|
||||||
|
("readonly_fans", Mapping[ReadonlyFanName, ReadonlyPWMFanNorm]),
|
||||||
|
("temps", Mapping[TempName, FilteredTemp]),
|
||||||
|
("mappings", Mapping[MappingName, FansTempsRelation]),
|
||||||
|
]
|
||||||
|
# fmt: on
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def parse_config(config_path: Path, daemon_cli_config: DaemonCLIConfig) -> ParsedConfig:
|
def parse_config(config_path: Path, daemon_cli_config: DaemonCLIConfig) -> ParsedConfig:
|
||||||
@ -132,13 +183,13 @@ def parse_config(config_path: Path, daemon_cli_config: DaemonCLIConfig) -> Parse
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise RuntimeError("Unable to parse %s:\n%s" % (config_path, e))
|
raise RuntimeError("Unable to parse %s:\n%s" % (config_path, e))
|
||||||
|
|
||||||
daemon, programs = _parse_daemon(config, daemon_cli_config)
|
daemon, hddtemp = _parse_daemon(config, daemon_cli_config)
|
||||||
report_cmd, global_commands = _parse_actions(config)
|
report_cmd, global_commands = _parse_actions(config)
|
||||||
arduino_connections = _parse_arduino_connections(config)
|
arduino_connections = _parse_arduino_connections(config)
|
||||||
filters = _parse_filters(config)
|
filters = _parse_filters(config)
|
||||||
temps, temp_commands = _parse_temps(config, programs, filters)
|
temps, temp_commands = _parse_temps(config, hddtemp, filters)
|
||||||
fans = _parse_fans(config, arduino_connections)
|
fans = _parse_fans(config, arduino_connections)
|
||||||
readonly_fans = _parse_readonly_fans(config, arduino_connections, programs)
|
readonly_fans = _parse_readonly_fans(config, arduino_connections)
|
||||||
_check_fans_namespace(fans, readonly_fans)
|
_check_fans_namespace(fans, readonly_fans)
|
||||||
mappings = _parse_mappings(config, fans, temps)
|
mappings = _parse_mappings(config, fans, temps)
|
||||||
|
|
||||||
@ -165,35 +216,110 @@ def first_not_none(*parts: Optional[T]) -> Optional[T]:
|
|||||||
|
|
||||||
def _parse_daemon(
|
def _parse_daemon(
|
||||||
config: configparser.ConfigParser, daemon_cli_config: DaemonCLIConfig
|
config: configparser.ConfigParser, daemon_cli_config: DaemonCLIConfig
|
||||||
) -> Tuple[DaemonConfig, Programs]:
|
) -> Tuple[DaemonConfig, str]:
|
||||||
section: ConfigParserSection[str] = ConfigParserSection(config["daemon"])
|
daemon = config["daemon"]
|
||||||
daemon_config = DaemonConfig.from_configparser(section, daemon_cli_config)
|
keys = set(daemon.keys())
|
||||||
programs = Programs.from_configparser(section)
|
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
return daemon_config, programs
|
pidfile = first_not_none(
|
||||||
|
daemon_cli_config.pidfile, daemon.get("pidfile"), DEFAULT_PIDFILE
|
||||||
|
)
|
||||||
|
if pidfile is not None and not pidfile.strip():
|
||||||
|
pidfile = None
|
||||||
|
keys.discard("pidfile")
|
||||||
|
|
||||||
|
logfile = first_not_none(daemon_cli_config.logfile, daemon.get("logfile"))
|
||||||
|
keys.discard("logfile")
|
||||||
|
|
||||||
|
interval = daemon.getint("interval", fallback=DEFAULT_INTERVAL)
|
||||||
|
keys.discard("interval")
|
||||||
|
|
||||||
|
exporter_listen_host = first_not_none(
|
||||||
|
daemon_cli_config.exporter_listen_host, daemon.get("exporter_listen_host")
|
||||||
|
)
|
||||||
|
keys.discard("exporter_listen_host")
|
||||||
|
|
||||||
|
hddtemp = daemon.get("hddtemp") or DEFAULT_HDDTEMP
|
||||||
|
keys.discard("hddtemp")
|
||||||
|
|
||||||
|
if keys:
|
||||||
|
raise RuntimeError("Unknown options in the [daemon] section: %s" % (keys,))
|
||||||
|
|
||||||
|
return (
|
||||||
|
DaemonConfig(
|
||||||
|
pidfile=pidfile,
|
||||||
|
logfile=logfile,
|
||||||
|
interval=interval,
|
||||||
|
exporter_listen_host=exporter_listen_host,
|
||||||
|
),
|
||||||
|
hddtemp,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def _parse_actions(config: configparser.ConfigParser) -> Tuple[str, Actions]:
|
def _parse_actions(config: configparser.ConfigParser) -> Tuple[str, Actions]:
|
||||||
section: ConfigParserSection[str] = ConfigParserSection(config["actions"])
|
actions = config["actions"]
|
||||||
report_cmd = section.get("report_cmd", fallback=DEFAULT_REPORT_CMD)
|
keys = set(actions.keys())
|
||||||
actions = Actions.from_configparser(section)
|
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
return report_cmd, actions
|
report_cmd = first_not_none(actions.get("report_cmd"), DEFAULT_REPORT_CMD)
|
||||||
|
assert report_cmd is not None
|
||||||
|
keys.discard("report_cmd")
|
||||||
|
|
||||||
|
panic = AlertCommands(
|
||||||
|
enter_cmd=first_not_none(actions.get("panic_enter_cmd")),
|
||||||
|
leave_cmd=first_not_none(actions.get("panic_leave_cmd")),
|
||||||
|
)
|
||||||
|
keys.discard("panic_enter_cmd")
|
||||||
|
keys.discard("panic_leave_cmd")
|
||||||
|
|
||||||
|
threshold = AlertCommands(
|
||||||
|
enter_cmd=first_not_none(actions.get("threshold_enter_cmd")),
|
||||||
|
leave_cmd=first_not_none(actions.get("threshold_leave_cmd")),
|
||||||
|
)
|
||||||
|
keys.discard("threshold_enter_cmd")
|
||||||
|
keys.discard("threshold_leave_cmd")
|
||||||
|
|
||||||
|
if keys:
|
||||||
|
raise RuntimeError("Unknown options in the [actions] section: %s" % (keys,))
|
||||||
|
|
||||||
|
return report_cmd, Actions(panic=panic, threshold=threshold)
|
||||||
|
|
||||||
|
|
||||||
def _parse_arduino_connections(
|
def _parse_arduino_connections(
|
||||||
config: configparser.ConfigParser,
|
config: configparser.ConfigParser,
|
||||||
) -> Mapping[ArduinoName, ArduinoConnection]:
|
) -> Mapping[ArduinoName, ArduinoConnection]:
|
||||||
arduino_connections: Dict[ArduinoName, ArduinoConnection] = {}
|
arduino_connections = {} # type: Dict[ArduinoName, ArduinoConnection]
|
||||||
for section in iter_sections(config, "arduino", ArduinoName):
|
for section_name in config.sections():
|
||||||
if section.name in arduino_connections:
|
section_name_parts = section_name.split(":", 1)
|
||||||
|
|
||||||
|
if section_name_parts[0].strip().lower() != "arduino":
|
||||||
|
continue
|
||||||
|
|
||||||
|
arduino_name = ArduinoName(section_name_parts[1].strip())
|
||||||
|
arduino = config[section_name]
|
||||||
|
keys = set(arduino.keys())
|
||||||
|
|
||||||
|
serial_url = arduino["serial_url"]
|
||||||
|
keys.discard("serial_url")
|
||||||
|
|
||||||
|
baudrate = arduino.getint("baudrate", fallback=DEFAULT_BAUDRATE)
|
||||||
|
keys.discard("baudrate")
|
||||||
|
status_ttl = arduino.getint("status_ttl", fallback=DEFAULT_STATUS_TTL)
|
||||||
|
keys.discard("status_ttl")
|
||||||
|
|
||||||
|
if keys:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Duplicate arduino section declaration for '%s'" % section.name
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
|
)
|
||||||
|
|
||||||
|
if arduino_name in arduino_connections:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Duplicate arduino section declaration for '%s'" % arduino_name
|
||||||
|
)
|
||||||
|
arduino_connections[arduino_name] = ArduinoConnection(
|
||||||
|
name=arduino_name,
|
||||||
|
serial_url=serial_url,
|
||||||
|
baudrate=baudrate,
|
||||||
|
status_ttl=status_ttl,
|
||||||
)
|
)
|
||||||
arduino_connections[section.name] = ArduinoConnection.from_configparser(section)
|
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
# Empty arduino_connections is ok
|
# Empty arduino_connections is ok
|
||||||
return arduino_connections
|
return arduino_connections
|
||||||
@ -202,14 +328,49 @@ def _parse_arduino_connections(
|
|||||||
def _parse_filters(
|
def _parse_filters(
|
||||||
config: configparser.ConfigParser,
|
config: configparser.ConfigParser,
|
||||||
) -> Mapping[FilterName, TempFilter]:
|
) -> Mapping[FilterName, TempFilter]:
|
||||||
filters: Dict[FilterName, TempFilter] = {}
|
filters = {} # type: Dict[FilterName, TempFilter]
|
||||||
for section in iter_sections(config, "filter", FilterName):
|
for section_name in config.sections():
|
||||||
if section.name in filters:
|
section_name_parts = section_name.split(":", 1)
|
||||||
|
|
||||||
|
if section_name_parts[0].strip().lower() != "filter":
|
||||||
|
continue
|
||||||
|
|
||||||
|
filter_name = FilterName(section_name_parts[1].strip())
|
||||||
|
filter = config[section_name]
|
||||||
|
keys = set(filter.keys())
|
||||||
|
|
||||||
|
filter_type = filter["type"]
|
||||||
|
keys.discard("type")
|
||||||
|
|
||||||
|
if filter_type == "moving_median":
|
||||||
|
window_size = filter.getint("window_size", fallback=DEFAULT_WINDOW_SIZE)
|
||||||
|
keys.discard("window_size")
|
||||||
|
|
||||||
|
f = MovingMedianFilter(window_size=window_size) # type: TempFilter
|
||||||
|
elif filter_type == "moving_quantile":
|
||||||
|
window_size = filter.getint("window_size", fallback=DEFAULT_WINDOW_SIZE)
|
||||||
|
keys.discard("window_size")
|
||||||
|
|
||||||
|
quantile = filter.getfloat("quantile")
|
||||||
|
keys.discard("quantile")
|
||||||
|
f = MovingQuantileFilter(quantile=quantile, window_size=window_size)
|
||||||
|
else:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Duplicate filter section declaration for '%s'" % section.name
|
"Unsupported filter type '%s' for filter '%s'. "
|
||||||
|
"Supported types: `moving_median`, `moving_quantile`."
|
||||||
|
% (filter_type, filter_name)
|
||||||
)
|
)
|
||||||
filters[section.name] = afancontrol.filters.from_configparser(section)
|
|
||||||
section.ensure_no_unused_keys()
|
if keys:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
|
)
|
||||||
|
|
||||||
|
if filter_name in filters:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Duplicate filter section declaration for '%s'" % filter_name
|
||||||
|
)
|
||||||
|
filters[filter_name] = f
|
||||||
|
|
||||||
# Empty filters is ok
|
# Empty filters is ok
|
||||||
return filters
|
return filters
|
||||||
@ -217,19 +378,98 @@ def _parse_filters(
|
|||||||
|
|
||||||
def _parse_temps(
|
def _parse_temps(
|
||||||
config: configparser.ConfigParser,
|
config: configparser.ConfigParser,
|
||||||
programs: Programs,
|
hddtemp: str,
|
||||||
filters: Mapping[FilterName, TempFilter],
|
filters: Mapping[FilterName, TempFilter],
|
||||||
) -> Tuple[Mapping[TempName, FilteredTemp], Mapping[TempName, Actions]]:
|
) -> Tuple[Mapping[TempName, FilteredTemp], Mapping[TempName, Actions]]:
|
||||||
temps: Dict[TempName, FilteredTemp] = {}
|
temps = {} # type: Dict[TempName, FilteredTemp]
|
||||||
temp_commands: Dict[TempName, Actions] = {}
|
temp_commands = {} # type: Dict[TempName, Actions]
|
||||||
for section in iter_sections(config, "temp", TempName):
|
for section_name in config.sections():
|
||||||
if section.name in temps:
|
section_name_parts = section_name.split(":", 1)
|
||||||
raise RuntimeError(
|
|
||||||
"Duplicate temp section declaration for '%s'" % section.name
|
if section_name_parts[0].strip().lower() != "temp":
|
||||||
|
continue
|
||||||
|
|
||||||
|
temp_name = TempName(section_name_parts[1].strip())
|
||||||
|
temp = config[section_name]
|
||||||
|
keys = set(temp.keys())
|
||||||
|
|
||||||
|
actions_panic = AlertCommands(
|
||||||
|
enter_cmd=first_not_none(temp.get("panic_enter_cmd")),
|
||||||
|
leave_cmd=first_not_none(temp.get("panic_leave_cmd")),
|
||||||
|
)
|
||||||
|
keys.discard("panic_enter_cmd")
|
||||||
|
keys.discard("panic_leave_cmd")
|
||||||
|
|
||||||
|
actions_threshold = AlertCommands(
|
||||||
|
enter_cmd=first_not_none(temp.get("threshold_enter_cmd")),
|
||||||
|
leave_cmd=first_not_none(temp.get("threshold_leave_cmd")),
|
||||||
|
)
|
||||||
|
keys.discard("threshold_enter_cmd")
|
||||||
|
keys.discard("threshold_leave_cmd")
|
||||||
|
|
||||||
|
panic = TempCelsius(temp.getfloat("panic"))
|
||||||
|
threshold = TempCelsius(temp.getfloat("threshold"))
|
||||||
|
min = TempCelsius(temp.getfloat("min"))
|
||||||
|
max = TempCelsius(temp.getfloat("max"))
|
||||||
|
keys.discard("panic")
|
||||||
|
keys.discard("threshold")
|
||||||
|
keys.discard("min")
|
||||||
|
keys.discard("max")
|
||||||
|
|
||||||
|
type = temp["type"]
|
||||||
|
keys.discard("type")
|
||||||
|
|
||||||
|
if type == "file":
|
||||||
|
t = FileTemp(
|
||||||
|
temp["path"], min=min, max=max, panic=panic, threshold=threshold
|
||||||
|
) # type: Temp
|
||||||
|
keys.discard("path")
|
||||||
|
elif type == "hdd":
|
||||||
|
if min is None or max is None:
|
||||||
|
raise RuntimeError(
|
||||||
|
"hdd temp '%s' doesn't define the mandatory `min` and `max` temps"
|
||||||
|
% temp_name
|
||||||
|
)
|
||||||
|
t = HDDTemp(
|
||||||
|
temp["path"],
|
||||||
|
min=min,
|
||||||
|
max=max,
|
||||||
|
panic=panic,
|
||||||
|
threshold=threshold,
|
||||||
|
hddtemp_bin=hddtemp,
|
||||||
|
)
|
||||||
|
keys.discard("path")
|
||||||
|
elif type == "exec":
|
||||||
|
t = CommandTemp(
|
||||||
|
temp["command"], min=min, max=max, panic=panic, threshold=threshold
|
||||||
|
)
|
||||||
|
keys.discard("command")
|
||||||
|
else:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Unsupported temp type '%s' for temp '%s'" % (type, temp_name)
|
||||||
|
)
|
||||||
|
|
||||||
|
filter_name = temp.get("filter")
|
||||||
|
keys.discard("filter")
|
||||||
|
|
||||||
|
if filter_name is None:
|
||||||
|
filter = NullFilter()
|
||||||
|
else:
|
||||||
|
filter = filters[FilterName(filter_name.strip())].copy()
|
||||||
|
|
||||||
|
if keys:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
|
)
|
||||||
|
|
||||||
|
if temp_name in temps:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Duplicate temp section declaration for '%s'" % temp_name
|
||||||
|
)
|
||||||
|
temps[temp_name] = FilteredTemp(temp=t, filter=filter)
|
||||||
|
temp_commands[temp_name] = Actions(
|
||||||
|
panic=actions_panic, threshold=actions_threshold
|
||||||
)
|
)
|
||||||
temps[section.name] = FilteredTemp.from_configparser(section, filters, programs)
|
|
||||||
temp_commands[section.name] = Actions.from_configparser(section)
|
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
return temps, temp_commands
|
return temps, temp_commands
|
||||||
|
|
||||||
@ -238,14 +478,95 @@ def _parse_fans(
|
|||||||
config: configparser.ConfigParser,
|
config: configparser.ConfigParser,
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
||||||
) -> Mapping[FanName, PWMFanNorm]:
|
) -> Mapping[FanName, PWMFanNorm]:
|
||||||
fans: Dict[FanName, PWMFanNorm] = {}
|
fans = {} # type: Dict[FanName, PWMFanNorm]
|
||||||
for section in iter_sections(config, "fan", FanName):
|
for section_name in config.sections():
|
||||||
if section.name in fans:
|
section_name_parts = section_name.split(":", 1)
|
||||||
raise RuntimeError(
|
|
||||||
"Duplicate fan section declaration for '%s'" % section.name
|
if section_name_parts[0].strip().lower() != "fan":
|
||||||
|
continue
|
||||||
|
|
||||||
|
fan_name = FanName(section_name_parts[1].strip())
|
||||||
|
fan = config[section_name]
|
||||||
|
keys = set(fan.keys())
|
||||||
|
|
||||||
|
fan_type = fan.get("type", fallback=DEFAULT_FAN_TYPE)
|
||||||
|
keys.discard("type")
|
||||||
|
|
||||||
|
if fan_type == "linux":
|
||||||
|
pwm = PWMDevice(fan["pwm"])
|
||||||
|
fan_input = FanInputDevice(fan["fan_input"])
|
||||||
|
keys.discard("pwm")
|
||||||
|
keys.discard("fan_input")
|
||||||
|
|
||||||
|
fan_speed = LinuxFanSpeed(fan_input) # type: BaseFanSpeed
|
||||||
|
pwm_read = LinuxFanPWMRead(pwm) # type: BaseFanPWMRead
|
||||||
|
pwm_write = LinuxFanPWMWrite(pwm) # type: BaseFanPWMWrite
|
||||||
|
elif fan_type == "arduino":
|
||||||
|
arduino_name = ArduinoName(fan["arduino_name"])
|
||||||
|
keys.discard("arduino_name")
|
||||||
|
pwm_pin = ArduinoPin(fan.getint("pwm_pin"))
|
||||||
|
keys.discard("pwm_pin")
|
||||||
|
tacho_pin = ArduinoPin(fan.getint("tacho_pin"))
|
||||||
|
keys.discard("tacho_pin")
|
||||||
|
|
||||||
|
if arduino_name not in arduino_connections:
|
||||||
|
raise ValueError("[arduino:%s] section is missing" % arduino_name)
|
||||||
|
|
||||||
|
fan_speed = ArduinoFanSpeed(
|
||||||
|
arduino_connections[arduino_name], tacho_pin=tacho_pin
|
||||||
|
)
|
||||||
|
pwm_read = ArduinoFanPWMRead(
|
||||||
|
arduino_connections[arduino_name], pwm_pin=pwm_pin
|
||||||
|
)
|
||||||
|
pwm_write = ArduinoFanPWMWrite(
|
||||||
|
arduino_connections[arduino_name], pwm_pin=pwm_pin
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
raise ValueError(
|
||||||
|
"Unsupported FAN type %s. Supported ones are "
|
||||||
|
"`linux` and `arduino`." % fan_type
|
||||||
|
)
|
||||||
|
|
||||||
|
never_stop = fan.getboolean("never_stop", fallback=DEFAULT_NEVER_STOP)
|
||||||
|
keys.discard("never_stop")
|
||||||
|
|
||||||
|
pwm_line_start = PWMValue(
|
||||||
|
fan.getint("pwm_line_start", fallback=DEFAULT_PWM_LINE_START)
|
||||||
|
)
|
||||||
|
keys.discard("pwm_line_start")
|
||||||
|
|
||||||
|
pwm_line_end = PWMValue(
|
||||||
|
fan.getint("pwm_line_end", fallback=DEFAULT_PWM_LINE_END)
|
||||||
|
)
|
||||||
|
keys.discard("pwm_line_end")
|
||||||
|
|
||||||
|
for pwm_value in (pwm_line_start, pwm_line_end):
|
||||||
|
if not (pwm_read.min_pwm <= pwm_value <= pwm_read.max_pwm):
|
||||||
|
raise RuntimeError(
|
||||||
|
"Incorrect PWM value '%s' for fan '%s': it must be within [%s;%s]"
|
||||||
|
% (pwm_value, fan_name, pwm_read.min_pwm, pwm_read.max_pwm)
|
||||||
|
)
|
||||||
|
if pwm_line_start >= pwm_line_end:
|
||||||
|
raise RuntimeError(
|
||||||
|
"`pwm_line_start` PWM value must be less than `pwm_line_end` for fan '%s'"
|
||||||
|
% (fan_name,)
|
||||||
|
)
|
||||||
|
|
||||||
|
if keys:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
|
)
|
||||||
|
|
||||||
|
if fan_name in fans:
|
||||||
|
raise RuntimeError("Duplicate fan section declaration for '%s'" % fan_name)
|
||||||
|
fans[fan_name] = PWMFanNorm(
|
||||||
|
fan_speed,
|
||||||
|
pwm_read,
|
||||||
|
pwm_write,
|
||||||
|
pwm_line_start=pwm_line_start,
|
||||||
|
pwm_line_end=pwm_line_end,
|
||||||
|
never_stop=never_stop,
|
||||||
)
|
)
|
||||||
fans[section.name] = PWMFanNorm.from_configparser(section, arduino_connections)
|
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
return fans
|
return fans
|
||||||
|
|
||||||
@ -253,18 +574,75 @@ def _parse_fans(
|
|||||||
def _parse_readonly_fans(
|
def _parse_readonly_fans(
|
||||||
config: configparser.ConfigParser,
|
config: configparser.ConfigParser,
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
||||||
programs: Programs,
|
|
||||||
) -> Mapping[ReadonlyFanName, ReadonlyPWMFanNorm]:
|
) -> Mapping[ReadonlyFanName, ReadonlyPWMFanNorm]:
|
||||||
readonly_fans: Dict[ReadonlyFanName, ReadonlyPWMFanNorm] = {}
|
readonly_fans = {} # type: Dict[ReadonlyFanName, ReadonlyPWMFanNorm]
|
||||||
for section in iter_sections(config, "readonly_fan", ReadonlyFanName):
|
for section_name in config.sections():
|
||||||
if section.name in readonly_fans:
|
section_name_parts = section_name.split(":", 1)
|
||||||
|
|
||||||
|
if section_name_parts[0].strip().lower() != "readonly_fan":
|
||||||
|
continue
|
||||||
|
|
||||||
|
fan_name = ReadonlyFanName(section_name_parts[1].strip())
|
||||||
|
fan = config[section_name]
|
||||||
|
keys = set(fan.keys())
|
||||||
|
|
||||||
|
fan_type = fan.get("type", fallback=DEFAULT_FAN_TYPE)
|
||||||
|
keys.discard("type")
|
||||||
|
|
||||||
|
if fan_type == "linux":
|
||||||
|
fan_input = FanInputDevice(fan["fan_input"])
|
||||||
|
keys.discard("fan_input")
|
||||||
|
fan_speed = LinuxFanSpeed(fan_input) # type: BaseFanSpeed
|
||||||
|
pwm_read = None # type: Optional[BaseFanPWMRead]
|
||||||
|
if "pwm" in fan:
|
||||||
|
pwm = PWMDevice(fan["pwm"])
|
||||||
|
keys.discard("pwm")
|
||||||
|
pwm_read = LinuxFanPWMRead(pwm)
|
||||||
|
elif fan_type == "arduino":
|
||||||
|
arduino_name = ArduinoName(fan["arduino_name"])
|
||||||
|
keys.discard("arduino_name")
|
||||||
|
tacho_pin = ArduinoPin(fan.getint("tacho_pin"))
|
||||||
|
keys.discard("tacho_pin")
|
||||||
|
|
||||||
|
if arduino_name not in arduino_connections:
|
||||||
|
raise ValueError("[arduino:%s] section is missing" % arduino_name)
|
||||||
|
|
||||||
|
fan_speed = ArduinoFanSpeed(
|
||||||
|
arduino_connections[arduino_name], tacho_pin=tacho_pin
|
||||||
|
)
|
||||||
|
pwm_read = None
|
||||||
|
if "pwm_pin" in fan:
|
||||||
|
pwm_pin = ArduinoPin(fan.getint("pwm_pin"))
|
||||||
|
keys.discard("pwm_pin")
|
||||||
|
pwm_read = ArduinoFanPWMRead(
|
||||||
|
arduino_connections[arduino_name], pwm_pin=pwm_pin
|
||||||
|
)
|
||||||
|
elif fan_type == "freeipmi":
|
||||||
|
name = fan["name"]
|
||||||
|
keys.discard("name")
|
||||||
|
|
||||||
|
ipmi_sensors_extra_args = fan.get("ipmi_sensors_extra_args", fallback="")
|
||||||
|
|
||||||
|
fan_speed = FreeIPMIFanSpeed(
|
||||||
|
name, ipmi_sensors_extra_args=ipmi_sensors_extra_args
|
||||||
|
)
|
||||||
|
pwm_read = None
|
||||||
|
else:
|
||||||
|
raise ValueError(
|
||||||
|
"Unsupported FAN type %s. Supported ones are "
|
||||||
|
"`linux`, `arduino`, `freeipmi`." % fan_type
|
||||||
|
)
|
||||||
|
|
||||||
|
if keys:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Duplicate readonly_fan section declaration for '%s'" % section.name
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
)
|
)
|
||||||
readonly_fans[section.name] = ReadonlyPWMFanNorm.from_configparser(
|
|
||||||
section, arduino_connections, programs
|
if fan_name in readonly_fans:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Duplicate readonly_fan section declaration for '%s'" % fan_name
|
||||||
)
|
)
|
||||||
section.ensure_no_unused_keys()
|
readonly_fans[fan_name] = ReadonlyPWMFanNorm(fan_speed, pwm_read)
|
||||||
|
|
||||||
return readonly_fans
|
return readonly_fans
|
||||||
|
|
||||||
@ -287,35 +665,45 @@ def _parse_mappings(
|
|||||||
temps: Mapping[TempName, FilteredTemp],
|
temps: Mapping[TempName, FilteredTemp],
|
||||||
) -> Mapping[MappingName, FansTempsRelation]:
|
) -> Mapping[MappingName, FansTempsRelation]:
|
||||||
|
|
||||||
mappings: Dict[MappingName, FansTempsRelation] = {}
|
mappings = {} # type: Dict[MappingName, FansTempsRelation]
|
||||||
for section in iter_sections(config, "mapping", MappingName):
|
for section_name in config.sections():
|
||||||
|
section_name_parts = section_name.split(":", 1)
|
||||||
|
|
||||||
|
if section_name_parts[0].lower() != "mapping":
|
||||||
|
continue
|
||||||
|
|
||||||
|
mapping_name = MappingName(section_name_parts[1])
|
||||||
|
mapping = config[section_name]
|
||||||
|
keys = set(mapping.keys())
|
||||||
|
|
||||||
# temps:
|
# temps:
|
||||||
|
|
||||||
mapping_temps = [
|
mapping_temps = [
|
||||||
TempName(temp_name.strip()) for temp_name in section["temps"].split(",")
|
TempName(temp_name.strip()) for temp_name in mapping["temps"].split(",")
|
||||||
]
|
]
|
||||||
mapping_temps = [s for s in mapping_temps if s]
|
mapping_temps = [s for s in mapping_temps if s]
|
||||||
|
keys.discard("temps")
|
||||||
if not mapping_temps:
|
if not mapping_temps:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Temps must not be empty in the '%s' mapping" % section.name
|
"Temps must not be empty in the '%s' mapping" % mapping_name
|
||||||
)
|
)
|
||||||
for temp_name in mapping_temps:
|
for temp_name in mapping_temps:
|
||||||
if temp_name not in temps:
|
if temp_name not in temps:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Unknown temp '%s' in mapping '%s'" % (temp_name, section.name)
|
"Unknown temp '%s' in mapping '%s'" % (temp_name, mapping_name)
|
||||||
)
|
)
|
||||||
if len(mapping_temps) != len(set(mapping_temps)):
|
if len(mapping_temps) != len(set(mapping_temps)):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"There are duplicate temps in mapping '%s'" % section.name
|
"There are duplicate temps in mapping '%s'" % mapping_name
|
||||||
)
|
)
|
||||||
|
|
||||||
# fans:
|
# fans:
|
||||||
|
|
||||||
fans_with_speed = [
|
fans_with_speed = [
|
||||||
fan_with_speed.strip() for fan_with_speed in section["fans"].split(",")
|
fan_with_speed.strip() for fan_with_speed in mapping["fans"].split(",")
|
||||||
]
|
]
|
||||||
fans_with_speed = [s for s in fans_with_speed if s]
|
fans_with_speed = [s for s in fans_with_speed if s]
|
||||||
|
keys.discard("fans")
|
||||||
|
|
||||||
fan_speed_pairs = [
|
fan_speed_pairs = [
|
||||||
fan_with_speed.split("*") for fan_with_speed in fans_with_speed
|
fan_with_speed.split("*") for fan_with_speed in fans_with_speed
|
||||||
@ -324,7 +712,7 @@ def _parse_mappings(
|
|||||||
if len(fan_speed_pair) not in (1, 2):
|
if len(fan_speed_pair) not in (1, 2):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Invalid fan specification '%s' in mapping '%s'"
|
"Invalid fan specification '%s' in mapping '%s'"
|
||||||
% (fan_speed_pair, section.name)
|
% (fan_speed_pair, mapping_name)
|
||||||
)
|
)
|
||||||
mapping_fans = [
|
mapping_fans = [
|
||||||
FanSpeedModifier(
|
FanSpeedModifier(
|
||||||
@ -341,7 +729,7 @@ def _parse_mappings(
|
|||||||
if fan_speed_modifier.fan not in fans:
|
if fan_speed_modifier.fan not in fans:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Unknown fan '%s' in mapping '%s'"
|
"Unknown fan '%s' in mapping '%s'"
|
||||||
% (fan_speed_modifier.fan, section.name)
|
% (fan_speed_modifier.fan, mapping_name)
|
||||||
)
|
)
|
||||||
if not (0 < fan_speed_modifier.modifier <= 1.0):
|
if not (0 < fan_speed_modifier.modifier <= 1.0):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
@ -349,7 +737,7 @@ def _parse_mappings(
|
|||||||
"the allowed range is (0.0;1.0]."
|
"the allowed range is (0.0;1.0]."
|
||||||
% (
|
% (
|
||||||
fan_speed_modifier.modifier,
|
fan_speed_modifier.modifier,
|
||||||
section.name,
|
mapping_name,
|
||||||
fan_speed_modifier.fan,
|
fan_speed_modifier.fan,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@ -357,17 +745,21 @@ def _parse_mappings(
|
|||||||
set(fan_speed_modifier.fan for fan_speed_modifier in mapping_fans)
|
set(fan_speed_modifier.fan for fan_speed_modifier in mapping_fans)
|
||||||
):
|
):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"There are duplicate fans in mapping '%s'" % section.name
|
"There are duplicate fans in mapping '%s'" % mapping_name
|
||||||
)
|
)
|
||||||
|
|
||||||
if section.name in mappings:
|
if keys:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Duplicate mapping section declaration for '%s'" % section.name
|
"Unknown options in the [%s] section: %s" % (section_name, keys)
|
||||||
)
|
)
|
||||||
mappings[section.name] = FansTempsRelation(
|
|
||||||
|
if mapping_name in fans:
|
||||||
|
raise RuntimeError(
|
||||||
|
"Duplicate mapping section declaration for '%s'" % mapping_name
|
||||||
|
)
|
||||||
|
mappings[mapping_name] = FansTempsRelation(
|
||||||
temps=mapping_temps, fans=mapping_fans
|
temps=mapping_temps, fans=mapping_fans
|
||||||
)
|
)
|
||||||
section.ensure_no_unused_keys()
|
|
||||||
|
|
||||||
unused_temps = set(temps.keys())
|
unused_temps = set(temps.keys())
|
||||||
unused_fans = set(fans.keys())
|
unused_fans = set(fans.keys())
|
||||||
|
@ -1,129 +0,0 @@
|
|||||||
import configparser
|
|
||||||
from typing import Any, Generic, Iterator, Optional, Type, TypeVar, Union, overload
|
|
||||||
|
|
||||||
T = TypeVar("T", bound=str)
|
|
||||||
F = TypeVar("F", None, Any)
|
|
||||||
|
|
||||||
_UNSET = object()
|
|
||||||
|
|
||||||
|
|
||||||
def iter_sections(
|
|
||||||
config: configparser.ConfigParser, section_type: str, name_typevar: Type[T]
|
|
||||||
) -> Iterator["ConfigParserSection[T]"]:
|
|
||||||
for section_name in config.sections():
|
|
||||||
section_name_parts = section_name.split(":", 1)
|
|
||||||
|
|
||||||
if section_name_parts[0].strip().lower() != section_type:
|
|
||||||
continue
|
|
||||||
|
|
||||||
name = name_typevar(section_name_parts[1].strip())
|
|
||||||
section = ConfigParserSection(config[section_name], name)
|
|
||||||
yield section
|
|
||||||
|
|
||||||
|
|
||||||
class ConfigParserSection(Generic[T]):
|
|
||||||
def __init__(
|
|
||||||
self, section: configparser.SectionProxy, name: Optional[T] = None
|
|
||||||
) -> None:
|
|
||||||
self.__name = name
|
|
||||||
self.__section = section
|
|
||||||
self.__unused_keys = set(section.keys())
|
|
||||||
|
|
||||||
@property
|
|
||||||
def name(self) -> T:
|
|
||||||
assert self.__name is not None
|
|
||||||
return self.__name
|
|
||||||
|
|
||||||
def ensure_no_unused_keys(self) -> None:
|
|
||||||
if self.__unused_keys:
|
|
||||||
raise RuntimeError(
|
|
||||||
"Unknown options in the [%s] section: %s"
|
|
||||||
% (self.__section.name, self.__unused_keys)
|
|
||||||
)
|
|
||||||
|
|
||||||
def __contains__(self, key):
|
|
||||||
return self.__section.__contains__(key)
|
|
||||||
|
|
||||||
def __getitem__(self, key):
|
|
||||||
self.__unused_keys.discard(key)
|
|
||||||
return self.__section.__getitem__(key)
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def get(self, option: str) -> str:
|
|
||||||
...
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def get(self, option: str, *, fallback: F) -> Union[str, F]:
|
|
||||||
...
|
|
||||||
|
|
||||||
def get(self, option: str, *, fallback=_UNSET) -> Union[str, F]:
|
|
||||||
kwargs = {}
|
|
||||||
if fallback is not _UNSET:
|
|
||||||
kwargs["fallback"] = fallback
|
|
||||||
self.__unused_keys.discard(option)
|
|
||||||
res = self.__section.get(option, **kwargs)
|
|
||||||
if res is None and fallback is _UNSET:
|
|
||||||
raise ValueError(
|
|
||||||
"[%s] %r option is expected to be set" % (self.__section.name, option)
|
|
||||||
)
|
|
||||||
return res
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getint(self, option: str) -> int:
|
|
||||||
...
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getint(self, option: str, *, fallback: F) -> Union[int, F]:
|
|
||||||
...
|
|
||||||
|
|
||||||
def getint(self, option: str, *, fallback=_UNSET) -> Union[int, F]:
|
|
||||||
kwargs = {}
|
|
||||||
if fallback is not _UNSET:
|
|
||||||
kwargs["fallback"] = fallback
|
|
||||||
self.__unused_keys.discard(option)
|
|
||||||
res = self.__section.getint(option, **kwargs)
|
|
||||||
if res is None and fallback is _UNSET:
|
|
||||||
raise ValueError(
|
|
||||||
"[%s] %r option is expected to be set" % (self.__section.name, option)
|
|
||||||
)
|
|
||||||
return res
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getfloat(self, option: str) -> float:
|
|
||||||
...
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getfloat(self, option: str, *, fallback: F) -> Union[float, F]:
|
|
||||||
...
|
|
||||||
|
|
||||||
def getfloat(self, option: str, *, fallback=_UNSET) -> Union[float, F]:
|
|
||||||
kwargs = {}
|
|
||||||
if fallback is not _UNSET:
|
|
||||||
kwargs["fallback"] = fallback
|
|
||||||
self.__unused_keys.discard(option)
|
|
||||||
res = self.__section.getfloat(option, **kwargs)
|
|
||||||
if res is None and fallback is _UNSET:
|
|
||||||
raise ValueError(
|
|
||||||
"[%s] %r option is expected to be set" % (self.__section.name, option)
|
|
||||||
)
|
|
||||||
return res
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getboolean(self, option: str) -> bool:
|
|
||||||
...
|
|
||||||
|
|
||||||
@overload
|
|
||||||
def getboolean(self, option: str, *, fallback: F) -> Union[bool, F]:
|
|
||||||
...
|
|
||||||
|
|
||||||
def getboolean(self, option: str, *, fallback=_UNSET) -> Union[bool, F]:
|
|
||||||
kwargs = {}
|
|
||||||
if fallback is not _UNSET:
|
|
||||||
kwargs["fallback"] = fallback
|
|
||||||
self.__unused_keys.discard(option)
|
|
||||||
res = self.__section.getboolean(option, **kwargs)
|
|
||||||
if res is None and fallback is _UNSET:
|
|
||||||
raise ValueError(
|
|
||||||
"[%s] %r option is expected to be set" % (self.__section.name, option)
|
|
||||||
)
|
|
||||||
return res
|
|
@ -66,7 +66,9 @@ def daemon(
|
|||||||
parsed_config = parse_config(config_path, daemon_cli_config)
|
parsed_config = parse_config(config_path, daemon_cli_config)
|
||||||
|
|
||||||
if parsed_config.daemon.exporter_listen_host:
|
if parsed_config.daemon.exporter_listen_host:
|
||||||
metrics: Metrics = PrometheusMetrics(parsed_config.daemon.exporter_listen_host)
|
metrics = PrometheusMetrics(
|
||||||
|
parsed_config.daemon.exporter_listen_host
|
||||||
|
) # type: Metrics
|
||||||
else:
|
else:
|
||||||
metrics = NullMetrics()
|
metrics = NullMetrics()
|
||||||
|
|
||||||
@ -81,7 +83,7 @@ def daemon(
|
|||||||
metrics=metrics,
|
metrics=metrics,
|
||||||
)
|
)
|
||||||
|
|
||||||
pidfile_instance: Optional[PidFile] = None
|
pidfile_instance = None # type: Optional[PidFile]
|
||||||
if parsed_config.daemon.pidfile is not None:
|
if parsed_config.daemon.pidfile is not None:
|
||||||
pidfile_instance = PidFile(parsed_config.daemon.pidfile)
|
pidfile_instance = PidFile(parsed_config.daemon.pidfile)
|
||||||
|
|
||||||
|
@ -1,22 +1,8 @@
|
|||||||
import subprocess
|
import subprocess
|
||||||
from typing import NamedTuple
|
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
|
|
||||||
|
|
||||||
class Programs(NamedTuple):
|
|
||||||
hddtemp: str
|
|
||||||
ipmi_sensors: str
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> "Programs":
|
|
||||||
return cls(
|
|
||||||
hddtemp=section.get("hddtemp", fallback="hddtemp"),
|
|
||||||
ipmi_sensors=section.get("ipmi_sensors", fallback="ipmi-sensors"),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def exec_shell_command(shell_command: str, timeout: int = 5) -> str:
|
def exec_shell_command(shell_command: str, timeout: int = 5) -> str:
|
||||||
try:
|
try:
|
||||||
p = subprocess.run(
|
p = subprocess.run(
|
||||||
|
@ -2,8 +2,8 @@ import itertools
|
|||||||
from contextlib import ExitStack
|
from contextlib import ExitStack
|
||||||
from typing import Iterator, Mapping, MutableSet, Optional, Tuple, Union, cast
|
from typing import Iterator, Mapping, MutableSet, Optional, Tuple, Union, cast
|
||||||
|
|
||||||
|
from afancontrol.config import AnyFanName, FanName, ReadonlyFanName
|
||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
from afancontrol.pwmfan import AnyFanName, FanName, ReadonlyFanName
|
|
||||||
from afancontrol.pwmfannorm import PWMFanNorm, PWMValueNorm, ReadonlyPWMFanNorm
|
from afancontrol.pwmfannorm import PWMFanNorm, PWMValueNorm, ReadonlyPWMFanNorm
|
||||||
from afancontrol.report import Report
|
from afancontrol.report import Report
|
||||||
|
|
||||||
@ -19,13 +19,13 @@ class Fans:
|
|||||||
self.fans = fans
|
self.fans = fans
|
||||||
self.readonly_fans = readonly_fans
|
self.readonly_fans = readonly_fans
|
||||||
self.report = report
|
self.report = report
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
|
|
||||||
# Set of fans marked as failing (which speed is 0)
|
# Set of fans marked as failing (which speed is 0)
|
||||||
self._failed_fans: MutableSet[AnyFanName] = set()
|
self._failed_fans = set() # type: MutableSet[AnyFanName]
|
||||||
|
|
||||||
# Set of fans that will be skipped on speed check
|
# Set of fans that will be skipped on speed check
|
||||||
self._stopped_fans: MutableSet[AnyFanName] = set()
|
self._stopped_fans = set() # type: MutableSet[AnyFanName]
|
||||||
|
|
||||||
def is_fan_failing(self, fan_name: AnyFanName) -> bool:
|
def is_fan_failing(self, fan_name: AnyFanName) -> bool:
|
||||||
return fan_name in self._failed_fans
|
return fan_name in self._failed_fans
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import abc
|
import abc
|
||||||
import sys
|
import sys
|
||||||
from time import sleep
|
from time import sleep
|
||||||
from typing import Optional
|
from typing import NamedTuple, Optional
|
||||||
|
|
||||||
import click
|
import click
|
||||||
|
|
||||||
@ -15,6 +15,9 @@ from afancontrol.pwmfan import (
|
|||||||
ArduinoFanPWMRead,
|
ArduinoFanPWMRead,
|
||||||
ArduinoFanPWMWrite,
|
ArduinoFanPWMWrite,
|
||||||
ArduinoFanSpeed,
|
ArduinoFanSpeed,
|
||||||
|
BaseFanPWMRead,
|
||||||
|
BaseFanPWMWrite,
|
||||||
|
BaseFanSpeed,
|
||||||
FanInputDevice,
|
FanInputDevice,
|
||||||
FanValue,
|
FanValue,
|
||||||
LinuxFanPWMRead,
|
LinuxFanPWMRead,
|
||||||
@ -22,7 +25,6 @@ from afancontrol.pwmfan import (
|
|||||||
LinuxFanSpeed,
|
LinuxFanSpeed,
|
||||||
PWMDevice,
|
PWMDevice,
|
||||||
PWMValue,
|
PWMValue,
|
||||||
ReadWriteFan,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# Time to wait before measuring fan speed after setting a PWM value.
|
# Time to wait before measuring fan speed after setting a PWM value.
|
||||||
@ -72,6 +74,15 @@ HELP_PWM_STEP_SIZE = (
|
|||||||
"faster."
|
"faster."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Fan = NamedTuple(
|
||||||
|
"Fan",
|
||||||
|
[
|
||||||
|
("fan_speed", BaseFanSpeed),
|
||||||
|
("pwm_read", BaseFanPWMRead),
|
||||||
|
("pwm_write", BaseFanPWMWrite),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
@click.command()
|
||||||
@click.option(
|
@click.option(
|
||||||
@ -162,6 +173,7 @@ def fantest(
|
|||||||
|
|
||||||
Before starting the test ensure that no fan control software is currently
|
Before starting the test ensure that no fan control software is currently
|
||||||
controlling the fan you're going to test.
|
controlling the fan you're going to test.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
if fan_type == "linux":
|
if fan_type == "linux":
|
||||||
@ -179,7 +191,7 @@ def fantest(
|
|||||||
|
|
||||||
assert linux_fan_pwm is not None
|
assert linux_fan_pwm is not None
|
||||||
assert linux_fan_input is not None
|
assert linux_fan_input is not None
|
||||||
fan = ReadWriteFan(
|
fan = Fan(
|
||||||
fan_speed=LinuxFanSpeed(FanInputDevice(linux_fan_input)),
|
fan_speed=LinuxFanSpeed(FanInputDevice(linux_fan_input)),
|
||||||
pwm_read=LinuxFanPWMRead(PWMDevice(linux_fan_pwm)),
|
pwm_read=LinuxFanPWMRead(PWMDevice(linux_fan_pwm)),
|
||||||
pwm_write=LinuxFanPWMWrite(PWMDevice(linux_fan_pwm)),
|
pwm_write=LinuxFanPWMWrite(PWMDevice(linux_fan_pwm)),
|
||||||
@ -218,7 +230,7 @@ def fantest(
|
|||||||
)
|
)
|
||||||
assert arduino_pwm_pin is not None
|
assert arduino_pwm_pin is not None
|
||||||
assert arduino_tacho_pin is not None
|
assert arduino_tacho_pin is not None
|
||||||
fan = ReadWriteFan(
|
fan = Fan(
|
||||||
fan_speed=ArduinoFanSpeed(
|
fan_speed=ArduinoFanSpeed(
|
||||||
arduino_connection, tacho_pin=ArduinoPin(arduino_tacho_pin)
|
arduino_connection, tacho_pin=ArduinoPin(arduino_tacho_pin)
|
||||||
),
|
),
|
||||||
@ -256,7 +268,7 @@ def fantest(
|
|||||||
|
|
||||||
|
|
||||||
def run_fantest(
|
def run_fantest(
|
||||||
fan: ReadWriteFan, pwm_step_size: PWMValue, output: "MeasurementsOutput"
|
fan: Fan, pwm_step_size: PWMValue, output: "MeasurementsOutput"
|
||||||
) -> None:
|
) -> None:
|
||||||
with fan.fan_speed, fan.pwm_read, fan.pwm_write:
|
with fan.fan_speed, fan.pwm_read, fan.pwm_write:
|
||||||
start = fan.pwm_read.min_pwm
|
start = fan.pwm_read.min_pwm
|
||||||
|
@ -1,32 +1,10 @@
|
|||||||
import abc
|
import abc
|
||||||
import collections
|
import collections
|
||||||
from typing import TYPE_CHECKING, Deque, NewType, Optional, TypeVar
|
from typing import Deque, Optional, TypeVar
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from afancontrol.temp import TempStatus
|
from afancontrol.temp import TempStatus
|
||||||
|
|
||||||
T = TypeVar("T")
|
T = TypeVar("T")
|
||||||
FilterName = NewType("FilterName", str)
|
|
||||||
|
|
||||||
|
|
||||||
def from_configparser(section: ConfigParserSection[FilterName]) -> "TempFilter":
|
|
||||||
filter_type = section["type"]
|
|
||||||
|
|
||||||
if filter_type == "moving_median":
|
|
||||||
window_size = section.getint("window_size", fallback=3)
|
|
||||||
return MovingMedianFilter(window_size=window_size)
|
|
||||||
elif filter_type == "moving_quantile":
|
|
||||||
window_size = section.getint("window_size", fallback=3)
|
|
||||||
quantile = section.getfloat("quantile")
|
|
||||||
return MovingQuantileFilter(quantile=quantile, window_size=window_size)
|
|
||||||
else:
|
|
||||||
raise RuntimeError(
|
|
||||||
"Unsupported filter type '%s' for filter '%s'. "
|
|
||||||
"Supported types: `moving_median`, `moving_quantile`."
|
|
||||||
% (filter_type, section.name)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class TempFilter(abc.ABC):
|
class TempFilter(abc.ABC):
|
||||||
@ -35,7 +13,7 @@ class TempFilter(abc.ABC):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def apply(self, status: Optional["TempStatus"]) -> Optional["TempStatus"]:
|
def apply(self, status: Optional[TempStatus]) -> Optional[TempStatus]:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def __enter__(self): # reusable
|
def __enter__(self): # reusable
|
||||||
@ -49,7 +27,7 @@ class NullFilter(TempFilter):
|
|||||||
def copy(self: T) -> T:
|
def copy(self: T) -> T:
|
||||||
return type(self)()
|
return type(self)()
|
||||||
|
|
||||||
def apply(self, status: Optional["TempStatus"]) -> Optional["TempStatus"]:
|
def apply(self, status: Optional[TempStatus]) -> Optional[TempStatus]:
|
||||||
return status
|
return status
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
@ -65,7 +43,7 @@ class NullFilter(TempFilter):
|
|||||||
return "%s()" % (type(self).__name__,)
|
return "%s()" % (type(self).__name__,)
|
||||||
|
|
||||||
|
|
||||||
def _temp_status_sorting_key(status: Optional["TempStatus"]) -> float:
|
def _temp_status_sorting_key(status: Optional[TempStatus]) -> float:
|
||||||
if status is None:
|
if status is None:
|
||||||
return float("+inf")
|
return float("+inf")
|
||||||
return status.temp
|
return status.temp
|
||||||
@ -75,14 +53,14 @@ class MovingQuantileFilter(TempFilter):
|
|||||||
def __init__(self, quantile: float, *, window_size: int) -> None:
|
def __init__(self, quantile: float, *, window_size: int) -> None:
|
||||||
self.quantile = quantile
|
self.quantile = quantile
|
||||||
self.window_size = window_size
|
self.window_size = window_size
|
||||||
self.history: Optional[Deque[Optional["TempStatus"]]] = None
|
self.history = None # type: Optional[Deque[Optional[TempStatus]]]
|
||||||
|
|
||||||
def copy(self: T) -> T:
|
def copy(self: T) -> T:
|
||||||
return type(self)( # type: ignore
|
return type(self)( # type: ignore
|
||||||
quantile=self.quantile, window_size=self.window_size # type: ignore
|
quantile=self.quantile, window_size=self.window_size # type: ignore
|
||||||
)
|
)
|
||||||
|
|
||||||
def apply(self, status: Optional["TempStatus"]) -> Optional["TempStatus"]:
|
def apply(self, status: Optional[TempStatus]) -> Optional[TempStatus]:
|
||||||
assert self.history is not None
|
assert self.history is not None
|
||||||
self.history.append(status)
|
self.history.append(status)
|
||||||
|
|
||||||
|
@ -41,7 +41,7 @@ class Manager:
|
|||||||
self.mappings = mappings
|
self.mappings = mappings
|
||||||
self.triggers = Triggers(triggers_config, report)
|
self.triggers = Triggers(triggers_config, report)
|
||||||
self.metrics = metrics
|
self.metrics = metrics
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
|
|
||||||
def __enter__(self): # reusable
|
def __enter__(self): # reusable
|
||||||
self._stack = ExitStack()
|
self._stack = ExitStack()
|
||||||
@ -88,7 +88,9 @@ class Manager:
|
|||||||
for temp_name, temp_status in temps.items()
|
for temp_name, temp_status in temps.items()
|
||||||
}
|
}
|
||||||
|
|
||||||
fan_speeds: Dict[FanName, PWMValueNorm] = defaultdict(lambda: PWMValueNorm(0.0))
|
fan_speeds = defaultdict(
|
||||||
|
lambda: PWMValueNorm(0.0)
|
||||||
|
) # type: Dict[FanName, PWMValueNorm]
|
||||||
|
|
||||||
for mapping_name, relation in self.mappings.items():
|
for mapping_name, relation in self.mappings.items():
|
||||||
mapping_speed = max(temp_speeds[temp_name] for temp_name in relation.temps)
|
mapping_speed = max(temp_speeds[temp_name] for temp_name in relation.temps)
|
||||||
|
@ -1,20 +1,23 @@
|
|||||||
import abc
|
import abc
|
||||||
import contextlib
|
import contextlib
|
||||||
import threading
|
import threading
|
||||||
from http.server import HTTPServer
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
||||||
from socketserver import ThreadingMixIn
|
from socketserver import ThreadingMixIn
|
||||||
from timeit import default_timer
|
from timeit import default_timer
|
||||||
from typing import ContextManager, Mapping, Optional, Union
|
from typing import TYPE_CHECKING, Mapping, Optional, Union
|
||||||
|
from urllib.parse import parse_qs, urlparse
|
||||||
|
|
||||||
from afancontrol.arduino import ArduinoConnection, ArduinoName
|
from afancontrol.arduino import ArduinoConnection, ArduinoName
|
||||||
from afancontrol.config import TempName
|
from afancontrol.config import AnyFanName, FanName, ReadonlyFanName, TempName
|
||||||
from afancontrol.fans import Fans
|
from afancontrol.fans import Fans
|
||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
from afancontrol.pwmfan import AnyFanName, FanName, ReadonlyFanName
|
|
||||||
from afancontrol.pwmfannorm import PWMFanNorm, ReadonlyPWMFanNorm
|
from afancontrol.pwmfannorm import PWMFanNorm, ReadonlyPWMFanNorm
|
||||||
from afancontrol.temps import ObservedTempStatus
|
from afancontrol.temps import ObservedTempStatus
|
||||||
from afancontrol.trigger import Triggers
|
from afancontrol.trigger import Triggers
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from typing import ContextManager # Added in 3.6
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import prometheus_client as prom
|
import prometheus_client as prom
|
||||||
|
|
||||||
@ -43,7 +46,7 @@ class Metrics(abc.ABC):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def measure_tick(self) -> ContextManager[None]:
|
def measure_tick(self) -> "ContextManager[None]":
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
@ -63,7 +66,7 @@ class NullMetrics(Metrics):
|
|||||||
) -> None:
|
) -> None:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def measure_tick(self) -> ContextManager[None]:
|
def measure_tick(self) -> "ContextManager[None]":
|
||||||
@contextlib.contextmanager
|
@contextlib.contextmanager
|
||||||
def null_context_manager():
|
def null_context_manager():
|
||||||
yield
|
yield
|
||||||
@ -82,7 +85,7 @@ class PrometheusMetrics(Metrics):
|
|||||||
self._listen_addr, port_str = listen_host.rsplit(":", 1)
|
self._listen_addr, port_str = listen_host.rsplit(":", 1)
|
||||||
self._listen_port = int(port_str)
|
self._listen_port = int(port_str)
|
||||||
|
|
||||||
self._http_server: Optional[HTTPServer] = None
|
self._http_server = None # type: Optional[HTTPServer]
|
||||||
|
|
||||||
self._last_metrics_collect_clock = float("nan")
|
self._last_metrics_collect_clock = float("nan")
|
||||||
|
|
||||||
@ -252,7 +255,7 @@ class PrometheusMetrics(Metrics):
|
|||||||
def _start(self):
|
def _start(self):
|
||||||
# `prometheus_client.start_http_server` which persists a server reference
|
# `prometheus_client.start_http_server` which persists a server reference
|
||||||
# so it could be stopped later.
|
# so it could be stopped later.
|
||||||
CustomMetricsHandler = prom.MetricsHandler.factory(self.registry)
|
CustomMetricsHandler = MetricsHandler.factory(self.registry)
|
||||||
httpd = _ThreadingSimpleServer(
|
httpd = _ThreadingSimpleServer(
|
||||||
(self._listen_addr, self._listen_port), CustomMetricsHandler
|
(self._listen_addr, self._listen_port), CustomMetricsHandler
|
||||||
)
|
)
|
||||||
@ -332,7 +335,7 @@ class PrometheusMetrics(Metrics):
|
|||||||
|
|
||||||
self._last_metrics_collect_clock = self._clock()
|
self._last_metrics_collect_clock = self._clock()
|
||||||
|
|
||||||
def measure_tick(self) -> ContextManager[None]:
|
def measure_tick(self) -> "ContextManager[None]":
|
||||||
return self.tick_duration.time()
|
return self.tick_duration.time()
|
||||||
|
|
||||||
def _collect_fan_metrics(
|
def _collect_fan_metrics(
|
||||||
@ -390,3 +393,40 @@ class _ThreadingSimpleServer(ThreadingMixIn, HTTPServer):
|
|||||||
# Enabling daemon threads virtually makes ``_ThreadingSimpleServer`` the
|
# Enabling daemon threads virtually makes ``_ThreadingSimpleServer`` the
|
||||||
# same as Python 3.7's ``ThreadingHTTPServer``.
|
# same as Python 3.7's ``ThreadingHTTPServer``.
|
||||||
daemon_threads = True
|
daemon_threads = True
|
||||||
|
|
||||||
|
|
||||||
|
# `MetricsHandler` of `prometheus_client==0.0.18` doesn't support exposing
|
||||||
|
# a custom registry. This is backported below:
|
||||||
|
if prometheus_available:
|
||||||
|
if hasattr(prom.MetricsHandler, "factory"):
|
||||||
|
MetricsHandler = prom.MetricsHandler
|
||||||
|
else:
|
||||||
|
from prometheus_client.exposition import generate_latest, CONTENT_TYPE_LATEST
|
||||||
|
|
||||||
|
class MetricsHandler(BaseHTTPRequestHandler): # type: ignore
|
||||||
|
# https://github.com/prometheus/client_python/blob/31f5557e2e84ca4ffa9a03abf6e3f4d0c8b8c3eb/prometheus_client/exposition.py#L141-L177 # noqa
|
||||||
|
registry = prom.REGISTRY
|
||||||
|
|
||||||
|
def do_GET(self):
|
||||||
|
registry = self.registry
|
||||||
|
params = parse_qs(urlparse(self.path).query)
|
||||||
|
if "name[]" in params:
|
||||||
|
registry = registry.restricted_registry(params["name[]"])
|
||||||
|
try:
|
||||||
|
output = generate_latest(registry)
|
||||||
|
except Exception:
|
||||||
|
self.send_error(500, "error generating metric output")
|
||||||
|
raise
|
||||||
|
self.send_response(200)
|
||||||
|
self.send_header("Content-Type", CONTENT_TYPE_LATEST)
|
||||||
|
self.end_headers()
|
||||||
|
self.wfile.write(output)
|
||||||
|
|
||||||
|
def log_message(self, format, *args):
|
||||||
|
"""Log nothing."""
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def factory(cls, registry):
|
||||||
|
cls_name = str(cls.__name__)
|
||||||
|
MyMetricsHandler = type(cls_name, (cls, object), {"registry": registry})
|
||||||
|
return MyMetricsHandler
|
||||||
|
@ -1,8 +1,3 @@
|
|||||||
from typing import Mapping, NamedTuple, NewType, Optional, Union
|
|
||||||
|
|
||||||
from afancontrol.arduino import ArduinoConnection, ArduinoName
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.exec import Programs
|
|
||||||
from afancontrol.pwmfan.arduino import (
|
from afancontrol.pwmfan.arduino import (
|
||||||
ArduinoFanPWMRead,
|
ArduinoFanPWMRead,
|
||||||
ArduinoFanPWMWrite,
|
ArduinoFanPWMWrite,
|
||||||
@ -40,92 +35,3 @@ __all__ = (
|
|||||||
"PWMDevice",
|
"PWMDevice",
|
||||||
"PWMValue",
|
"PWMValue",
|
||||||
)
|
)
|
||||||
|
|
||||||
DEFAULT_FAN_TYPE = "linux"
|
|
||||||
|
|
||||||
FanName = NewType("FanName", str)
|
|
||||||
ReadonlyFanName = NewType("ReadonlyFanName", str)
|
|
||||||
AnyFanName = Union[FanName, ReadonlyFanName]
|
|
||||||
|
|
||||||
|
|
||||||
class ReadOnlyFan(NamedTuple):
|
|
||||||
fan_speed: BaseFanSpeed
|
|
||||||
pwm_read: Optional[BaseFanPWMRead]
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection[ReadonlyFanName],
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
programs: Programs,
|
|
||||||
) -> "ReadOnlyFan":
|
|
||||||
fan_type = section.get("type", fallback=DEFAULT_FAN_TYPE)
|
|
||||||
|
|
||||||
if fan_type == "linux":
|
|
||||||
return cls(
|
|
||||||
fan_speed=LinuxFanSpeed.from_configparser(section),
|
|
||||||
pwm_read=(
|
|
||||||
LinuxFanPWMRead.from_configparser(section)
|
|
||||||
if "pwm" in section
|
|
||||||
else None
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif fan_type == "arduino":
|
|
||||||
return cls(
|
|
||||||
fan_speed=ArduinoFanSpeed.from_configparser(
|
|
||||||
section, arduino_connections
|
|
||||||
),
|
|
||||||
pwm_read=(
|
|
||||||
ArduinoFanPWMRead.from_configparser(section, arduino_connections)
|
|
||||||
if "pwm_pin" in section
|
|
||||||
else None
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif fan_type == "freeipmi":
|
|
||||||
return cls(
|
|
||||||
fan_speed=FreeIPMIFanSpeed.from_configparser(section, programs),
|
|
||||||
pwm_read=None,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
raise ValueError(
|
|
||||||
"Unsupported FAN type %s. Supported ones are "
|
|
||||||
"`linux`, `arduino`, `freeipmi`." % fan_type
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class ReadWriteFan(NamedTuple):
|
|
||||||
fan_speed: BaseFanSpeed
|
|
||||||
pwm_read: BaseFanPWMRead
|
|
||||||
pwm_write: BaseFanPWMWrite
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection[FanName],
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> "ReadWriteFan":
|
|
||||||
fan_type = section.get("type", fallback=DEFAULT_FAN_TYPE)
|
|
||||||
|
|
||||||
if fan_type == "linux":
|
|
||||||
return cls(
|
|
||||||
fan_speed=LinuxFanSpeed.from_configparser(section),
|
|
||||||
pwm_read=LinuxFanPWMRead.from_configparser(section),
|
|
||||||
pwm_write=LinuxFanPWMWrite.from_configparser(section),
|
|
||||||
)
|
|
||||||
elif fan_type == "arduino":
|
|
||||||
return cls(
|
|
||||||
fan_speed=ArduinoFanSpeed.from_configparser(
|
|
||||||
section, arduino_connections
|
|
||||||
),
|
|
||||||
pwm_read=ArduinoFanPWMRead.from_configparser(
|
|
||||||
section, arduino_connections
|
|
||||||
),
|
|
||||||
pwm_write=ArduinoFanPWMWrite.from_configparser(
|
|
||||||
section, arduino_connections
|
|
||||||
),
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
raise ValueError(
|
|
||||||
"Unsupported FAN type %s. Supported ones are "
|
|
||||||
"`linux` and `arduino`." % fan_type
|
|
||||||
)
|
|
||||||
|
@ -1,7 +1,4 @@
|
|||||||
from typing import Mapping
|
from afancontrol.arduino import ArduinoConnection, ArduinoPin
|
||||||
|
|
||||||
from afancontrol.arduino import ArduinoConnection, ArduinoName, ArduinoPin
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.pwmfan.base import (
|
from afancontrol.pwmfan.base import (
|
||||||
BaseFanPWMRead,
|
BaseFanPWMRead,
|
||||||
BaseFanPWMWrite,
|
BaseFanPWMWrite,
|
||||||
@ -11,14 +8,6 @@ from afancontrol.pwmfan.base import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def _ensure_arduino_connection(
|
|
||||||
arduino_name: ArduinoName,
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> None:
|
|
||||||
if arduino_name not in arduino_connections:
|
|
||||||
raise ValueError("[arduino:%s] section is missing" % arduino_name)
|
|
||||||
|
|
||||||
|
|
||||||
class ArduinoFanSpeed(BaseFanSpeed):
|
class ArduinoFanSpeed(BaseFanSpeed):
|
||||||
__slots__ = "_conn", "_tacho_pin"
|
__slots__ = "_conn", "_tacho_pin"
|
||||||
|
|
||||||
@ -28,17 +17,6 @@ class ArduinoFanSpeed(BaseFanSpeed):
|
|||||||
self._conn = arduino_connection
|
self._conn = arduino_connection
|
||||||
self._tacho_pin = tacho_pin
|
self._tacho_pin = tacho_pin
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection,
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> BaseFanSpeed:
|
|
||||||
arduino_name = ArduinoName(section["arduino_name"])
|
|
||||||
_ensure_arduino_connection(arduino_name, arduino_connections)
|
|
||||||
tacho_pin = ArduinoPin(section.getint("tacho_pin"))
|
|
||||||
return cls(arduino_connections[arduino_name], tacho_pin=tacho_pin)
|
|
||||||
|
|
||||||
def get_speed(self) -> FanValue:
|
def get_speed(self) -> FanValue:
|
||||||
return FanValue(self._conn.get_rpm(self._tacho_pin))
|
return FanValue(self._conn.get_rpm(self._tacho_pin))
|
||||||
|
|
||||||
@ -57,25 +35,11 @@ class ArduinoFanPWMRead(BaseFanPWMRead):
|
|||||||
min_pwm = PWMValue(0)
|
min_pwm = PWMValue(0)
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self, arduino_connection: ArduinoConnection, *, pwm_pin: ArduinoPin
|
||||||
arduino_connection: ArduinoConnection,
|
|
||||||
*,
|
|
||||||
pwm_pin: ArduinoPin,
|
|
||||||
) -> None:
|
) -> None:
|
||||||
self._conn = arduino_connection
|
self._conn = arduino_connection
|
||||||
self._pwm_pin = pwm_pin
|
self._pwm_pin = pwm_pin
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection,
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> BaseFanPWMRead:
|
|
||||||
arduino_name = ArduinoName(section["arduino_name"])
|
|
||||||
_ensure_arduino_connection(arduino_name, arduino_connections)
|
|
||||||
pwm_pin = ArduinoPin(section.getint("pwm_pin"))
|
|
||||||
return cls(arduino_connections[arduino_name], pwm_pin=pwm_pin)
|
|
||||||
|
|
||||||
def get(self) -> PWMValue:
|
def get(self) -> PWMValue:
|
||||||
return PWMValue(int(self._conn.get_pwm(self._pwm_pin)))
|
return PWMValue(int(self._conn.get_pwm(self._pwm_pin)))
|
||||||
|
|
||||||
@ -93,25 +57,11 @@ class ArduinoFanPWMWrite(BaseFanPWMWrite):
|
|||||||
read_cls = ArduinoFanPWMRead
|
read_cls = ArduinoFanPWMRead
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self, arduino_connection: ArduinoConnection, *, pwm_pin: ArduinoPin
|
||||||
arduino_connection: ArduinoConnection,
|
|
||||||
*,
|
|
||||||
pwm_pin: ArduinoPin,
|
|
||||||
) -> None:
|
) -> None:
|
||||||
self._conn = arduino_connection
|
self._conn = arduino_connection
|
||||||
self._pwm_pin = pwm_pin
|
self._pwm_pin = pwm_pin
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection,
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> BaseFanPWMWrite:
|
|
||||||
arduino_name = ArduinoName(section["arduino_name"])
|
|
||||||
_ensure_arduino_connection(arduino_name, arduino_connections)
|
|
||||||
pwm_pin = ArduinoPin(section.getint("pwm_pin"))
|
|
||||||
return cls(arduino_connections[arduino_name], pwm_pin=pwm_pin)
|
|
||||||
|
|
||||||
def _set_raw(self, pwm: PWMValue) -> None:
|
def _set_raw(self, pwm: PWMValue) -> None:
|
||||||
self._conn.set_pwm(self._pwm_pin, pwm)
|
self._conn.set_pwm(self._pwm_pin, pwm)
|
||||||
|
|
||||||
|
@ -40,8 +40,8 @@ class BaseFanSpeed(abc.ABC, _SlotsReprMixin):
|
|||||||
|
|
||||||
|
|
||||||
class BaseFanPWMRead(abc.ABC, _SlotsReprMixin):
|
class BaseFanPWMRead(abc.ABC, _SlotsReprMixin):
|
||||||
max_pwm: PWMValue
|
max_pwm = None # type: PWMValue
|
||||||
min_pwm: PWMValue
|
min_pwm = None # type: PWMValue
|
||||||
|
|
||||||
def is_stopped(self) -> bool:
|
def is_stopped(self) -> bool:
|
||||||
return type(self).is_pwm_stopped(self.get())
|
return type(self).is_pwm_stopped(self.get())
|
||||||
@ -62,7 +62,7 @@ class BaseFanPWMRead(abc.ABC, _SlotsReprMixin):
|
|||||||
|
|
||||||
|
|
||||||
class BaseFanPWMWrite(abc.ABC, _SlotsReprMixin):
|
class BaseFanPWMWrite(abc.ABC, _SlotsReprMixin):
|
||||||
read_cls: Type[BaseFanPWMRead]
|
read_cls = None # type: Type[BaseFanPWMRead]
|
||||||
|
|
||||||
def set(self, pwm: PWMValue) -> None:
|
def set(self, pwm: PWMValue) -> None:
|
||||||
if not (self.read_cls.min_pwm <= pwm <= self.read_cls.max_pwm):
|
if not (self.read_cls.min_pwm <= pwm <= self.read_cls.max_pwm):
|
||||||
|
@ -1,8 +1,7 @@
|
|||||||
import csv
|
import csv
|
||||||
import io
|
import io
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
from afancontrol.exec import exec_shell_command
|
||||||
from afancontrol.exec import Programs, exec_shell_command
|
|
||||||
from afancontrol.pwmfan.base import BaseFanSpeed, FanValue
|
from afancontrol.pwmfan.base import BaseFanSpeed, FanValue
|
||||||
|
|
||||||
# TODO maybe switch to `python3-pyghmi`? although it looks like the current version
|
# TODO maybe switch to `python3-pyghmi`? although it looks like the current version
|
||||||
@ -19,16 +18,6 @@ class FreeIPMIFanSpeed(BaseFanSpeed):
|
|||||||
self._ipmi_sensors_bin = ipmi_sensors_bin
|
self._ipmi_sensors_bin = ipmi_sensors_bin
|
||||||
self._ipmi_sensors_extra_args = ipmi_sensors_extra_args
|
self._ipmi_sensors_extra_args = ipmi_sensors_extra_args
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls, section: ConfigParserSection, programs: Programs
|
|
||||||
) -> BaseFanSpeed:
|
|
||||||
return cls(
|
|
||||||
section["name"],
|
|
||||||
ipmi_sensors_bin=programs.ipmi_sensors,
|
|
||||||
ipmi_sensors_extra_args=section.get("ipmi_sensors_extra_args", fallback=""),
|
|
||||||
)
|
|
||||||
|
|
||||||
def get_speed(self) -> FanValue:
|
def get_speed(self) -> FanValue:
|
||||||
out = self._call_ipmi_sensors()
|
out = self._call_ipmi_sensors()
|
||||||
reader = csv.DictReader(io.StringIO(out))
|
reader = csv.DictReader(io.StringIO(out))
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import NewType
|
from typing import NewType
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.pwmfan.base import (
|
from afancontrol.pwmfan.base import (
|
||||||
BaseFanPWMRead,
|
BaseFanPWMRead,
|
||||||
BaseFanPWMWrite,
|
BaseFanPWMWrite,
|
||||||
@ -20,10 +19,6 @@ class LinuxFanSpeed(BaseFanSpeed):
|
|||||||
def __init__(self, fan_input: FanInputDevice) -> None:
|
def __init__(self, fan_input: FanInputDevice) -> None:
|
||||||
self._fan_input = Path(fan_input)
|
self._fan_input = Path(fan_input)
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> BaseFanSpeed:
|
|
||||||
return cls(FanInputDevice(section["fan_input"]))
|
|
||||||
|
|
||||||
def get_speed(self) -> FanValue:
|
def get_speed(self) -> FanValue:
|
||||||
return FanValue(int(self._fan_input.read_text()))
|
return FanValue(int(self._fan_input.read_text()))
|
||||||
|
|
||||||
@ -37,10 +32,6 @@ class LinuxFanPWMRead(BaseFanPWMRead):
|
|||||||
def __init__(self, pwm: PWMDevice) -> None:
|
def __init__(self, pwm: PWMDevice) -> None:
|
||||||
self._pwm = Path(pwm)
|
self._pwm = Path(pwm)
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> BaseFanPWMRead:
|
|
||||||
return cls(PWMDevice(section["pwm"]))
|
|
||||||
|
|
||||||
def get(self) -> PWMValue:
|
def get(self) -> PWMValue:
|
||||||
return PWMValue(int(self._pwm.read_text()))
|
return PWMValue(int(self._pwm.read_text()))
|
||||||
|
|
||||||
@ -54,10 +45,6 @@ class LinuxFanPWMWrite(BaseFanPWMWrite):
|
|||||||
self._pwm = Path(pwm)
|
self._pwm = Path(pwm)
|
||||||
self._pwm_enable = Path(pwm + "_enable")
|
self._pwm_enable = Path(pwm + "_enable")
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> BaseFanPWMWrite:
|
|
||||||
return cls(PWMDevice(section["pwm"]))
|
|
||||||
|
|
||||||
def _set_raw(self, pwm: PWMValue) -> None:
|
def _set_raw(self, pwm: PWMValue) -> None:
|
||||||
self._pwm.write_text(str(int(pwm)))
|
self._pwm.write_text(str(int(pwm)))
|
||||||
|
|
||||||
|
@ -1,20 +1,13 @@
|
|||||||
import math
|
import math
|
||||||
from contextlib import ExitStack
|
from contextlib import ExitStack
|
||||||
from typing import Mapping, NewType, Optional
|
from typing import NewType, Optional
|
||||||
|
|
||||||
from afancontrol.arduino import ArduinoConnection, ArduinoName
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.exec import Programs
|
|
||||||
from afancontrol.pwmfan import (
|
from afancontrol.pwmfan import (
|
||||||
BaseFanPWMRead,
|
BaseFanPWMRead,
|
||||||
BaseFanPWMWrite,
|
BaseFanPWMWrite,
|
||||||
BaseFanSpeed,
|
BaseFanSpeed,
|
||||||
FanName,
|
|
||||||
FanValue,
|
FanValue,
|
||||||
PWMValue,
|
PWMValue,
|
||||||
ReadOnlyFan,
|
|
||||||
ReadonlyFanName,
|
|
||||||
ReadWriteFan,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
PWMValueNorm = NewType("PWMValueNorm", float) # [0..1]
|
PWMValueNorm = NewType("PWMValueNorm", float) # [0..1]
|
||||||
@ -26,19 +19,7 @@ class ReadonlyPWMFanNorm:
|
|||||||
) -> None:
|
) -> None:
|
||||||
self.fan_speed = fan_speed
|
self.fan_speed = fan_speed
|
||||||
self.pwm_read = pwm_read
|
self.pwm_read = pwm_read
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection[ReadonlyFanName],
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
programs: Programs,
|
|
||||||
) -> "ReadonlyPWMFanNorm":
|
|
||||||
readonly_fan = ReadOnlyFan.from_configparser(
|
|
||||||
section, arduino_connections, programs
|
|
||||||
)
|
|
||||||
return cls(readonly_fan.fan_speed, readonly_fan.pwm_read)
|
|
||||||
|
|
||||||
def __enter__(self):
|
def __enter__(self):
|
||||||
self._stack = ExitStack()
|
self._stack = ExitStack()
|
||||||
@ -117,48 +98,7 @@ class PWMFanNorm:
|
|||||||
"Invalid pwm_line_end. Expected: pwm_line_end <= max_pwm. "
|
"Invalid pwm_line_end. Expected: pwm_line_end <= max_pwm. "
|
||||||
"Got: %s <= %s" % (self.pwm_line_end, type(self.pwm_read).max_pwm)
|
"Got: %s <= %s" % (self.pwm_line_end, type(self.pwm_read).max_pwm)
|
||||||
)
|
)
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection[FanName],
|
|
||||||
arduino_connections: Mapping[ArduinoName, ArduinoConnection],
|
|
||||||
) -> "PWMFanNorm":
|
|
||||||
readwrite_fan = ReadWriteFan.from_configparser(section, arduino_connections)
|
|
||||||
never_stop = section.getboolean("never_stop", fallback=True)
|
|
||||||
pwm_line_start = PWMValue(section.getint("pwm_line_start", fallback=100))
|
|
||||||
pwm_line_end = PWMValue(section.getint("pwm_line_end", fallback=240))
|
|
||||||
|
|
||||||
for pwm_value in (pwm_line_start, pwm_line_end):
|
|
||||||
if not (
|
|
||||||
readwrite_fan.pwm_read.min_pwm
|
|
||||||
<= pwm_value
|
|
||||||
<= readwrite_fan.pwm_read.max_pwm
|
|
||||||
):
|
|
||||||
raise RuntimeError(
|
|
||||||
"Incorrect PWM value '%s' for fan '%s': it must be within [%s;%s]"
|
|
||||||
% (
|
|
||||||
pwm_value,
|
|
||||||
section.name,
|
|
||||||
readwrite_fan.pwm_read.min_pwm,
|
|
||||||
readwrite_fan.pwm_read.max_pwm,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
if pwm_line_start >= pwm_line_end:
|
|
||||||
raise RuntimeError(
|
|
||||||
"`pwm_line_start` PWM value must be less than `pwm_line_end` for fan '%s'"
|
|
||||||
% (section.name,)
|
|
||||||
)
|
|
||||||
|
|
||||||
return cls(
|
|
||||||
readwrite_fan.fan_speed,
|
|
||||||
readwrite_fan.pwm_read,
|
|
||||||
readwrite_fan.pwm_write,
|
|
||||||
pwm_line_start=pwm_line_start,
|
|
||||||
pwm_line_end=pwm_line_end,
|
|
||||||
never_stop=never_stop,
|
|
||||||
)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
if isinstance(other, type(self)):
|
if isinstance(other, type(self)):
|
||||||
|
@ -1,8 +1,3 @@
|
|||||||
from typing import Mapping, NamedTuple, NewType
|
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.exec import Programs
|
|
||||||
from afancontrol.filters import FilterName, NullFilter, TempFilter
|
|
||||||
from afancontrol.temp.base import Temp, TempCelsius, TempStatus
|
from afancontrol.temp.base import Temp, TempCelsius, TempStatus
|
||||||
from afancontrol.temp.command import CommandTemp
|
from afancontrol.temp.command import CommandTemp
|
||||||
from afancontrol.temp.file import FileTemp
|
from afancontrol.temp.file import FileTemp
|
||||||
@ -16,40 +11,3 @@ __all__ = (
|
|||||||
"TempCelsius",
|
"TempCelsius",
|
||||||
"TempStatus",
|
"TempStatus",
|
||||||
)
|
)
|
||||||
|
|
||||||
TempName = NewType("TempName", str)
|
|
||||||
|
|
||||||
|
|
||||||
class FilteredTemp(NamedTuple):
|
|
||||||
temp: Temp
|
|
||||||
filter: TempFilter
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls,
|
|
||||||
section: ConfigParserSection[TempName],
|
|
||||||
filters: Mapping[FilterName, TempFilter],
|
|
||||||
programs: Programs,
|
|
||||||
) -> "FilteredTemp":
|
|
||||||
|
|
||||||
type = section["type"]
|
|
||||||
|
|
||||||
if type == "file":
|
|
||||||
temp: Temp = FileTemp.from_configparser(section)
|
|
||||||
elif type == "hdd":
|
|
||||||
temp = HDDTemp.from_configparser(section, programs)
|
|
||||||
elif type == "exec":
|
|
||||||
temp = CommandTemp.from_configparser(section)
|
|
||||||
else:
|
|
||||||
raise RuntimeError(
|
|
||||||
"Unsupported temp type '%s' for temp '%s'" % (type, section.name)
|
|
||||||
)
|
|
||||||
|
|
||||||
filter_name = section.get("filter", fallback=None)
|
|
||||||
|
|
||||||
if filter_name is None:
|
|
||||||
filter: TempFilter = NullFilter()
|
|
||||||
else:
|
|
||||||
filter = filters[FilterName(filter_name.strip())].copy()
|
|
||||||
|
|
||||||
return cls(temp=temp, filter=filter)
|
|
||||||
|
@ -3,15 +3,18 @@ from typing import NamedTuple, NewType, Optional, Tuple
|
|||||||
|
|
||||||
TempCelsius = NewType("TempCelsius", float)
|
TempCelsius = NewType("TempCelsius", float)
|
||||||
|
|
||||||
|
TempStatus = NamedTuple(
|
||||||
class TempStatus(NamedTuple):
|
"TempStatus",
|
||||||
temp: TempCelsius
|
[
|
||||||
min: TempCelsius
|
("temp", TempCelsius),
|
||||||
max: TempCelsius
|
("min", TempCelsius),
|
||||||
panic: Optional[TempCelsius]
|
("max", TempCelsius),
|
||||||
threshold: Optional[TempCelsius]
|
("panic", Optional[TempCelsius]),
|
||||||
is_panic: bool
|
("threshold", Optional[TempCelsius]),
|
||||||
is_threshold: bool
|
("is_panic", bool),
|
||||||
|
("is_threshold", bool),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class Temp(abc.ABC):
|
class Temp(abc.ABC):
|
||||||
|
@ -1,6 +1,5 @@
|
|||||||
from typing import Optional, Tuple
|
from typing import Optional, Tuple
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.exec import exec_shell_command
|
from afancontrol.exec import exec_shell_command
|
||||||
from afancontrol.temp.base import Temp, TempCelsius
|
from afancontrol.temp.base import Temp, TempCelsius
|
||||||
|
|
||||||
@ -20,16 +19,6 @@ class CommandTemp(Temp):
|
|||||||
self._min = min
|
self._min = min
|
||||||
self._max = max
|
self._max = max
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> Temp:
|
|
||||||
panic = TempCelsius(section.getfloat("panic", fallback=None))
|
|
||||||
threshold = TempCelsius(section.getfloat("threshold", fallback=None))
|
|
||||||
min = TempCelsius(section.getfloat("min", fallback=None))
|
|
||||||
max = TempCelsius(section.getfloat("max", fallback=None))
|
|
||||||
return cls(
|
|
||||||
section["command"], min=min, max=max, panic=panic, threshold=threshold
|
|
||||||
)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
if isinstance(other, type(self)):
|
if isinstance(other, type(self)):
|
||||||
return (
|
return (
|
||||||
|
@ -3,7 +3,6 @@ import re
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional, Tuple
|
from typing import Optional, Tuple
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
|
||||||
from afancontrol.temp.base import Temp, TempCelsius
|
from afancontrol.temp.base import Temp, TempCelsius
|
||||||
|
|
||||||
|
|
||||||
@ -42,14 +41,6 @@ class FileTemp(Temp):
|
|||||||
self._min = min
|
self._min = min
|
||||||
self._max = max
|
self._max = max
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(cls, section: ConfigParserSection) -> Temp:
|
|
||||||
panic = TempCelsius(section.getfloat("panic", fallback=None))
|
|
||||||
threshold = TempCelsius(section.getfloat("threshold", fallback=None))
|
|
||||||
min = TempCelsius(section.getfloat("min", fallback=None))
|
|
||||||
max = TempCelsius(section.getfloat("max", fallback=None))
|
|
||||||
return cls(section["path"], min=min, max=max, panic=panic, threshold=threshold)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
if isinstance(other, type(self)):
|
if isinstance(other, type(self)):
|
||||||
return (
|
return (
|
||||||
|
@ -1,7 +1,6 @@
|
|||||||
from typing import Optional, Tuple
|
from typing import Optional, Tuple
|
||||||
|
|
||||||
from afancontrol.configparser import ConfigParserSection
|
from afancontrol.exec import exec_shell_command
|
||||||
from afancontrol.exec import Programs, exec_shell_command
|
|
||||||
from afancontrol.temp.base import Temp, TempCelsius
|
from afancontrol.temp.base import Temp, TempCelsius
|
||||||
|
|
||||||
|
|
||||||
@ -33,23 +32,6 @@ class HDDTemp(Temp):
|
|||||||
self._max = max
|
self._max = max
|
||||||
self._hddtemp_bin = hddtemp_bin
|
self._hddtemp_bin = hddtemp_bin
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_configparser(
|
|
||||||
cls, section: ConfigParserSection, programs: Programs
|
|
||||||
) -> Temp:
|
|
||||||
panic = TempCelsius(section.getfloat("panic", fallback=None))
|
|
||||||
threshold = TempCelsius(section.getfloat("threshold", fallback=None))
|
|
||||||
min = TempCelsius(section.getfloat("min"))
|
|
||||||
max = TempCelsius(section.getfloat("max"))
|
|
||||||
return cls(
|
|
||||||
section["path"],
|
|
||||||
min=min,
|
|
||||||
max=max,
|
|
||||||
panic=panic,
|
|
||||||
threshold=threshold,
|
|
||||||
hddtemp_bin=programs.hddtemp,
|
|
||||||
)
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
if isinstance(other, type(self)):
|
if isinstance(other, type(self)):
|
||||||
return (
|
return (
|
||||||
|
@ -7,10 +7,10 @@ from afancontrol.filters import TempFilter
|
|||||||
from afancontrol.logger import logger
|
from afancontrol.logger import logger
|
||||||
from afancontrol.temp import Temp, TempStatus
|
from afancontrol.temp import Temp, TempStatus
|
||||||
|
|
||||||
|
ObservedTempStatus = NamedTuple(
|
||||||
class ObservedTempStatus(NamedTuple):
|
"ObservedTempStatus",
|
||||||
raw: Optional[TempStatus]
|
[("raw", Optional[TempStatus]), ("filtered", Optional[TempStatus])],
|
||||||
filtered: Optional[TempStatus]
|
)
|
||||||
|
|
||||||
|
|
||||||
def filtered_temps(
|
def filtered_temps(
|
||||||
@ -25,8 +25,8 @@ def filtered_temps(
|
|||||||
class Temps:
|
class Temps:
|
||||||
def __init__(self, temps: Mapping[TempName, FilteredTemp]) -> None:
|
def __init__(self, temps: Mapping[TempName, FilteredTemp]) -> None:
|
||||||
self.temps = temps
|
self.temps = temps
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
self._executor: Optional[concurrent.futures.Executor] = None
|
self._executor = None # type: Optional[concurrent.futures.Executor]
|
||||||
|
|
||||||
def __enter__(self): # reusable
|
def __enter__(self): # reusable
|
||||||
self._stack = ExitStack()
|
self._stack = ExitStack()
|
||||||
@ -64,7 +64,7 @@ def _get_temp_status(
|
|||||||
name: TempName, temp: Temp, filter: TempFilter
|
name: TempName, temp: Temp, filter: TempFilter
|
||||||
) -> ObservedTempStatus:
|
) -> ObservedTempStatus:
|
||||||
try:
|
try:
|
||||||
sensor_value: Optional[TempStatus] = temp.get()
|
sensor_value = temp.get() # type: Optional[TempStatus]
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
sensor_value = None
|
sensor_value = None
|
||||||
logger.warning("Temp sensor [%s] has failed: %s", name, e, exc_info=True)
|
logger.warning("Temp sensor [%s] has failed: %s", name, e, exc_info=True)
|
||||||
|
@ -20,7 +20,7 @@ class Trigger(abc.ABC):
|
|||||||
self.global_commands = global_commands
|
self.global_commands = global_commands
|
||||||
self.temp_commands = temp_commands
|
self.temp_commands = temp_commands
|
||||||
self.report = report
|
self.report = report
|
||||||
self._alerting_temps: Set[TempName] = set()
|
self._alerting_temps = set() # type: Set[TempName]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
@ -184,7 +184,7 @@ class Triggers:
|
|||||||
},
|
},
|
||||||
report=report,
|
report=report,
|
||||||
)
|
)
|
||||||
self._stack: Optional[ExitStack] = None
|
self._stack = None # type: Optional[ExitStack]
|
||||||
|
|
||||||
def __enter__(self): # reusable
|
def __enter__(self): # reusable
|
||||||
self._stack = ExitStack()
|
self._stack = ExitStack()
|
||||||
|
@ -430,69 +430,3 @@ fan_input = /sys/class/hwmon/hwmon0/device/fan1_input
|
|||||||
},
|
},
|
||||||
mappings={},
|
mappings={},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_multiline_mapping():
|
|
||||||
daemon_cli_config = DaemonCLIConfig(
|
|
||||||
pidfile=None, logfile=None, exporter_listen_host=None
|
|
||||||
)
|
|
||||||
|
|
||||||
config = """
|
|
||||||
[daemon]
|
|
||||||
|
|
||||||
[actions]
|
|
||||||
|
|
||||||
[temp:cpu]
|
|
||||||
type = file
|
|
||||||
path = /sys/class/hwmon/hwmon0/device/temp1_input
|
|
||||||
|
|
||||||
[temp:mobo]
|
|
||||||
type = file
|
|
||||||
path = /sys/class/hwmon/hwmon0/device/temp2_input
|
|
||||||
|
|
||||||
[fan: case]
|
|
||||||
pwm = /sys/class/hwmon/hwmon0/device/pwm2
|
|
||||||
fan_input = /sys/class/hwmon/hwmon0/device/fan2_input
|
|
||||||
|
|
||||||
[fan: hdd]
|
|
||||||
pwm = /sys/class/hwmon/hwmon0/device/pwm2
|
|
||||||
fan_input = /sys/class/hwmon/hwmon0/device/fan2_input
|
|
||||||
|
|
||||||
[mapping:1]
|
|
||||||
fans =
|
|
||||||
case*0.6,
|
|
||||||
hdd,
|
|
||||||
temps =
|
|
||||||
mobo,
|
|
||||||
cpu
|
|
||||||
"""
|
|
||||||
parsed = parse_config(path_from_str(config), daemon_cli_config)
|
|
||||||
assert parsed.mappings == {
|
|
||||||
MappingName("1"): FansTempsRelation(
|
|
||||||
temps=[TempName("mobo"), TempName("cpu")],
|
|
||||||
fans=[
|
|
||||||
FanSpeedModifier(fan=FanName("case"), modifier=0.6),
|
|
||||||
FanSpeedModifier(fan=FanName("hdd"), modifier=1.0),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def test_extraneous_keys_raises():
|
|
||||||
daemon_cli_config = DaemonCLIConfig(
|
|
||||||
pidfile=None, logfile=None, exporter_listen_host=None
|
|
||||||
)
|
|
||||||
|
|
||||||
config = """
|
|
||||||
[daemon]
|
|
||||||
|
|
||||||
[actions]
|
|
||||||
|
|
||||||
[temp: mobo]
|
|
||||||
type = file
|
|
||||||
path = /sys/class/hwmon/hwmon0/device/temp1_input
|
|
||||||
aa = 55
|
|
||||||
"""
|
|
||||||
with pytest.raises(RuntimeError) as cm:
|
|
||||||
parse_config(path_from_str(config), daemon_cli_config)
|
|
||||||
assert str(cm.value) == "Unknown options in the [temp: mobo] section: {'aa'}"
|
|
||||||
|
@ -8,6 +8,7 @@ from click.testing import CliRunner
|
|||||||
from afancontrol import fantest
|
from afancontrol import fantest
|
||||||
from afancontrol.fantest import (
|
from afancontrol.fantest import (
|
||||||
CSVMeasurementsOutput,
|
CSVMeasurementsOutput,
|
||||||
|
Fan,
|
||||||
HumanMeasurementsOutput,
|
HumanMeasurementsOutput,
|
||||||
MeasurementsOutput,
|
MeasurementsOutput,
|
||||||
fantest as main,
|
fantest as main,
|
||||||
@ -23,7 +24,6 @@ from afancontrol.pwmfan import (
|
|||||||
LinuxFanSpeed,
|
LinuxFanSpeed,
|
||||||
PWMDevice,
|
PWMDevice,
|
||||||
PWMValue,
|
PWMValue,
|
||||||
ReadWriteFan,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -65,7 +65,7 @@ def test_main_smoke(temp_path):
|
|||||||
args, kwargs = mocked_fantest.call_args
|
args, kwargs = mocked_fantest.call_args
|
||||||
assert not args
|
assert not args
|
||||||
assert kwargs.keys() == {"fan", "pwm_step_size", "output"}
|
assert kwargs.keys() == {"fan", "pwm_step_size", "output"}
|
||||||
assert kwargs["fan"] == ReadWriteFan(
|
assert kwargs["fan"] == Fan(
|
||||||
fan_speed=LinuxFanSpeed(FanInputDevice(str(fan_input_path))),
|
fan_speed=LinuxFanSpeed(FanInputDevice(str(fan_input_path))),
|
||||||
pwm_read=LinuxFanPWMRead(PWMDevice(str(pwm_path))),
|
pwm_read=LinuxFanPWMRead(PWMDevice(str(pwm_path))),
|
||||||
pwm_write=LinuxFanPWMWrite(PWMDevice(str(pwm_path))),
|
pwm_write=LinuxFanPWMWrite(PWMDevice(str(pwm_path))),
|
||||||
@ -77,11 +77,11 @@ def test_main_smoke(temp_path):
|
|||||||
@pytest.mark.parametrize("pwm_step_size", [5, -5])
|
@pytest.mark.parametrize("pwm_step_size", [5, -5])
|
||||||
@pytest.mark.parametrize("output_cls", [HumanMeasurementsOutput, CSVMeasurementsOutput])
|
@pytest.mark.parametrize("output_cls", [HumanMeasurementsOutput, CSVMeasurementsOutput])
|
||||||
def test_fantest(output_cls: Type[MeasurementsOutput], pwm_step_size: PWMValue):
|
def test_fantest(output_cls: Type[MeasurementsOutput], pwm_step_size: PWMValue):
|
||||||
fan: Any = ReadWriteFan(
|
fan = Fan(
|
||||||
fan_speed=MagicMock(spec=BaseFanSpeed),
|
fan_speed=MagicMock(spec=BaseFanSpeed),
|
||||||
pwm_read=MagicMock(spec=BaseFanPWMRead),
|
pwm_read=MagicMock(spec=BaseFanPWMRead),
|
||||||
pwm_write=MagicMock(spec=BaseFanPWMWrite),
|
pwm_write=MagicMock(spec=BaseFanPWMWrite),
|
||||||
)
|
) # type: Any
|
||||||
fan.pwm_read.min_pwm = 0
|
fan.pwm_read.min_pwm = 0
|
||||||
fan.pwm_read.max_pwm = 255
|
fan.pwm_read.max_pwm = 255
|
||||||
output = output_cls()
|
output = output_cls()
|
||||||
|
@ -1,5 +1,4 @@
|
|||||||
from contextlib import ExitStack
|
from contextlib import ExitStack
|
||||||
from typing import cast
|
|
||||||
from unittest.mock import MagicMock, patch, sentinel
|
from unittest.mock import MagicMock, patch, sentinel
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
@ -69,7 +68,7 @@ def test_manager(report):
|
|||||||
|
|
||||||
manager.tick()
|
manager.tick()
|
||||||
|
|
||||||
mocked_triggers = cast(MagicMock, manager.triggers)
|
mocked_triggers = manager.triggers # type: MagicMock
|
||||||
assert mocked_triggers.check.call_count == 1
|
assert mocked_triggers.check.call_count == 1
|
||||||
assert mocked_case_fan.__enter__.call_count == 1
|
assert mocked_case_fan.__enter__.call_count == 1
|
||||||
assert mocked_metrics.__enter__.call_count == 1
|
assert mocked_metrics.__enter__.call_count == 1
|
||||||
|
6
tox.ini
6
tox.ini
@ -1,7 +1,11 @@
|
|||||||
[tox]
|
[tox]
|
||||||
envlist=py{36,37,38,39,310}{,-arduino,-metrics},lint,check-docs
|
envlist=py{35,36,37,38,39}{,-arduino,-metrics},lint,check-docs
|
||||||
|
|
||||||
[testenv]
|
[testenv]
|
||||||
|
deps =
|
||||||
|
; Python 3.5 still ships an old setuptools version which doesn't support
|
||||||
|
; declarative setup.cfg format.
|
||||||
|
setuptools>=41.4.0
|
||||||
extras =
|
extras =
|
||||||
arduino: arduino
|
arduino: arduino
|
||||||
dev
|
dev
|
||||||
|
Loading…
Reference in New Issue
Block a user