From a2aba983d162dc8432521fdc2c847976b326af34 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?K=C3=A1lm=C3=A1n=20=E2=80=9EKAMI=E2=80=9D=20Szalai?= Date: Sat, 23 Jul 2022 09:26:45 +0200 Subject: [PATCH] Add initial support of EPG id --- make_playlist.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/make_playlist.py b/make_playlist.py index cac4342..6d7697f 100755 --- a/make_playlist.py +++ b/make_playlist.py @@ -8,16 +8,24 @@ class Channel(): def __init__(self, group, md_line): self.group = group md_line = md_line.strip() - (before, number, name, url, logo, after) = md_line.split("|") + try: + (before, number, name, url, logo, epg, after) = md_line.split("|") + except ValueError: + (before, number, name, url, logo, after) = md_line.split("|") + epg = None self.number = number.strip() self.name = name.strip() self.url = url.strip() self.url = self.url[self.url.find("(")+1:self.url.rfind(")")] self.logo = logo.strip() self.logo = self.logo[self.logo.find('src="')+5:self.logo.rfind('"')] + self.epg = epg.strip() if epg else None def to_m3u_line(self): - return (f'#EXTINF:-1 tvg-name="{self.name}" tvg-logo="{self.logo}" group-title="{self.group}",{self.name}\n{self.url}') + if self.epg is None: + return (f'#EXTINF:-1 tvg-name="{self.name}" tvg-logo="{self.logo}" group-title="{self.group}",{self.name}\n{self.url}') + else: + return (f'#EXTINF:-1 tvg-name="{self.name}" tvg-logo="{self.logo}" tvg-id="{self.epg}" group-title="{self.group}",{self.name}\n{self.url}') if __name__ == "__main__": with open("playlist.m3u8", "w", encoding='utf-8') as playlist: @@ -35,4 +43,3 @@ if __name__ == "__main__": continue channel = Channel(group, line) print(channel.to_m3u_line(), file=playlist) -