add EPG source support to m3u import

This commit is contained in:
DYefremov
2024-01-28 13:58:29 +03:00
parent 07c8034393
commit ee7eb01af5
2 changed files with 157 additions and 10 deletions

View File

@@ -705,30 +705,55 @@ class M3uImportDialog(IptvListDialog):
self._apply_button.set_label(translate("Import"))
# Extra box.
builder = get_builder(f"{UI_RESOURCES_PATH}m3u.glade", use_str=True, objects=("import_m3u_box",))
self._data_box.add(builder.get_object("import_m3u_box"))
self._info_label = builder.get_object("info_label")
self._progress_bar = builder.get_object("progress_bar")
self._spinner = builder.get_object("spinner")
self._spinner.bind_property("active", self._start_values_grid, "sensitive", 4)
self._picon_switch = builder.get_object("picon_switch")
self._picon_box = builder.get_object("picon_box")
# builder.get_object("import_type_box").set_visible(False)
# Type import buttons.
self._current_bq_button = builder.get_object("current_bq_button")
self._single_bq_button = builder.get_object("single_bq_button")
self._group_bq_button = builder.get_object("group_bq_button")
self._sub_bq_button = builder.get_object("sub_bq_button")
# EPG src.
self._epg_links_button = builder.get_object("epg_links_box")
self._add_epg_src_switch = builder.get_object("add_epg_src_switch")
m3u_box = builder.get_object("import_m3u_box")
if s_type is SettingsType.ENIGMA_2:
self._data_box.add(m3u_box)
else:
self._data_box.set_visible(False)
self._sub_bq_button.set_sensitive(False)
m3u_box.set_margin_start(5)
m3u_box.set_margin_end(5)
self._dialog.get_content_area().pack_start(m3u_box, True, True, 0)
self.get_m3u(m3_path, s_type)
@run_task
def get_m3u(self, path, s_type):
try:
GLib.idle_add(self._spinner.set_property, "active", True)
GLib.idle_add(self._spinner.start)
self._epg_src, self._services = parse_m3u(path, s_type)
for s in self._services:
if s.picon:
GLib.idle_add(self._picon_box.set_sensitive, True)
break
finally:
msg = f"{translate('Streams detected:')} {len(self._services) if self._services else 0}."
GLib.idle_add(self._info_label.set_text, msg)
GLib.idle_add(self._spinner.set_property, "active", False)
self.update_info()
@run_idle
def update_info(self):
msg = f"{translate('Streams detected:')} {len(self._services) if self._services else 0}."
self._info_label.set_text(msg)
self._spinner.stop()
if self._epg_src:
self._epg_links_button.set_visible(True)
[self._epg_links_button.append(u, u) for u in self._epg_src]
self._epg_links_button.set_active(0)
def on_apply(self, item):
if not self._app.current_bouquet:
@@ -766,6 +791,9 @@ class M3uImportDialog(IptvListDialog):
services.append(s._replace(picon=None, picon_id=picon_id, data_id=None, fav_id=fav_id))
if self._add_epg_src_switch.get_active():
self.on_add_epg_source()
if self._picon_switch.get_active():
if self.is_default_values():
msg = "Set values for TID, NID and Namespace for correct naming of the picons!"
@@ -778,6 +806,18 @@ class M3uImportDialog(IptvListDialog):
self._app.append_imported_services(services)
def on_add_epg_source(self):
active_src = self._epg_links_button.get_active_id()
settings = self._app.app_settings
sources = settings.epg_xml_sources
log(f"Adding an EPG source -> {active_src}")
if active_src not in set(sources):
sources.append(active_src)
settings.epg_xml_sources = sources
self._app.emit("epg-settings-changed", None)
else:
log(f"{translate('This URL already exists!')}")
@run_task
def download_picons(self, picons):
self._is_download = True

View File

@@ -464,7 +464,7 @@ Author: Dmitriy Yefremov
<property name="can-focus">False</property>
<property name="layout-style">expand</property>
<child>
<object class="GtkRadioButton" id="cuurent_bq_button">
<object class="GtkRadioButton" id="current_bq_button">
<property name="label" translatable="yes">Current bouquet</property>
<property name="name">header-button</property>
<property name="visible">True</property>
@@ -490,7 +490,7 @@ Author: Dmitriy Yefremov
<property name="receives-default">False</property>
<property name="active">True</property>
<property name="draw-indicator">False</property>
<property name="group">cuurent_bq_button</property>
<property name="group">current_bq_button</property>
</object>
<packing>
<property name="expand">True</property>
@@ -499,7 +499,7 @@ Author: Dmitriy Yefremov
</packing>
</child>
<child>
<object class="GtkRadioButton" id="broup_bq_button">
<object class="GtkRadioButton" id="group_bq_button">
<property name="label" translatable="yes">Split by groups</property>
<property name="name">header-button</property>
<property name="visible">True</property>
@@ -526,7 +526,7 @@ Author: Dmitriy Yefremov
<property name="receives-default">False</property>
<property name="active">True</property>
<property name="draw-indicator">False</property>
<property name="group">cuurent_bq_button</property>
<property name="group">current_bq_button</property>
</object>
<packing>
<property name="expand">True</property>
@@ -648,6 +648,113 @@ Author: Dmitriy Yefremov
<property name="position">1</property>
</packing>
</child>
<child>
<object class="GtkBox" id="epg_src_box">
<property name="height-request">30</property>
<property name="visible">True</property>
<property name="can-focus">False</property>
<child>
<object class="GtkLabel" id="epg_source_label">
<property name="visible">True</property>
<property name="can-focus">False</property>
<property name="label" translatable="yes">EPG source</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">0</property>
</packing>
</child>
<child>
<object class="GtkLabel">
<property name="visible">True</property>
<property name="can-focus">False</property>
<property name="label">:</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">1</property>
</packing>
</child>
<child>
<object class="GtkComboBoxText" id="epg_links_box">
<property name="can-focus">True</property>
<property name="halign">start</property>
<property name="margin-start">5</property>
<property name="margin-end">5</property>
<property name="active">0</property>
<property name="has-entry">True</property>
<child internal-child="entry">
<object class="GtkEntry">
<property name="can-focus">False</property>
</object>
</child>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">False</property>
<property name="position">2</property>
</packing>
</child>
<child>
<object class="GtkLabel" id="epg_info_label">
<property name="visible" bind-source="epg_links_box" bind-property="visible" bind-flags="invert-boolean">True</property>
<property name="can-focus">False</property>
<property name="margin-start">5</property>
<property name="margin-end">5</property>
<property name="label" translatable="yes">Not found.</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">3</property>
</packing>
</child>
<child>
<object class="GtkBox" id="add_epg_src_box">
<property name="visible" bind-source="epg_links_box" bind-property="visible">False</property>
<property name="can-focus">False</property>
<property name="margin-start">5</property>
<property name="spacing">5</property>
<child>
<object class="GtkLabel" id="add_epg_src_label">
<property name="visible">True</property>
<property name="can-focus">False</property>
<property name="label" translatable="yes">Add to EPG sources list</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">0</property>
</packing>
</child>
<child>
<object class="GtkSwitch" id="add_epg_src_switch">
<property name="visible">True</property>
<property name="can-focus">True</property>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">1</property>
</packing>
</child>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="pack-type">end</property>
<property name="position">4</property>
</packing>
</child>
</object>
<packing>
<property name="expand">False</property>
<property name="fill">True</property>
<property name="position">2</property>
</packing>
</child>
</object>
</child>
<style>