From 00fbdb3a02fe95915e5e4774d16d3086afa6f67f Mon Sep 17 00:00:00 2001
From: Hendrik Melse <hendrik.melse@wsu.edu>
Date: Mon, 16 Aug 2021 22:24:47 -0700
Subject: [PATCH] Added TV streaming

---
 README.rst         | 1 +
 berserk/clients.py | 9 +++++++++
 2 files changed, 10 insertions(+)

diff --git a/README.rst b/README.rst
index 098a41b..eadac1b 100644
--- a/README.rst
+++ b/README.rst
@@ -95,6 +95,7 @@ Most if not all of the API is available:
     client.board.stream_incoming_events
     client.board.seek
     client.board.stream_game_state
+    client.board.stream_tv_game
     client.board.make_move
     client.board.post_message
     client.board.abort_game
diff --git a/berserk/clients.py b/berserk/clients.py
index f0b13fd..4f914f8 100644
--- a/berserk/clients.py
+++ b/berserk/clients.py
@@ -693,6 +693,15 @@ def stream_game_state(self, game_id):
         yield from self._r.get(path, stream=True,
                                converter=models.GameState.convert)
 
+    def stream_tv_game(self):
+        """Get the stream of events for the current game on Lichess TV.
+
+        :return iterator over game states
+        """
+        path = 'api/tv/feed'
+        yield from self._r.get(path, stream=True,
+                               converter=models.GameState.convert)
+
     def make_move(self, game_id, move):
         """Make a move in a board game.