Update run.py

run.py

@@ -7,7 +7,7 @@ from streamingwatch import streamingwatch
 import json
 import config
 import logging
-
+from okru import okru_get_url
 # Configure logging
 FILMPERTUTTI = config.FILMPERTUTTI
 STREAMINGCOMMUNITY = config.STREAMINGCOMMUNITY
@@ -94,6 +94,11 @@ STREAM = {
     ]
 }
 
+okru = {
+    "rai1": "https://ok.ru/videoembed/7703488765552?nochat=1",
+    "rai2": "https://ok.ru/videoembed/7805618364016?nochat=1"
+}
+
 MANIFEST["catalogs"].append({"type": "tv", "id": "channels", "name": "Channels"})
 
 
@@ -165,10 +170,17 @@ def addon_stream(type, id):
     if type == "tv":
         for channel in STREAM["channels"]:
             if channel["id"] == id:
-                streams['streams'].append({
-                    'title': channel['title'],
-                    'url': channel['url']
-                })
+                if id in okru:
+                    channel_url = okru_get_url(id)
+                    streams['streams'].append({
+                        'title': "okru"+channel['title'],
+                        'url': channel_url
+                    })
+                else:
+                    streams['streams'].append({
+                        'title': channel['title'],
+                        'url': channel['url']
+                    })
         if not streams['streams']:
             abort(404)
         return respond_with(streams)
@@ -215,4 +227,4 @@ def addon_stream(type, id):
 
     return respond_with({"meta": meta})
 if __name__ == '__main__':
-    app.run(host=HOST, port=PORT)
+    app.run(host=HOST, port=PORT)