Streamer.py

import sys
import os
import glob
import re
import requests
import json
import wget
from bs4 import BeautifulSoup
from yt_dlp import YoutubeDL
import base64
import threading
import time
import os

class Streamer:
    def __init__(self, name:str, red_url:str, id:int):
        self.name = name
        self.red_url = red_url
        self.id = id
        self.url = self.construct_episode_url()
        self.m3u8_url = self.get_m3u8_url()

    def construct_episode_url(self):
        return "https://aniworld.to" + str(self.red_url)

    def set_m3u8_url(self, m3u8_url:str):
        self.m3u8_url = m3u8_url

    def get_m3u8_url(self):
        if self.name == "VOE":
            return self.extract_m3u8_links_VOE(self.url)
        else:
            return dict()

    def extract_m3u8_links_VOE(self, URL):
        URL = str(URL)
        headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:127.0) Gecko/20100101 Firefox/127.0",
            "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8",
            "Accept-Language": "de,en-US;q=0.7,en;q=0.3",
            "Upgrade-Insecure-Requests": "1",
            "Sec-Fetch-Dest": "document",
            "Sec-Fetch-Mode": "navigate",
            "Sec-Fetch-Site": "none",
            "Sec-Fetch-User": "?1",
            "Priority": "u=1"
        }
        html_page = requests.get(URL, headers=headers)

        soup = BeautifulSoup(html_page.content, 'html.parser')
        if html_page.text.startswith("<script>"):
            START = "window.location.href = '"
            L = len(START)
            i0 = html_page.text.find(START)
            i1 = html_page.text.find("'",i0+L)
            url = html_page.text[i0+L:i1]
            return self.extract_m3u8_links_VOE(url)

        name_find = soup.find('meta', attrs={"name":"og:title"})
        name = name_find["content"]
        name = name.replace(" ","_")
    #     print("Name of file: " + name)

        sources_find = soup.find_all(string = re.compile("var sources")) #searching for the script tag containing the link to the mp4
        sources_find = str(sources_find)
        #slice_start = sources_find.index("const sources")
        slice_start = sources_find.index("var sources")
        source = sources_find[slice_start:] #cutting everything before 'var sources' in the script tag
        slice_end = source.index(";")
        source = source[:slice_end] #cutting everything after ';' in the remaining String to make it ready for the JSON parser

        source = source.replace("var sources = ","")    #
        source = source.replace("\'","\"")                #Making the JSON valid
        source = source.replace("\\n","")                 #
        source = source.replace("\\","")                  #

        strToReplace = ","
        replacementStr = ""
        source = replacementStr.join(source.rsplit(strToReplace, 1)) #complicated but needed replacement of the last comma in the source String to make it JSON valid

        source_json = json.loads(source) #parsing the JSON
        result = {}
        try:
            link = source_json["mp4"] #extracting the link to the mp4 file
            link = base64.b64decode(link)
            link = link.decode("utf-8")
            result['mp4'] = link
        except KeyError:
            try:
                link = source_json["hls"]
                link = base64.b64decode(link)
                link = link.decode("utf-8")
                result['hls'] = link
            except KeyError:
                print("Could not find downloadable URL. Voe might have change their site. Check that you are running the latest version of voe-dl, and if so file an issue on GitHub.")
        return result