SSL Support / Air Date (#7)

Author: ctalkington

sonarr/models.py

@@ -139,6 +139,7 @@ class Episode:
     identifier: str
     title: str
     overview: str
+    airdate: str
     airs: datetime
     downloaded: bool
     downloading: bool
@@ -163,6 +164,7 @@ def from_dict(data: dict):
             identifier=identifier,
             title=data.get("title", ""),
             overview=data.get("overview", ""),
+            airdate=data.get("airDate", ""),
             airs=airs,
             downloaded=data.get("hasFile", False),
             downloading=data.get("downloading", False),

sonarr/sonarr.py

@@ -33,6 +33,8 @@ def __init__(
         port: int = 8989,
         request_timeout: int = 8,
         session: aiohttp.client.ClientSession = None,
+        tls: bool = False,
+        verify_ssl: bool = True,
         user_agent: str = None,
     ) -> None:
         """Initialize connection with receiver."""
@@ -44,6 +46,8 @@ def __init__(
         self.host = host
         self.port = port
         self.request_timeout = request_timeout
+        self.tls = tls
+        self.verify_ssl = verify_ssl
         self.user_agent = user_agent
 
         if user_agent is None:
@@ -60,7 +64,7 @@ async def _request(
         params: Optional[Mapping[str, str]] = None,
     ) -> Any:
         """Handle a request to API."""
-        scheme = "http"
+        scheme = "https" if self.tls else "http"
 
         url = URL.build(
             scheme=scheme, host=self.host, port=self.port, path=self.base_path
@@ -79,7 +83,12 @@ async def _request(
         try:
             with async_timeout.timeout(self.request_timeout):
                 response = await self._session.request(
-                    method, url, data=data, params=params, headers=headers,
+                    method,
+                    url,
+                    data=data,
+                    params=params,
+                    headers=headers,
+                    ssl=self.verify_ssl,
                 )
         except asyncio.TimeoutError as exception:
             raise SonarrConnectionError(
@@ -144,7 +153,7 @@ async def update(self, full_update: bool = False) -> Application:
         self._application.update_from_dict({"diskspace": diskspace})
         return self._application
 
-    async def calendar(self, start: int = None, end: int = None) -> List[Episode]:
+    async def calendar(self, start: str = None, end: str = None) -> List[Episode]:
         """Get upcoming episodes.
 
         If start/end are not supplied, episodes airing

tests/test_models.py

@@ -96,6 +96,7 @@ def test_episode() -> None:
     assert isinstance(episode.series, models.Series)
     assert episode.title == "Easy Com-mercial, Easy Go-mercial"
     assert episode.overview == overview.replace("\n", " ")
+    assert episode.airdate == "2014-01-26"
     assert episode.airs == datetime(2014, 1, 27, 1, 30, tzinfo=timezone.utc)
     assert not episode.downloaded
     assert not episode.downloading