Add shared code to a common module

2020-09-23 19:58:24 +02:00
parent 98612a0d70
commit fefee6944f
7 changed files with 201 additions and 197 deletions
--- a/custom_components/localtuya/light.py
+++ b/custom_components/localtuya/light.py
@@ -30,11 +30,10 @@ from homeassistant.components.light import (

 from . import (
    BASE_PLATFORM_SCHEMA,
-    TuyaDevice,
-    LocalTuyaEntity,
    import_from_yaml,
-    prepare_setup_entities,
 )
+from .common import LocalTuyaEntity, TuyaDevice, prepare_setup_entities
+

 _LOGGER = logging.getLogger(__name__)