Merge pull request #5 from tmcarr/master

Revert imports "cleanup"

Author: maykar

custom_components/sensor/plex_recently_added.py

@@ -7,26 +7,16 @@
 https://github.com/custom-cards/upcoming-media-card
 
 """
-import json
-import logging
-import math
-import os
 import os.path
-import re
-import time
-import xml.etree.ElementTree as ET
-from datetime import datetime
-from unicodedata import normalize
-from urllib.parse import quote
-
+import logging
+import json
 import requests
-
-import homeassistant.helpers.config_validation as cv
 import voluptuous as vol
+import homeassistant.helpers.config_validation as cv
+from datetime import datetime
 from homeassistant.components.sensor import PLATFORM_SCHEMA
 from homeassistant.const import CONF_HOST, CONF_PORT, CONF_SSL
 from homeassistant.helpers.entity import Entity
-from pytz import timezone, utc
 
 __version__ = '0.1.7'
 
@@ -60,6 +50,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
 class PlexRecentlyAddedSensor(Entity):
 
     def __init__(self, hass, conf):
+        from pytz import timezone
         self.conf_dir = str(hass.config.path()) + '/'
         self._dir = conf.get(CONF_IMG_CACHE)
         self.img = '{0}{1}{2}{3}{4}.jpg'.format(
@@ -96,6 +87,7 @@ def state(self):
 
     @property
     def device_state_attributes(self):
+        import math
         attributes = {}
         if self.change_detected:
             self.card_json = []
@@ -185,6 +177,8 @@ def device_state_attributes(self):
         return attributes
 
     def update(self):
+        import re
+        import os
         plex = requests.Session()
         if not self.cert:
             """Default SSL certificate is for plex.tv not our api server"""
@@ -291,6 +285,7 @@ def update(self):
 
 def image_url(url_elements, cert_check, img):
     """Plex can resize images with a long & partially % encoded url."""
+    from urllib.parse import quote
     ssl, host, local, port, token, self_cert, dl_images = url_elements
     if not cert_check and not self_cert:
         ssl = ''
@@ -314,6 +309,8 @@ def image_url(url_elements, cert_check, img):
 
 def get_server_ip(name, token):
     """With a token and server name we get server's ip, local ip, and port"""
+    import xml.etree.ElementTree as ET
+    from unicodedata import normalize
     plex_tv = requests.get(
         'https://plex.tv/api/servers.xml?X-Plex-Token=' + token, timeout=10)
     plex_xml = ET.fromstring(plex_tv.content)
@@ -326,6 +323,8 @@ def get_server_ip(name, token):
 
 
 def days_since(date, tz):
+    import time
+    from pytz import utc
     date = datetime.utcfromtimestamp(date).isoformat() + 'Z'
     date = datetime.strptime(date, '%Y-%m-%dT%H:%M:%SZ')
     date = str(date.replace(tzinfo=utc).astimezone(tz))[:10]
@@ -366,7 +365,11 @@ def view_count(data):
 
 def get_info(title):
     tmdb_url = requests.get('https://api.themoviedb.org/3/search/movie?'
-                            'api_key=1f7708bb9a218ab891a5d438b1b63992&query='
+                            + 'api_key=1f7708bb9a218ab891a5d438b1b63992&query='
                             + title)
-    tmdb_json = tmdb_url.json()
-    return tmdb_json['results'][0]['overview']
+    try:
+        tmdb_json = tmdb_url.json()['results'][0]['overview']
+    except:
+        _LOGGER.warning("Failed to get information from TMDB")
+        tmdb_json = {}
+    return tmdb_json