Add client fetch streams (#183)

* implemented a fetch_streams method for the Client class

* put 1lines <120 back as they were

* removed print, corrected docstring

* changing per IAmTomahawkx's feedback for the PR

* changed user_ids and game_ids to be List[int] per pr feedback

Author: bpafoshizle

twitchio/client.py

@@ -91,7 +91,11 @@ def __init__(
 
     @classmethod
     def from_client_credentials(
-        cls, client_id: str, client_secret: str, *, loop: asyncio.AbstractEventLoop = None
+        cls,
+        client_id: str,
+        client_secret: str,
+        *,
+        loop: asyncio.AbstractEventLoop = None,
     ) -> "Client":
         """
         creates a client application token from your client credentials.
@@ -121,7 +125,7 @@ def from_client_credentials(
         self.loop = loop or asyncio.get_event_loop()
         self._http = TwitchHTTP(self, client_id=client_id, client_secret=client_secret)
         self._connection = WSConnection(
-            client=self, loop=self.loop, initial_channels=None, heartbeat=self._heartbeat
+            client=self, loop=self.loop, initial_channels=None
         )  # The only reason we're even creating this is to avoid attribute errors
         self._events = {}
         self._waiting = []
@@ -480,6 +484,46 @@ async def fetch_tags(self, ids: List[str] = None):
         data = await self._http.get_stream_tags(ids)
         return [models.Tag(x) for x in data]
 
+    async def fetch_streams(
+        self,
+        user_ids: List[int] = None,
+        game_ids: List[int] = None,
+        user_logins: List[str] = None,
+        languages: List[str] = None,
+        token: str = None,
+    ):
+        """|coro|
+        Fetches live streams from the helix API
+
+        Parameters
+        -----------
+        user_ids: Optional[List[:class:`int`]]
+            user ids of people whose streams to fetch
+        game_ids: Optional[List[:class:`int`]]
+            game ids of streams to fetch
+        user_logins: Optional[List[:class:`str`]]
+            user login names of people whose streams to fetch
+        languages: Optional[List:class:`str]]
+            language for the stream(s). ISO 639-1 or two letter code for supported stream language
+        token: Optional[:class:`str`]
+            An optional OAuth token to use instead of the bot OAuth token
+
+        Returns
+        --------
+        List[:class:`twitchio.Stream`]
+        """
+        from .models import Stream
+
+        assert user_ids or game_ids or user_logins
+        data = await self._http.get_streams(
+            game_ids=game_ids,
+            user_ids=user_ids,
+            user_logins=user_logins,
+            languages=languages,
+            token=token,
+        )
+        return [Stream(self._http, x) for x in data]
+
     async def search_categories(self, query: str):
         """|coro|
         Searches twitches categories

twitchio/models.py

@@ -530,3 +530,38 @@ def __init__(self, data: dict):
 
     def __repr__(self):
         return f"<WebhookSubscription callback={self.callback} topic={self.topic} expires_at={self.expires_at}>"
+
+
+class Stream:
+
+    __slots__ = (
+        "id",
+        "user",
+        "game_id",
+        "game_name",
+        "type",
+        "title",
+        "viewer_count",
+        "started_at",
+        "language",
+        "thumbnail_url",
+        "tag_ids",
+        "is_mature",
+    )
+
+    def __init__(self, http: "TwitchHTTP", data: dict):
+        self.id: int = data["id"]
+        self.user = PartialUser(http, data["user_id"], data["user_name"], data["user_login"])
+        self.game_id: int = data["game_id"]
+        self.game_name: str = data["game_name"]
+        self.type: str = data["type"]
+        self.title: str = data["title"]
+        self.viewer_count: int = data["viewer_count"]
+        self.started_at = datetime.datetime.strptime(data["started_at"], "%Y-%m-%dT%H:%M:%SZ")
+        self.language: str = data["language"]
+        self.thumbnail_url: str = data["thumbnail_url"]
+        self.tag_ids: List[str] = data["tag_ids"]
+        self.is_mature: bool = data["is_mature"]
+
+    def __repr__(self):
+        return f"<Stream id={self.id} user={self.user} title={self.title} started_at={self.started_at}>"