Prioritize Arabic groups in playlist generation

pull/1084/head
copilot-swe-agent[bot] 2026-06-19 14:55:39 +00:00 committed by GitHub
parent f1f3666ec9
commit 795acf0e58
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 1 deletions

View File

@ -90,6 +90,16 @@ COUNTRY_CODES = {
"venezuela": "VE", "venezuela": "VE",
} }
ARABIC_PRIORITY_COUNTRIES = {
"chad",
"egypt",
"iraq",
"qatar",
"saudi_arabia",
"somalia",
"united_arab_emirates",
}
class Channel: class Channel:
def __init__(self, group, md_line, country_code=""): def __init__(self, group, md_line, country_code=""):
@ -120,6 +130,12 @@ class Channel:
return (f'#EXTINF:-1 tvg-name="{self.name}" tvg-logo="{self.logo}" tvg-id="{self.epg}"{chno}{country} group-title="{self.group}",{self.name}\n{self.url}') return (f'#EXTINF:-1 tvg-name="{self.name}" tvg-logo="{self.logo}" tvg-id="{self.epg}"{chno}{country} group-title="{self.group}",{self.name}\n{self.url}')
def playlist_sort_key(filename):
country_key = filename[:-3]
arabic_priority = country_key.endswith("_ar") or country_key in ARABIC_PRIORITY_COUNTRIES
return (0 if arabic_priority else 1, filename)
def main(): def main():
base_dir = os.path.dirname(os.path.abspath(__file__)) base_dir = os.path.dirname(os.path.abspath(__file__))
lists_dir = os.path.join(base_dir, "lists") lists_dir = os.path.join(base_dir, "lists")
@ -135,7 +151,7 @@ def main():
with open(os.path.join(base_dir, "playlist.m3u8"), "w", encoding='utf-8') as playlist: with open(os.path.join(base_dir, "playlist.m3u8"), "w", encoding='utf-8') as playlist:
playlist.write(head_playlist) playlist.write(head_playlist)
for filename in sorted(os.listdir(lists_dir)): for filename in sorted(os.listdir(lists_dir), key=playlist_sort_key):
if filename == "README.md" or not filename.endswith(".md"): if filename == "README.md" or not filename.endswith(".md"):
continue continue
markup_path = os.path.join(lists_dir, filename) markup_path = os.path.join(lists_dir, filename)