From 898b69931131cd844732f3ab4edf56402c4b92a8 Mon Sep 17 00:00:00 2001
From: "David F. Mulcahey" <david.mulcahey@icloud.com>
Date: Sun, 10 Feb 2019 09:56:27 -0500
Subject: [PATCH] Add quirks info to ZHA device (#20923)

* add quirks info to zha device

* move import

* remove device entity part
---
 homeassistant/components/zha/core/const.py  |  3 +++
 homeassistant/components/zha/core/device.py | 13 +++++++++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/zha/core/const.py b/homeassistant/components/zha/core/const.py
index cb3a311c985..5edcadc7fce 100644
--- a/homeassistant/components/zha/core/const.py
+++ b/homeassistant/components/zha/core/const.py
@@ -88,6 +88,9 @@ SIGNAL_STATE_ATTR = "update_state_attribute"
 SIGNAL_AVAILABLE = 'available'
 SIGNAL_REMOVE = 'remove'
 
+QUIRK_APPLIED = 'quirk_applied'
+QUIRK_CLASS = 'quirk_class'
+
 
 class RadioType(enum.Enum):
     """Possible options for radio type."""
diff --git a/homeassistant/components/zha/core/device.py b/homeassistant/components/zha/core/device.py
index 2322df5452c..6bbfcd43a94 100644
--- a/homeassistant/components/zha/core/device.py
+++ b/homeassistant/components/zha/core/device.py
@@ -14,7 +14,8 @@ from .const import (
     ATTR_MANUFACTURER, LISTENER_BATTERY, SIGNAL_AVAILABLE, IN, OUT,
     ATTR_CLUSTER_ID, ATTR_ATTRIBUTE, ATTR_VALUE, ATTR_COMMAND, SERVER,
     ATTR_COMMAND_TYPE, ATTR_ARGS, CLIENT_COMMANDS, SERVER_COMMANDS,
-    ATTR_ENDPOINT_ID, IEEE, MODEL, NAME, UNKNOWN
+    ATTR_ENDPOINT_ID, IEEE, MODEL, NAME, UNKNOWN, QUIRK_APPLIED,
+    QUIRK_CLASS
 )
 from .listeners import EventRelayListener
 
@@ -52,6 +53,12 @@ class ZHADevice:
             self._available_signal,
             self.async_initialize
         )
+        from zigpy.quirks import CustomDevice
+        self.quirk_applied = isinstance(self._zigpy_device, CustomDevice)
+        self.quirk_class = "{}.{}".format(
+            self._zigpy_device.__class__.__module__,
+            self._zigpy_device.__class__.__name__
+        )
 
     @property
     def name(self):
@@ -143,7 +150,9 @@ class ZHADevice:
             IEEE: ieee,
             ATTR_MANUFACTURER: self.manufacturer,
             MODEL: self.model,
-            NAME: self.name or ieee
+            NAME: self.name or ieee,
+            QUIRK_APPLIED: self.quirk_applied,
+            QUIRK_CLASS: self.quirk_class
         }
 
     def add_cluster_listener(self, cluster_listener):