Dispatcharr/core/xtream_codes.py

import requests

class Client:
    host = ""
    username = ""
    password = ""
    user_agent = ""

    def __init__(self, host, username, password, user_agent):
        self.host = host
        self.username = username
        self.password = password

        # Handle UserAgent objects by extracting the string value
        if hasattr(user_agent, 'user_agent'):  # Check if it's a UserAgent model object
            self.user_agent = user_agent.user_agent  # Extract the string attribute
        else:
            self.user_agent = str(user_agent)  # Ensure it's a string in any case

    def authenticate(self):
        response = requests.get(f"{self.host}/player_api.php?username={self.username}&password={self.password}", headers={
            'User-Agent': self.user_agent,
        })
        return response.json()

    def get_live_categories(self):
        response = requests.get(f"{self.host}/player_api.php?username={self.username}&password={self.password}&action=get_live_categories", headers={
            'User-Agent': self.user_agent,
        })
        return response.json()

    def get_live_category_streams(self, category_id):
        response = requests.get(f"{self.host}/player_api.php?username={self.username}&password={self.password}&action=get_live_streams&category_id={category_id}", headers={
            'User-Agent': self.user_agent,
        })
        return response.json()

    def get_stream_url(self, stream_id):
        return f"{self.host}/{self.username}/{self.password}/{stream_id}"