streamora_backend / get_streams.py
raannakasturi's picture
Add initial implementation of streaming service with multiple providers
e37348d
import json
from providers.auto_embed import AutoEmbed
from providers.embed import Embed
from providers.vidsrc import VidSrc
from providers.two_embed import TwoEmbed
from providers.vidzee import Vidzee
from providers.netfree import NetFree
providers = [
AutoEmbed,
Embed,
VidSrc,
TwoEmbed,
Vidzee,
NetFree,
]
def get_streams(tmdb_id, imdb_id, media_type, title, year, season=None, episode=None):
streams = []
for provider in providers:
try:
encoded_data = provider().get_stream(tmdb_id=tmdb_id, imdb_id=imdb_id, media_type=media_type, title=title, year=year, season=season, episode=episode)
streams.extend(encoded_data)
except Exception as e:
print(f"Error with provider {provider.__name__}: {e}")
return json.dumps(streams, indent=4)
# if __name__ == "__main__":
# media_type = "movie"
# tmdb_id = "822119"
# imdb_id = "tt14513804"
# title = "Dhoom Dhaam"
# year = "2025"
# season = None
# episode = None
# stream_list = get_streams(
# tmdb_id=tmdb_id,
# imdb_id=imdb_id,
# media_type=media_type,
# title=title,
# year=year,
# season=season,
# episode=episode,
# )
# print(stream_list)
# print("-" * 25)
# for stream in json.loads(stream_list):
# print(f"Stream: {stream}")
# print(f"VideoData(videoSource: '{stream['videoSource']}', videoSourceUrl: '{stream['videoUrl']}', videoSourceHeaders: '{stream['videoHeaders']}',),")