Function main

make_playlist.py:123–158

()

def main():
    base_dir = os.path.dirname(os.path.abspath(__file__))
    lists_dir = os.path.join(base_dir, "lists")
    dir_playlists = os.path.join(base_dir, "playlists")

    if not os.path.isdir(dir_playlists):
        os.mkdir(dir_playlists)

    with open(os.path.join(base_dir, "epglist.txt"), encoding='utf-8') as epg_file:
        epg_urls = [line.strip() for line in epg_file if line.strip()]
    processed_epg_list = ", ".join(epg_urls)
    head_playlist = f'#EXTM3U x-tvg-url="{processed_epg_list}"\n'

    with open(os.path.join(base_dir, "playlist.m3u8"), "w", encoding='utf-8') as playlist:
        playlist.write(head_playlist)
        for filename in sorted(os.listdir(lists_dir)):
            if filename == "README.md" or not filename.endswith(".md"):
                continue
            markup_path = os.path.join(lists_dir, filename)
            country_path = os.path.join(dir_playlists, "playlist_" + filename[:-3] + ".m3u8")
            country_key = filename[:-3]
            group = country_key.replace("_", " ").title()
            country_code = COUNTRY_CODES.get(country_key, "")
            print(f"Generating {group}")
            with open(markup_path, encoding='utf-8') as markup_file, \
                 open(country_path, "w", encoding='utf-8') as playlist_country:
                playlist_country.write(head_playlist)
                for line in markup_file:
                    if "<h1>" in line.lower() and "</h1>" in line.lower():
                        group = re.sub('<[^<>]+>', '', line.strip())
                    if "[>]" not in line:
                        continue
                    channel = Channel(group, line, country_code)
                    m3u_line = channel.to_m3u_line()
                    print(m3u_line, file=playlist)
                    print(m3u_line, file=playlist_country)

if __name__ == "__main__":
    main()