Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 19:23:34 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Allow deriving from SmartModule without being registered (#1189)

Browse Source
This commit is contained in:
Steven B. 2024-10-23 21:42:01 +01:00 committed by GitHub
parent a88b677776
commit 51958d8078
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kasa/smart/smartmodule.py
View File

@ -76,9 +76,11 @@ class SmartModule(Module):
self._error_count = 0
def __init_subclass__(cls, **kwargs):
name = getattr(cls, "NAME", cls.__name__)
_LOGGER.debug("Registering %s", cls)
cls.REGISTERED_MODULES[name] = cls
# We only want to register submodules in a modules package so that
# other classes can inherit from smartmodule and not be registered
if cls.__module__.split(".")[-2] == "modules":
_LOGGER.debug("Registering %s", cls)
cls.REGISTERED_MODULES[cls.__name__] = cls
def _set_error(self, err: Exception | None):
if err is None: