sadorowo/m3u-category-matcher
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: first channel without category

Browse Source 
due to dynamic playlist changing, enumerate() wasn't working properly
This commit is contained in:
Franek 2024-03-25 20:30:20 +01:00 committed by GitHub
parent c497d690ca
commit 121b9cce15
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
__main__.py
View File

@ -54,10 +54,12 @@ def process(arguments):
# Add category "News" to channel 5. # Add category "News" to channel 5.
current_category = None current_category = None
channels_to_remove = []
for index, channel in enumerate(playlist): for index, channel in enumerate(playlist):
match = regex.match(channel.name) match = regex.match(channel.name)
if match: if match:
playlist.remove_channel(index) channels_to_remove.append(index)
current_category = match.group(1) current_category = match.group(1)
debug(f"Category matched: {current_category}") debug(f"Category matched: {current_category}")
@ -65,6 +67,9 @@ def process(arguments):
debug(f"Adding category {current_category} to channel {channel.name}") debug(f"Adding category {current_category} to channel {channel.name}")
channel.attributes["group-title"] = current_category channel.attributes["group-title"] = current_category
for index in channels_to_remove:
playlist.remove_channel(index)
# save playlist # save playlist
debug("Saving playlist...") debug("Saving playlist...")
with open(arguments.input, "w+", encoding="utf-8") as f: with open(arguments.input, "w+", encoding="utf-8") as f: