Use entity class atrributes (#50)

* use _attr

* icon

* remove typing

Author: yuvalabou

custom_components/feedparser/sensor.py

@@ -1,18 +1,20 @@
 """Feedparser sensor"""
+from __future__ import annotations
 
 import asyncio
 import re
-import feedparser
-import voluptuous as vol
 from datetime import timedelta
-from dateutil import parser
-from homeassistant.components.sensor import SensorEntity
+
 import homeassistant.helpers.config_validation as cv
-from homeassistant.components.sensor import PLATFORM_SCHEMA
+import voluptuous as vol
+from dateutil import parser
+from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
 from homeassistant.const import CONF_NAME
 import homeassistant.util.dt as dt
 
-__version__ = "0.1.2"
+import feedparser
+
+__version__ = "0.1.6"
 
 REQUIREMENTS = ["feedparser"]
 
@@ -27,7 +29,6 @@
 
 COMPONENT_REPO = "https://github.com/custom-components/sensor.feedparser/"
 SCAN_INTERVAL = timedelta(hours=1)
-ICON = "mdi:rss"
 
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
     {
@@ -70,32 +71,33 @@ def __init__(
         local_time: bool,
         exclusions: str,
         inclusions: str,
-    ):
+    ) -> None:
         self._feed = feed
-        self._name = name
+        self._attr_name = name
+        self._attr_icon = "mdi:rss"
         self._date_format = date_format
         self._local_time = local_time
         self._show_topn = show_topn
         self._inclusions = inclusions
         self._exclusions = exclusions
-        self._state = None
+        self._attr_state = None
         self._entries = []
 
     def update(self):
-        parsedFeed = feedparser.parse(self._feed)
+        parsed_feed = feedparser.parse(self._feed)
 
-        if not parsedFeed:
+        if not parsed_feed:
             return False
         else:
-            self._state = (
+            self._attr_state = (
                 self._show_topn
-                if len(parsedFeed.entries) > self._show_topn
-                else len(parsedFeed.entries)
+                if len(parsed_feed.entries) > self._show_topn
+                else len(parsed_feed.entries)
             )
             self._entries = []
 
-            for entry in parsedFeed.entries[: self._state]:
-                entryValue = {}
+            for entry in parsed_feed.entries[: self._attr_state]:
+                entry_value = {}
 
                 for key, value in entry.items():
                     if (
@@ -111,34 +113,22 @@ def update(self):
                         value = value.strftime(self._date_format)
 
 
-                    entryValue[key] = value
+                    entry_value[key] = value
 
-                if "image" in self._inclusions and "image" not in entryValue.keys():
+                if "image" in self._inclusions and "image" not in entry_value.keys():
                     images = []
                     if "summary" in entry.keys():
                         images = re.findall(
                             r"<img.+?src=\"(.+?)\".+?>", entry["summary"]
                         )
                     if images:
-                        entryValue["image"] = images[0]
+                        entry_value["image"] = images[0]
                     else:
-                        entryValue[
+                        entry_value[
                             "image"
                         ] = "https://www.home-assistant.io/images/favicon-192x192-full.png"
 
-                self._entries.append(entryValue)
-
-    @property
-    def name(self):
-        return self._name
-
-    @property
-    def state(self):
-        return self._state
-
-    @property
-    def icon(self):
-        return ICON
+                self._entries.append(entry_value)
 
     @property
     def device_state_attributes(self):