Files
DemonEditor/app/ui/main_app_window.py

1605 lines
70 KiB
Python
Raw Normal View History

2017-11-13 00:21:52 +03:00
import os
import shutil
2017-11-14 19:20:16 +03:00
from contextlib import suppress
2017-11-30 00:45:52 +03:00
from functools import lru_cache
2017-11-13 00:21:52 +03:00
2018-08-18 10:21:40 +03:00
from gi.repository import GLib
from app.commons import run_idle, log, run_task, run_with_delay
2017-12-24 16:43:05 +03:00
from app.eparser import get_blacklist, write_blacklist, parse_m3u
2018-01-01 23:42:40 +03:00
from app.eparser import get_services, get_bouquets, write_bouquets, write_services, Bouquets, Bouquet, Service
from app.eparser.ecommons import CAS, Flag
2018-01-01 17:28:19 +03:00
from app.eparser.enigma.bouquets import BqServiceType
2018-02-11 23:14:22 +03:00
from app.eparser.neutrino.bouquets import BqType
2017-12-30 21:51:57 +03:00
from app.properties import get_config, write_config, Profile
2018-04-29 15:36:35 +03:00
from app.tools.media import Player
2018-10-29 18:46:46 +03:00
from .new_download_dialog import DownloadDialog
from .iptv import IptvDialog, SearchUnavailableDialog, IptvListConfigurationDialog
2018-03-06 19:06:16 +03:00
from .search import SearchProvider
2018-11-02 23:13:31 +03:00
from .uicommons import Gtk, Gdk, UI_RESOURCES_PATH, LOCKED_ICON, HIDE_ICON, IPTV_ICON, MOVE_KEYS, KEY_MAP
2018-03-06 19:06:16 +03:00
from .dialogs import show_dialog, DialogType, get_chooser_dialog, WaitDialog, get_message
2018-10-25 16:00:25 +03:00
from .main_helper import insert_marker, move_items, rename, ViewTarget, set_flags, locate_in_services, \
2018-08-18 10:21:40 +03:00
scroll_to, get_base_model, update_picons_data, copy_picon_reference, assign_picon, remove_picon, \
2018-11-02 23:13:31 +03:00
is_only_one_item_selected, gen_bouquets, BqGenType, get_iptv_url, append_picons, get_selection, get_model_data
from .picons_downloader import PiconsDialog
2017-11-09 19:01:09 +03:00
from .satellites_dialog import show_satellites_dialog
from .settings_dialog import show_settings_dialog
from .service_details_dialog import ServiceDetailsDialog, Action
2017-11-09 19:01:09 +03:00
class MainAppWindow:
2018-03-03 20:55:08 +03:00
_TV_TYPES = ("TV", "TV (HD)", "TV (UHD)", "TV (H264)")
2018-01-24 13:39:11 +03:00
_SERVICE_LIST_NAME = "services_list_store"
2018-03-03 20:55:08 +03:00
2018-01-24 13:39:11 +03:00
_FAV_LIST_NAME = "fav_list_store"
2018-03-03 20:55:08 +03:00
2018-01-24 13:39:11 +03:00
_BOUQUETS_LIST_NAME = "bouquets_tree_store"
2018-03-03 20:55:08 +03:00
2017-11-09 19:01:09 +03:00
# dynamically active elements depending on the selected view
_SERVICE_ELEMENTS = ("services_popup_menu",)
2018-01-08 00:11:07 +03:00
_BOUQUET_ELEMENTS = ("edit_tool_button", "new_tool_button", "bouquets_new_popup_item", "bouquets_edit_popup_item",
"bouquets_cut_popup_item", "bouquets_copy_popup_item", "bouquets_paste_popup_item")
2018-01-08 00:11:07 +03:00
_COMMONS_ELEMENTS = ("edit_tool_button", "bouquets_remove_popup_item", "fav_remove_popup_item")
2018-01-08 00:11:07 +03:00
_FAV_ELEMENTS = ("fav_cut_popup_item", "fav_paste_popup_item", "fav_locate_popup_item", "fav_iptv_popup_item",
2018-09-16 16:59:34 +03:00
"fav_insert_marker_popup_item", "fav_edit_sub_menu_popup_item", "fav_edit_popup_item",
2018-09-18 14:40:24 +03:00
"fav_picon_popup_item", "fav_copy_popup_item")
2018-01-08 00:11:07 +03:00
2018-04-29 01:44:28 +03:00
_FAV_ENIGMA_ELEMENTS = ("fav_insert_marker_popup_item",)
2018-02-11 23:14:22 +03:00
_FAV_IPTV_ELEMENTS = ("fav_iptv_popup_item",)
2018-01-08 00:11:07 +03:00
2017-11-26 20:40:22 +03:00
_LOCK_HIDE_ELEMENTS = ("locked_tool_button", "hide_tool_button")
2018-01-08 00:11:07 +03:00
_DYNAMIC_ELEMENTS = ("services_popup_menu", "new_tool_button", "edit_tool_button", "locked_tool_button",
"fav_cut_popup_item", "fav_paste_popup_item", "bouquets_new_popup_item", "hide_tool_button",
"bouquets_remove_popup_item", "fav_remove_popup_item", "bouquets_edit_popup_item",
2018-09-16 16:59:34 +03:00
"fav_insert_marker_popup_item", "fav_edit_popup_item", "fav_edit_sub_menu_popup_item",
"fav_locate_popup_item", "fav_picon_popup_item", "fav_iptv_popup_item", "fav_copy_popup_item",
"bouquets_cut_popup_item", "bouquets_copy_popup_item", "bouquets_paste_popup_item")
2017-11-09 19:01:09 +03:00
def __init__(self):
2018-07-08 00:09:26 +03:00
handlers = {"on_close_app": self.on_close_app,
2017-11-09 19:01:09 +03:00
"on_resize": self.on_resize,
"on_about_app": self.on_about_app,
"on_preferences": self.on_preferences,
"on_download": self.on_download,
"on_data_open": self.on_data_open,
"on_data_save": self.on_data_save,
2018-09-18 14:40:24 +03:00
"on_tree_view_key_press": self.on_tree_view_key_press,
2017-11-09 19:01:09 +03:00
"on_tree_view_key_release": self.on_tree_view_key_release,
"on_bouquets_selection": self.on_bouquets_selection,
"on_satellite_editor_show": self.on_satellite_editor_show,
"on_services_selection": self.on_services_selection,
"on_up": self.on_up,
"on_down": self.on_down,
"on_fav_cut": self.on_fav_cut,
"on_bouquets_cut": self.on_bouquets_cut,
2018-09-18 10:35:10 +03:00
"on_services_copy": self.on_services_copy,
"on_fav_copy": self.on_fav_copy,
"on_bouquets_copy": self.on_bouquets_copy,
"on_fav_paste": self.on_fav_paste,
"on_bouquets_paste": self.on_bouquets_paste,
"on_edit": self.on_rename,
"on_rename_for_bouquet": self.on_rename_for_bouquet,
"on_set_default_name_for_bouquet": self.on_set_default_name_for_bouquet,
"on_service_edit": self.on_service_edit,
"on_services_add_new": self.on_services_add_new,
2017-11-09 19:01:09 +03:00
"on_delete": self.on_delete,
2017-12-25 19:50:35 +03:00
"on_tool_edit": self.on_tool_edit,
"on_to_fav_copy": self.on_to_fav_copy,
"on_to_fav_end_copy": self.on_to_fav_end_copy,
"on_view_drag_begin": self.on_view_drag_begin,
2018-09-19 11:46:41 +03:00
"on_view_drag_data_get": self.on_view_drag_data_get,
"on_view_drag_data_received": self.on_view_drag_data_received,
"on_bq_view_drag_data_received": self.on_bq_view_drag_data_received,
"on_view_press": self.on_view_press,
2017-11-09 19:01:09 +03:00
"on_view_popup_menu": self.on_view_popup_menu,
2018-07-12 11:57:02 +03:00
"on_popover_release": self.on_popover_release,
2017-11-23 16:59:21 +03:00
"on_view_focus": self.on_view_focus,
"on_hide": self.on_hide,
2017-11-29 00:26:12 +03:00
"on_locked": self.on_locked,
2017-12-08 18:32:28 +03:00
"on_model_changed": self.on_model_changed,
2017-12-19 22:57:04 +03:00
"on_import_m3u": self.on_import_m3u,
"on_insert_marker": self.on_insert_marker,
2018-04-29 01:44:28 +03:00
"on_fav_press": self.on_fav_press,
2018-01-08 22:00:48 +03:00
"on_locate_in_services": self.on_locate_in_services,
2018-01-23 16:18:28 +03:00
"on_picons_loader_show": self.on_picons_loader_show,
2018-01-29 18:07:47 +03:00
"on_filter_changed": self.on_filter_changed,
"on_assign_picon": self.on_assign_picon,
"on_remove_picon": self.on_remove_picon,
2018-01-31 00:13:42 +03:00
"on_reference_picon": self.on_reference_picon,
2018-01-31 16:02:26 +03:00
"on_filter_toggled": self.on_filter_toggled,
2018-02-02 12:45:58 +03:00
"on_search_toggled": self.on_search_toggled,
"on_search_down": self.on_search_down,
"on_search_up": self.on_search_up,
2018-02-14 00:00:52 +03:00
"on_search": self.on_search,
"on_iptv": self.on_iptv,
"on_iptv_list_configuration": self.on_iptv_list_configuration,
2018-08-01 11:05:29 +03:00
"on_play_stream": self.on_play_stream,
2018-08-25 15:30:12 +03:00
"on_player_play": self.on_player_play,
"on_player_stop": self.on_player_stop,
"on_player_previous": self.on_player_previous,
"on_player_next": self.on_player_next,
2018-08-25 15:30:12 +03:00
"on_player_close": self.on_player_close,
"on_player_press": self.on_player_press,
"on_full_screen": self.on_full_screen,
2018-08-25 15:30:12 +03:00
"on_drawing_area_realize": self.on_drawing_area_realize,
2018-09-30 23:16:30 +03:00
"on_player_drawing_area_draw": self.on_player_drawing_area_draw,
"on_main_window_state": self.on_main_window_state,
"on_remove_all_unavailable": self.on_remove_all_unavailable,
"on_new_bouquet": self.on_new_bouquet,
"on_bouquets_edit": self.on_bouquets_edit,
"on_create_bouquet_for_current_satellite": self.on_create_bouquet_for_current_satellite,
"on_create_bouquet_for_each_satellite": self.on_create_bouquet_for_each_satellite,
"on_create_bouquet_for_current_package": self.on_create_bouquet_for_current_package,
2018-04-07 23:49:36 +03:00
"on_create_bouquet_for_each_package": self.on_create_bouquet_for_each_package,
"on_create_bouquet_for_current_type": self.on_create_bouquet_for_current_type,
"on_create_bouquet_for_each_type": self.on_create_bouquet_for_each_type}
2017-11-09 19:01:09 +03:00
self._options = get_config()
self._profile = self._options.get("profile")
os.makedirs(os.path.dirname(self._options.get(self._profile).get("data_dir_path")), exist_ok=True)
2017-11-09 19:01:09 +03:00
# Used for copy/paste. When adding the previous data will not be deleted.
# Clearing only after the insertion!
self._rows_buffer = []
self._bouquets_buffer = []
self._services = {}
self._bouquets = {}
# For bouquets with different names of services in bouquet and main list
self._extra_bouquets = {}
self._picons = {}
self._blacklist = set()
self._current_bq_name = None
# Current satellite positions in the services list
self._sat_positions = []
2018-04-29 15:36:35 +03:00
# Player
self._player = None
2018-05-19 16:24:20 +03:00
self._full_screen = False
self._drawing_area_xid = None
2017-11-09 19:01:09 +03:00
builder = Gtk.Builder()
2018-02-27 14:55:03 +03:00
builder.set_translation_domain("demon-editor")
2017-12-25 19:50:35 +03:00
builder.add_from_file(UI_RESOURCES_PATH + "main_window.glade")
2018-01-08 22:00:48 +03:00
builder.connect_signals(handlers)
self._main_window = builder.get_object("main_window")
main_window_size = self._options.get("window_size", None)
2017-11-09 19:01:09 +03:00
# Setting the last size of the window if it was saved
if main_window_size:
self._main_window.resize(*main_window_size)
self._services_view = builder.get_object("services_tree_view")
self._fav_view = builder.get_object("fav_tree_view")
self._bouquets_view = builder.get_object("bouquets_tree_view")
self._fav_model = builder.get_object("fav_list_store")
self._services_model = builder.get_object("services_list_store")
self._bouquets_model = builder.get_object("bouquets_tree_store")
2018-09-30 00:12:15 +03:00
self._main_data_box = builder.get_object("main_data_box")
2018-08-25 15:30:12 +03:00
self._player_drawing_area = builder.get_object("player_drawing_area")
2018-09-30 23:16:30 +03:00
self._player_box = builder.get_object("player_box")
2018-09-30 00:12:15 +03:00
self._player_tool_bar = builder.get_object("player_tool_bar")
self._player_prev_button = builder.get_object("player_prev_button")
self._player_next_button = builder.get_object("player_next_button")
2018-09-30 00:12:15 +03:00
self._status_bar_box = builder.get_object("status_bar_box")
self._services_main_box = builder.get_object("services_main_box")
self._bouquets_main_box = builder.get_object("bouquets_main_box")
# Enabling events for the drawing area
2018-08-25 15:30:12 +03:00
self._player_drawing_area.set_events(Gdk.ModifierType.BUTTON1_MASK)
self._player_frame = builder.get_object("player_frame")
2018-08-15 10:51:19 +03:00
self._header_bar = builder.get_object("header_bar")
self._bq_name_label = builder.get_object("bq_name_label")
self._ip_label = builder.get_object("ip_label")
self._ip_label.set_text(self._options.get(self._profile).get("host"))
2018-06-01 11:16:30 +03:00
self.update_profile_label()
# Dynamically active elements depending on the selected view
self._tool_elements = {k: builder.get_object(k) for k in self._DYNAMIC_ELEMENTS}
self._cas_label = builder.get_object("cas_label")
self._fav_count_label = builder.get_object("fav_count_label")
self._bouquets_count_label = builder.get_object("bouquets_count_label")
self._tv_count_label = builder.get_object("tv_count_label")
self._radio_count_label = builder.get_object("radio_count_label")
self._data_count_label = builder.get_object("data_count_label")
2017-11-09 19:01:09 +03:00
self.init_drag_and_drop() # drag and drop
2018-01-22 14:51:34 +03:00
# Force ctrl press event for view. Multiple selections in lists only with Space key(as in file managers)!!!
self._services_view.connect("key-press-event", self.force_ctrl)
self._fav_view.connect("key-press-event", self.force_ctrl)
2018-01-30 12:37:04 +03:00
# Clipboard
self._clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
2018-02-18 17:14:02 +03:00
# Wait dialog
self._wait_dialog = WaitDialog(self._main_window)
2018-03-06 19:06:16 +03:00
# Filter
self._services_model_filter = builder.get_object("services_model_filter")
self._services_model_filter.set_visible_func(self.services_filter_function)
self._filter_entry = builder.get_object("filter_entry")
2018-07-08 00:09:26 +03:00
self._filter_bar = builder.get_object("filter_bar")
self._filter_types_box = builder.get_object("filter_types_box")
self._filter_sat_positions_box = builder.get_object("filter_sat_positions_box")
self._filter_types_model = builder.get_object("filter_types_list_store")
self._filter_sat_positions_model = builder.get_object("filter_sat_positions_list_store")
self._filter_only_free_button = builder.get_object("filter_only_free_button")
2018-03-06 19:06:16 +03:00
# Search
2018-07-08 00:09:26 +03:00
self._search_bar = builder.get_object("search_bar")
self._search_provider = SearchProvider((self._services_view, self._fav_view, self._bouquets_view),
builder.get_object("search_down_button"),
builder.get_object("search_up_button"))
self._main_window.show()
2017-11-09 19:01:09 +03:00
def init_drag_and_drop(self):
2018-09-19 11:46:41 +03:00
""" Enable drag-and-drop """
2017-11-09 19:01:09 +03:00
target = []
2018-09-19 11:46:41 +03:00
bq_target = []
self._services_view.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, target, Gdk.DragAction.COPY)
self._fav_view.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, target,
Gdk.DragAction.DEFAULT | Gdk.DragAction.MOVE)
self._fav_view.enable_model_drag_dest(target, Gdk.DragAction.DEFAULT | Gdk.DragAction.MOVE)
2018-09-19 11:46:41 +03:00
self._bouquets_view.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, bq_target,
Gdk.DragAction.DEFAULT | Gdk.DragAction.MOVE)
self._bouquets_view.enable_model_drag_dest(bq_target, Gdk.DragAction.DEFAULT | Gdk.DragAction.MOVE)
self._fav_view.drag_dest_set_target_list(None)
self._fav_view.drag_source_set_target_list(None)
self._fav_view.drag_dest_add_text_targets()
self._fav_view.drag_source_add_text_targets()
self._services_view.drag_source_set_target_list(None)
self._services_view.drag_source_add_text_targets()
2018-09-19 11:46:41 +03:00
self._bouquets_view.drag_dest_set_target_list(None)
self._bouquets_view.drag_source_set_target_list(None)
self._bouquets_view.drag_dest_add_text_targets()
self._bouquets_view.drag_source_add_text_targets()
2017-11-09 19:01:09 +03:00
2018-01-22 14:51:34 +03:00
def force_ctrl(self, view, event):
""" Function for force ctrl press event for view """
event.state |= Gdk.ModifierType.CONTROL_MASK
2018-04-29 15:36:35 +03:00
@run_idle
2018-07-08 00:09:26 +03:00
def on_close_app(self, *args):
2017-11-09 19:01:09 +03:00
""" Called before app quit """
write_config(self._options) # storing current config
2018-08-25 15:30:12 +03:00
self.on_player_close()
2017-11-09 19:01:09 +03:00
Gtk.main_quit()
def on_resize(self, window):
""" Stores new size properties for app window after resize """
self._options["window_size"] = window.get_size()
2017-11-09 19:01:09 +03:00
def on_up(self, item):
self.move_items(Gdk.KEY_Up)
def on_down(self, item):
self.move_items(Gdk.KEY_Down)
2018-07-12 11:57:02 +03:00
@run_idle
2017-11-09 19:01:09 +03:00
def on_about_app(self, item):
show_dialog(DialogType.ABOUT, self._main_window)
2017-11-09 19:01:09 +03:00
def move_items(self, key):
2017-12-25 19:50:35 +03:00
""" Move items in fav or bouquets tree view """
if self._services_view.is_focus():
2017-12-25 19:50:35 +03:00
return
move_items(key, self._fav_view if self._fav_view.is_focus() else self._bouquets_view)
2017-11-09 19:01:09 +03:00
# ***************** Copy - Cut - Paste *********************#
2018-09-18 10:35:10 +03:00
def on_services_copy(self, view):
self.on_copy(view, target=ViewTarget.FAV)
def on_fav_copy(self, view):
self.on_copy(view, target=ViewTarget.SERVICES)
def on_bouquets_copy(self, view):
self.on_copy(view, target=ViewTarget.BOUQUET)
def on_copy(self, view, target):
2017-11-09 19:01:09 +03:00
model, paths = view.get_selection().get_selected_rows()
2018-09-18 10:35:10 +03:00
if target is ViewTarget.FAV:
self._rows_buffer.extend((0, *model.get(model.get_iter(path), 2, 3, 4, 5, 7, 16, 18, 8)) for path in paths)
2018-09-18 10:35:10 +03:00
elif target is ViewTarget.SERVICES:
self._rows_buffer.extend(model[path][:] for path in paths)
2018-09-18 14:40:24 +03:00
elif target is ViewTarget.BOUQUET:
to_copy = list(map(model.get_iter, filter(lambda p: p.get_depth() == 2, paths)))
if to_copy:
self._bouquets_buffer.extend([model[i][:] for i in to_copy])
2017-11-09 19:01:09 +03:00
def on_fav_cut(self, view):
self.on_cut(view, ViewTarget.FAV)
def on_bouquets_cut(self, view):
self.on_cut(view, ViewTarget.BOUQUET)
def on_cut(self, view, target=None):
if target is ViewTarget.FAV:
for row in tuple(self.on_delete(view)):
self._rows_buffer.append(row)
elif target is ViewTarget.BOUQUET:
model, paths = view.get_selection().get_selected_rows()
to_cut = list(map(model.get_iter, filter(lambda p: p.get_depth() == 2, paths)))
if to_cut:
self._bouquets_buffer.extend([model[i][:] for i in to_cut])
list(map(model.remove, to_cut))
def on_fav_paste(self, view):
self.on_paste(view, ViewTarget.FAV)
def on_bouquets_paste(self, view):
self.on_paste(view, ViewTarget.BOUQUET)
def on_paste(self, view, target):
2017-11-09 19:01:09 +03:00
selection = view.get_selection()
if target is ViewTarget.FAV:
self.fav_paste(selection)
elif target is ViewTarget.BOUQUET:
self.bouquet_paste(selection)
self.on_view_focus(view, None)
def fav_paste(self, selection):
2017-11-09 19:01:09 +03:00
dest_index = 0
2018-04-10 11:15:50 +03:00
bq_selected = self.check_bouquet_selection()
2017-11-09 19:01:09 +03:00
if not bq_selected:
return
fav_bouquet = self._bouquets[bq_selected]
2017-11-09 19:01:09 +03:00
model, paths = selection.get_selected_rows()
if paths:
dest_index = int(paths[0][0])
for row in self._rows_buffer:
2017-11-09 19:01:09 +03:00
dest_index += 1
model.insert(dest_index, row)
2018-10-10 22:46:36 +03:00
fav_bouquet.insert(dest_index, row[7])
2017-11-09 19:01:09 +03:00
2018-01-24 13:39:11 +03:00
if model.get_name() == self._FAV_LIST_NAME:
2017-11-09 19:01:09 +03:00
self.update_fav_num_column(model)
self._rows_buffer.clear()
def bouquet_paste(self, selection):
model, paths = selection.get_selected_rows()
if len(paths) > 1:
show_dialog(DialogType.ERROR, self._main_window, "Please, select only one item!")
return
path = paths[0]
dest_iter = model.get_iter(path)
if path.get_depth() == 1:
list(map(lambda r: model.append(dest_iter, r), self._bouquets_buffer))
self._bouquets_view.expand_all()
else:
p_iter = model.iter_parent(dest_iter)
dest_index = path.get_indices()[1] + 1
for index, row in enumerate(self._bouquets_buffer):
model.insert(p_iter, dest_index + index, row)
self._bouquets_buffer.clear()
self.update_bouquets_type()
# ***************** Deletion *********************#
2017-11-09 19:01:09 +03:00
def on_delete(self, view):
""" Delete selected items from view
2017-11-09 19:01:09 +03:00
returns deleted rows list!
"""
selection = view.get_selection()
model, paths = selection.get_selected_rows()
model_name = get_base_model(model).get_name()
itrs = [model.get_iter(path) for path in paths]
rows = [model[in_itr][:] for in_itr in itrs]
if model_name == self._FAV_LIST_NAME:
2018-10-13 10:48:39 +03:00
next(self.remove_favs(itrs, model), False)
elif model_name == self._BOUQUETS_LIST_NAME:
self.delete_bouquets(itrs, model)
elif model_name == self._SERVICE_LIST_NAME:
2018-10-13 10:48:39 +03:00
next(self.delete_services(itrs, model, rows), False)
self.on_view_focus(view, None)
return rows
2017-11-09 19:01:09 +03:00
def remove_favs(self, itrs, model):
""" Deleting bouquet services """
bq_selected = self.get_selected_bouquet()
if bq_selected:
fav_bouquet = self._bouquets.get(bq_selected, None)
if fav_bouquet:
for itr in itrs:
del fav_bouquet[int(model.get_path(itr)[0])]
self._fav_model.remove(itr)
self.update_fav_num_column(model)
2018-10-13 10:48:39 +03:00
yield True
def delete_services(self, itrs, model, rows):
2017-11-16 01:24:16 +03:00
""" Deleting services """
srv_itrs = [self._services_model_filter.convert_iter_to_child_iter(
model.convert_iter_to_child_iter(itr)) for itr in itrs]
for s_itr in srv_itrs:
self._services_model.remove(s_itr)
2018-04-06 17:57:04 +03:00
srv_ids_to_delete = set()
2017-11-16 01:24:16 +03:00
for row in rows:
# There are channels with the same parameters except for the name.
# None because it can have duplicates! Need fix
fav_id = row[-2]
for bq in self._bouquets:
services = self._bouquets[bq]
2017-11-26 14:55:57 +03:00
if services:
with suppress(ValueError):
services.remove(fav_id)
2018-04-06 17:57:04 +03:00
srv_ids_to_delete.add(fav_id)
self._services.pop(fav_id, None)
2017-11-16 01:24:16 +03:00
2018-04-06 17:57:04 +03:00
for f_itr in filter(lambda r: r[7] in srv_ids_to_delete, self._fav_model):
self._fav_model.remove(f_itr.iter)
self.update_fav_num_column(self._fav_model)
self.update_sat_positions()
2018-10-13 10:48:39 +03:00
yield True
2017-11-16 01:24:16 +03:00
2018-09-21 11:15:20 +03:00
def delete_bouquets(self, itrs, model):
""" Deleting bouquets """
if len(itrs) == 1 and len(model.get_path(itrs[0])) < 2:
show_dialog(DialogType.ERROR, self._main_window, "This item is not allowed to be removed!")
return
for itr in itrs:
if len(model.get_path(itr)) < 2:
continue
2018-09-21 11:09:40 +03:00
self._fav_model.clear()
self._bouquets_model.remove(itr)
2018-01-05 14:32:14 +03:00
# ***************** ####### *********************#
2018-01-05 14:32:14 +03:00
def get_bouquet_file_name(self, bouquet):
bouquet_file_name = "{}userbouquet.{}.{}".format(self._options.get(self._profile).get("data_dir_path"),
2018-01-05 14:32:14 +03:00
*bouquet.split(":"))
return bouquet_file_name
2017-11-14 19:20:16 +03:00
2017-11-09 19:01:09 +03:00
def on_new_bouquet(self, view):
""" Creates a new item in the bouquets tree """
model, paths = view.get_selection().get_selected_rows()
if paths:
itr = model.get_iter(paths[0])
bq_type = model.get_value(itr, 3)
2017-11-09 19:01:09 +03:00
bq_name = "bouquet"
count = 0
key = "{}:{}".format(bq_name, bq_type)
# Generating name of new bouquet
while key in self._bouquets:
2017-11-09 19:01:09 +03:00
count += 1
bq_name = "bouquet{}".format(count)
key = "{}:{}".format(bq_name, bq_type)
response = show_dialog(DialogType.INPUT, self._main_window, bq_name)
2017-11-09 19:01:09 +03:00
if response == Gtk.ResponseType.CANCEL:
return
2018-01-05 14:53:53 +03:00
bq = response, None, None, bq_type
key = "{}:{}".format(response, bq_type)
self._current_bq_name = response
2017-11-09 19:01:09 +03:00
if model.iter_n_children(itr): # parent
2017-11-18 20:49:53 +03:00
ch_itr = model.insert(itr, 0, bq)
2017-12-25 19:50:35 +03:00
scroll_to(model.get_path(ch_itr), view, paths)
2017-11-09 19:01:09 +03:00
else:
2017-11-18 20:49:53 +03:00
p_itr = model.iter_parent(itr)
it = model.insert(p_itr, int(model.get_path(itr)[1]) + 1, bq) if p_itr else model.append(itr, bq)
2017-12-25 19:50:35 +03:00
scroll_to(model.get_path(it), view, paths)
self._bouquets[key] = []
2017-11-09 19:01:09 +03:00
2017-12-25 19:50:35 +03:00
def on_tool_edit(self, item):
""" Edit tool bar button """
if self._services_view.is_focus():
self.on_service_edit(self._services_view)
elif self._fav_view.is_focus():
self.on_service_edit(self._fav_view)
elif self._bouquets_view.is_focus():
self.on_rename(self._bouquets_view)
2017-11-18 20:49:53 +03:00
def on_to_fav_copy(self, view):
""" Copy items from main to beginning of fav list """
2017-11-09 19:01:09 +03:00
selection = self.get_selection(view)
if selection:
self.receive_selection(view=self._fav_view, drop_info=None, data=selection)
2017-11-09 19:01:09 +03:00
def on_to_fav_end_copy(self, view):
""" Copy items from main to end of fav list """
selection = self.get_selection(view)
if selection:
pos = Gtk.TreeViewDropPosition.AFTER
path = Gtk.TreePath.new()
mod_len = len(self._fav_model)
2018-10-16 14:12:07 +03:00
info = None
if mod_len > 0:
path.append_index(mod_len - 1)
info = (path, pos)
self.receive_selection(view=self._fav_view, drop_info=info, data=selection)
if mod_len > 0:
scroll_to(mod_len, self._fav_view)
2018-10-13 10:48:39 +03:00
@run_with_delay(1)
2018-09-19 11:46:41 +03:00
def update_fav_num_column(self, model):
""" Iterate through model and updates values for Num column """
gen = self.update_num_column(model)
GLib.idle_add(lambda: next(gen, False), priority=GLib.PRIORITY_LOW)
def update_num_column(self, model):
for index, row in enumerate(model):
row[0] = index + 1
2018-10-13 10:48:39 +03:00
yield True
2018-09-19 11:46:41 +03:00
def update_bouquet_list(self):
""" Update bouquet after move items """
bq_selected = self.get_selected_bouquet()
if bq_selected:
fav_bouquet = self._bouquets[bq_selected]
fav_bouquet.clear()
for row in self._fav_model:
fav_bouquet.append(row[7])
# ***************** Drag-and-drop *********************#
def on_view_drag_begin(self, view, context):
""" Selects a row under the cursor in the view at the dragging beginning. """
selection = view.get_selection()
if selection.count_selected_rows() > 1:
view.do_toggle_cursor_row(view)
2018-09-19 11:46:41 +03:00
def on_view_drag_data_get(self, view, drag_context, data, info, time):
selection = self.get_selection(view)
if selection:
data.set_text(selection, -1)
2018-09-19 11:46:41 +03:00
def on_view_drag_data_received(self, view, drag_context, x, y, data, info, time):
self.receive_selection(view=view, drop_info=view.get_dest_row_at_pos(x, y), data=data.get_text())
return False
2018-09-19 11:46:41 +03:00
def on_bq_view_drag_data_received(self, view, drag_context, x, y, data, info, time):
2018-11-02 23:13:31 +03:00
model_name, model = get_model_data(view)
2018-09-19 11:46:41 +03:00
drop_info = view.get_dest_row_at_pos(x, y)
data = data.get_text()
if not data:
return
2018-09-19 11:46:41 +03:00
itr_str, sep, source = data.partition("::::")
if source != self._BOUQUETS_LIST_NAME:
return
if drop_info:
path, position = drop_info
itrs = [model.get_iter_from_string(itr) for itr in itr_str.split(",")]
top_iter = model.get_iter(path)
parent_itr = model.iter_parent(top_iter) # parent
to_del = []
if parent_itr:
p_path = model.get_path(parent_itr)[0]
for itr in itrs:
p_itr = model.iter_parent(itr)
if not p_itr:
break
if p_itr and model.get_path(p_itr)[0] == p_path:
top_iter = model.move_before(itr, top_iter)
else:
model.insert(parent_itr, model.get_path(top_iter)[1], model[itr][:])
to_del.append(itr)
elif not model.iter_has_child(top_iter):
for itr in itrs:
model.append(top_iter, model[itr][:])
to_del.append(itr)
view.expand_all()
list(map(model.remove, to_del))
self.update_bouquets_type()
2018-09-19 11:46:41 +03:00
2017-11-09 19:01:09 +03:00
def get_selection(self, view):
""" Creates a string from the iterators of the selected rows """
2017-11-23 16:59:21 +03:00
model, paths = view.get_selection().get_selected_rows()
2018-02-06 14:20:59 +03:00
model = get_base_model(model)
2017-11-09 19:01:09 +03:00
if len(paths) > 0:
itrs = [model.get_iter(path) for path in paths]
2018-09-19 11:46:41 +03:00
return "{}::::{}".format(",".join([model.get_string_from_iter(itr) for itr in itrs]), model.get_name())
2017-11-09 19:01:09 +03:00
def receive_selection(self, *, view, drop_info, data):
""" Update fav view after data received """
try:
itr_str, sep, source = data.partition("::::")
if source == self._BOUQUETS_LIST_NAME:
return
2018-09-19 11:46:41 +03:00
bq_selected = self.check_bouquet_selection()
if not bq_selected:
return
model = get_base_model(view.get_model())
dest_index = 0
if drop_info:
path, position = drop_info
dest_iter = model.get_iter(path)
if dest_iter:
dest_index = model.get_value(dest_iter, 0)
fav_bouquet = self._bouquets[bq_selected]
2018-09-19 11:46:41 +03:00
itrs = itr_str.split(",")
2017-11-09 19:01:09 +03:00
2018-01-24 13:39:11 +03:00
if source == self._SERVICE_LIST_NAME:
ext_model = self._services_view.get_model()
2017-11-09 19:01:09 +03:00
ext_itrs = [ext_model.get_iter_from_string(itr) for itr in itrs]
2018-01-24 13:39:11 +03:00
ext_rows = [ext_model[ext_itr][:] for ext_itr in ext_itrs]
2017-11-09 19:01:09 +03:00
dest_index -= 1
for ext_row in ext_rows:
dest_index += 1
2018-02-06 14:20:59 +03:00
fav_id = ext_row[18]
ch = self._services[fav_id]
2018-02-06 14:20:59 +03:00
model.insert(dest_index, (0, ch.coded, ch.service, ch.locked, ch.hide,
ch.service_type, ch.pos, ch.fav_id, self._picons.get(ch.picon_id, None)))
2018-02-06 14:20:59 +03:00
fav_bouquet.insert(dest_index, ch.fav_id)
2018-01-24 13:39:11 +03:00
elif source == self._FAV_LIST_NAME:
2017-11-09 19:01:09 +03:00
in_itrs = [model.get_iter_from_string(itr) for itr in itrs]
in_rows = [model[in_itr][:] for in_itr in in_itrs]
2017-11-09 19:01:09 +03:00
for row in in_rows:
model.insert(dest_index, row)
2018-02-06 14:20:59 +03:00
fav_bouquet.insert(dest_index, row[7])
2017-11-09 19:01:09 +03:00
for in_itr in in_itrs:
del fav_bouquet[int(model.get_path(in_itr)[0])]
model.remove(in_itr)
self.update_fav_num_column(model)
except ValueError as e:
2018-09-19 11:46:41 +03:00
show_dialog(DialogType.ERROR, self._main_window, str(e))
2017-11-09 19:01:09 +03:00
def on_view_press(self, view, event):
if event.get_event_type() == Gdk.EventType.BUTTON_PRESS and event.button == Gdk.BUTTON_PRIMARY:
2018-11-02 23:13:31 +03:00
name, model = get_model_data(view)
self.delete_views_selection(name)
def delete_views_selection(self, name):
if name == self._SERVICE_LIST_NAME:
self.delete_selection(self._fav_view)
elif name == self._FAV_LIST_NAME:
self.delete_selection(self._services_view)
elif name == self._BOUQUETS_LIST_NAME:
self.delete_selection(self._services_view, self._fav_view)
2017-11-09 19:01:09 +03:00
def on_view_popup_menu(self, menu, event):
""" Shows popup menu for any view """
if event.get_event_type() == Gdk.EventType.BUTTON_PRESS and event.button == Gdk.BUTTON_SECONDARY:
name = Gtk.Buildable.get_name(menu)
if name == "services_popup_menu":
self.delete_selection(self._fav_view, self._bouquets_view)
self.on_view_focus(self._services_view, None)
elif name == "fav_popup_menu":
self.delete_selection(self._services_view, self._bouquets_view)
self.on_view_focus(self._fav_view, None)
elif name == "bouquets_popup_menu":
self.delete_selection(self._services_view, self._fav_view)
self.on_view_focus(self._bouquets_view, None)
2017-11-09 19:01:09 +03:00
menu.popup(None, None, None, None, event.button, event.time)
return True
2017-11-09 19:01:09 +03:00
2018-07-12 11:57:02 +03:00
def on_popover_release(self, menu, event):
""" Hides popover after mouse click. Used if element of Popover menu is Gtk.Button! """
menu.hide()
2018-07-12 11:57:02 +03:00
2018-01-05 14:32:14 +03:00
@run_idle
2017-11-09 19:01:09 +03:00
def on_satellite_editor_show(self, model):
""" Shows satellites editor dialog """
show_satellites_dialog(self._main_window, self._options.get(self._profile))
2017-11-09 19:01:09 +03:00
2018-07-12 11:57:02 +03:00
@run_idle
2017-11-09 19:01:09 +03:00
def on_data_open(self, model):
response = show_dialog(DialogType.CHOOSER, self._main_window, options=self._options.get(self._profile))
2018-01-11 17:59:59 +03:00
if response in (Gtk.ResponseType.CANCEL, Gtk.ResponseType.DELETE_EVENT):
2017-11-09 19:01:09 +03:00
return
2017-12-12 09:16:58 +03:00
self.open_data(response)
2017-11-09 19:01:09 +03:00
2017-12-12 09:16:58 +03:00
def open_data(self, data_path=None):
2017-11-09 19:01:09 +03:00
""" Opening data and fill views. """
self._wait_dialog.show()
self.clear_current_data()
2018-08-18 11:35:44 +03:00
GLib.idle_add(self.append_data, data_path, priority=GLib.PRIORITY_LOW)
2017-12-12 09:16:58 +03:00
2018-08-18 10:21:40 +03:00
def append_data(self, data_path):
2018-08-18 11:35:44 +03:00
profile = Profile(self._profile)
data_path = self._options.get(self._profile).get("data_dir_path") if data_path is None else data_path
2018-08-18 11:35:44 +03:00
2017-11-09 19:01:09 +03:00
try:
2018-08-18 11:35:44 +03:00
black_list = get_blacklist(data_path)
bouquets = get_bouquets(data_path, Profile(self._profile))
services = get_services(data_path, profile, self.get_format_version() if profile is Profile.ENIGMA_2 else 0)
2018-08-18 12:31:12 +03:00
update_picons_data(self._options.get(self._profile).get("picons_dir_path"), self._picons)
2017-11-09 19:01:09 +03:00
except FileNotFoundError as e:
2018-08-18 10:21:40 +03:00
self._wait_dialog.hide()
show_dialog(DialogType.ERROR, self._main_window, getattr(e, "message", str(e)) + "\n\n" +
2018-03-06 19:06:16 +03:00
get_message("Please, download files from receiver or setup your path for read data!"))
2017-12-20 16:46:15 +03:00
except SyntaxError as e:
self._wait_dialog.hide()
2018-08-18 10:21:40 +03:00
show_dialog(DialogType.ERROR, self._main_window, str(e))
2018-08-18 11:35:44 +03:00
except Exception as e:
self._wait_dialog.hide()
log("Append services error: " + str(e))
show_dialog(DialogType.ERROR, self._main_window, "Reading data error!\n" + str(e))
else:
self.append_blacklist(black_list)
self.append_bouquets(bouquets)
self.append_services(services)
self.update_sat_positions()
2018-08-18 11:35:44 +03:00
self.update_services_counts(len(self._services.values()))
2017-11-09 19:01:09 +03:00
2018-08-18 11:35:44 +03:00
def append_blacklist(self, black_list):
2017-11-25 15:55:24 +03:00
if black_list:
self._blacklist.update(black_list)
2017-11-25 15:55:24 +03:00
2018-08-18 11:35:44 +03:00
def append_bouquets(self, bqs):
for bouquet in bqs:
parent = self._bouquets_model.append(None, [bouquet.name, None, None, bouquet.type])
for bq in bouquet.bouquets:
self.append_bouquet(bq, parent)
def append_bouquet(self, bq, parent):
name, bt_type, locked, hidden = bq.name, bq.type, bq.locked, bq.hidden
self._bouquets_model.append(parent, [name, locked, hidden, bt_type])
bq_id = "{}:{}".format(name, bt_type)
services = []
extra_services = {} # for services with different names in bouquet and main list
2018-09-01 00:49:11 +03:00
agr = [None] * 7
for srv in bq.services:
fav_id = srv.data
# IPTV and MARKER services
s_type = srv.type
if s_type is BqServiceType.MARKER or s_type is BqServiceType.IPTV:
2018-09-01 00:49:11 +03:00
icon = None
picon_id = None
if s_type is BqServiceType.IPTV:
icon = IPTV_ICON
id_data = fav_id.lstrip().split(":")
picon_id = "{}_{}_{}_{}_{}_{}_{}_{}_{}_{}.png".format(*id_data[0:10])
srv = Service(*agr[0:2], icon, srv.name, *agr[0:3], s_type.name, self._picons.get(picon_id, None),
picon_id, *agr, srv.num, fav_id, None)
self._services[fav_id] = srv
elif srv.name:
extra_services[fav_id] = srv.name
services.append(fav_id)
self._bouquets[bq_id] = services
if extra_services:
self._extra_bouquets[bq_id] = extra_services
2017-11-14 19:20:16 +03:00
2018-08-18 11:35:44 +03:00
def append_services(self, services):
2018-09-21 10:16:30 +03:00
for srv in services:
# adding channels to dict with fav_id as keys
self._services[srv.fav_id] = srv
gen = self.append_services_data(services)
GLib.idle_add(lambda: next(gen, False), priority=GLib.PRIORITY_LOW)
2018-08-18 10:21:40 +03:00
def append_services_data(self, services):
for srv in services:
2018-08-18 12:31:12 +03:00
itr = self._services_model.append(srv)
self._services_model.set_value(itr, 8, self._picons.get(srv.picon_id, None))
2018-08-18 10:21:40 +03:00
yield True
self._wait_dialog.hide()
2017-11-14 19:20:16 +03:00
def clear_current_data(self):
""" Clearing current data from lists """
self._bouquets_model.clear()
self._fav_model.clear()
self._services_model.clear()
self._blacklist.clear()
self._services.clear()
self._rows_buffer.clear()
self._bouquets.clear()
self._extra_bouquets.clear()
2018-07-12 11:57:02 +03:00
self._current_bq_name = None
self._bq_name_label.set_text("")
self.init_sat_positions()
2018-07-12 11:57:02 +03:00
@run_idle
2017-11-09 19:01:09 +03:00
def on_data_save(self, *args):
if show_dialog(DialogType.QUESTION, self._main_window) == Gtk.ResponseType.CANCEL:
2017-11-09 19:01:09 +03:00
return
profile = Profile(self._profile)
path = self._options.get(self._profile).get("data_dir_path")
2018-02-01 14:02:59 +03:00
backup_path = path + "backup/"
os.makedirs(os.path.dirname(backup_path), exist_ok=True)
# backup files in data dir(skipping dirs and satellites.xml)
for file in filter(lambda f: f != "satellites.xml" and os.path.isfile(os.path.join(path, f)), os.listdir(path)):
shutil.move(os.path.join(path, file), backup_path + file)
2017-11-09 19:01:09 +03:00
bouquets = []
def parse_bouquets(model, b_path, itr):
bqs = None
2017-11-09 19:01:09 +03:00
if model.iter_has_child(itr):
bqs = []
num_of_children = model.iter_n_children(itr)
2017-11-09 19:01:09 +03:00
for num in range(num_of_children):
bq_itr = model.iter_nth_child(itr, num)
2018-01-05 14:32:14 +03:00
bq_name, locked, hidden, bq_type = model.get(bq_itr, 0, 1, 2, 3)
bq_id = "{}:{}".format(bq_name, bq_type)
favs = self._bouquets[bq_id]
ex_s = self._extra_bouquets.get(bq_id)
bq_s = list(filter(None, [self._services.get(f_id, None) for f_id in favs]))
if profile is Profile.ENIGMA_2:
bq_s = list(map(lambda s: s._replace(service=ex_s.get(s.fav_id, None) if ex_s else None), bq_s))
bq = Bouquet(bq_name, bq_type, bq_s, locked, hidden)
2017-11-09 19:01:09 +03:00
bqs.append(bq)
if len(b_path) == 1:
bouquets.append(Bouquets(*model.get(itr, 0, 3), bqs if bqs else []))
2017-11-26 20:40:22 +03:00
profile = Profile(self._profile)
2017-11-09 19:01:09 +03:00
# Getting bouquets
self._bouquets_view.get_model().foreach(parse_bouquets)
2018-01-04 20:58:22 +03:00
write_bouquets(path, bouquets, profile)
2017-11-09 19:01:09 +03:00
# Getting services
services_model = get_base_model(self._services_view.get_model())
2018-01-01 23:42:40 +03:00
services = [Service(*row[:]) for row in services_model]
2018-06-01 17:45:26 +03:00
write_services(path, services, profile, self.get_format_version() if profile is Profile.ENIGMA_2 else 0)
2018-01-04 20:58:22 +03:00
# removing bouquet files
2018-02-10 15:49:44 +03:00
if profile is Profile.ENIGMA_2:
2018-01-04 20:58:22 +03:00
# blacklist
write_blacklist(path, self._blacklist)
2017-11-09 19:01:09 +03:00
def on_services_selection(self, model, path, column):
2017-11-23 16:59:21 +03:00
self.update_service_bar(model, path)
def update_service_bar(self, model, path):
def_val = "Unknown"
2017-12-08 18:32:28 +03:00
cas = model.get_value(model.get_iter(path), 0)
if not cas:
return
cas_values = list(filter(lambda val: val.startswith("C:"), cas.split(",")))
self._cas_label.set_text(",".join(map(str, sorted(set(CAS.get(val, def_val) for val in cas_values)))))
2017-11-09 19:01:09 +03:00
def on_bouquets_selection(self, model, path, column):
self._current_bq_name = model[path][0] if len(path) > 1 else None
self._bq_name_label.set_text(self._current_bq_name if self._current_bq_name else "")
self._fav_model.clear()
2017-11-09 19:01:09 +03:00
if self._bouquets_view.row_expanded(path):
self._bouquets_view.collapse_row(path)
2017-11-09 19:01:09 +03:00
else:
self._bouquets_view.expand_row(path, column)
2017-11-09 19:01:09 +03:00
if len(path) > 1:
2018-10-13 10:48:39 +03:00
next(self.update_bouquet_services(model, path), False)
2017-11-09 19:01:09 +03:00
def update_bouquet_services(self, model, path, bq_key=None):
""" Updates list of bouquet services """
2017-11-09 19:01:09 +03:00
tree_iter = None
if path:
tree_iter = model.get_iter(path)
2018-01-05 14:32:14 +03:00
key = bq_key if bq_key else "{}:{}".format(*model.get(tree_iter, 0, 3))
services = self._bouquets.get(key, None)
ex_services = self._extra_bouquets.get(key, None)
if not services:
return
2017-11-09 19:01:09 +03:00
for num, srv_id in enumerate(services):
srv = self._services.get(srv_id, None)
ex_srv_name = None
if ex_services:
ex_srv_name = ex_services.get(srv_id)
if srv:
self._fav_model.append((num + 1, srv.coded, ex_srv_name if ex_srv_name else srv.service, srv.locked,
srv.hide, srv.service_type, srv.pos, srv.fav_id,
self._picons.get(srv.picon_id, None)))
2018-10-13 10:48:39 +03:00
yield True
2017-11-09 19:01:09 +03:00
2018-04-10 11:15:50 +03:00
def check_bouquet_selection(self):
""" checks and returns bouquet if selected """
bq_selected = self.get_selected_bouquet()
if not bq_selected:
show_dialog(DialogType.ERROR, self._main_window, "Error. No bouquet is selected!")
return
if Profile(self._profile) is Profile.NEUTRINO_MP and bq_selected.endswith(BqType.WEBTV.value):
show_dialog(DialogType.ERROR, self._main_window, "Operation not allowed in this context!")
return
return bq_selected
def get_selected_bouquet(self):
""" returns 'name:type' of last selected bouquet or False """
if self._current_bq_name is None:
2017-11-09 19:01:09 +03:00
return False
for row in self._bouquets_model:
chs_rows = row.iterchildren()
for ch_row in chs_rows:
name = ch_row[0]
if name == self._current_bq_name:
return "{}:{}".format(name, ch_row[3])
return False
2017-11-09 19:01:09 +03:00
@run_idle
def update_bouquets_type(self):
""" Update bouquets type in the model and dict """
for row in get_base_model(self._bouquets_view.get_model()):
bqs_rows = row.iterchildren()
if bqs_rows:
bq_type = row[-1]
for b_row in bqs_rows:
bq_id = "{}:{}".format(b_row[0], b_row[-1])
bq = self._bouquets.get(bq_id, None)
if bq:
b_row[-1] = bq_type
self._bouquets["{}:{}".format(b_row[0], b_row[-1])] = bq
2017-11-09 19:01:09 +03:00
def delete_selection(self, view, *args):
""" Used for clear selection on given view(s) """
for v in [view, *args]:
v.get_selection().unselect_all()
2018-01-18 00:57:58 +03:00
@run_idle
2017-11-09 19:01:09 +03:00
def on_preferences(self, item):
response = show_settings_dialog(self._main_window, self._options)
2017-12-30 21:51:57 +03:00
if response != Gtk.ResponseType.CANCEL:
profile = self._options.get("profile")
self._ip_label.set_text(self._options.get(profile).get("host"))
2018-06-01 11:16:30 +03:00
if profile != self._profile:
self._profile = profile
self.clear_current_data()
2018-01-06 15:24:56 +03:00
self.update_services_counts()
2017-11-09 19:01:09 +03:00
2018-06-01 11:16:30 +03:00
self.update_profile_label()
2018-09-18 14:40:24 +03:00
def on_tree_view_key_press(self, view, event):
""" Handling keystrokes on press """
key = event.keyval
2018-11-02 23:13:31 +03:00
map_key = KEY_MAP.get(key, None)
key = map_key if map_key else key
2018-09-18 14:40:24 +03:00
ctrl = event.state & Gdk.ModifierType.CONTROL_MASK
2018-11-02 23:13:31 +03:00
model_name, model = get_model_data(view)
2018-09-18 14:40:24 +03:00
if ctrl and key == Gdk.KEY_c or key == Gdk.KEY_C:
if model_name == self._SERVICE_LIST_NAME:
self.on_copy(view, ViewTarget.FAV)
elif model_name == self._FAV_LIST_NAME:
self.on_copy(view, ViewTarget.SERVICES)
else:
self.on_copy(view, ViewTarget.BOUQUET)
elif ctrl and key == Gdk.KEY_x or key == Gdk.KEY_X:
if model_name == self._FAV_LIST_NAME:
self.on_cut(view, ViewTarget.FAV)
elif model_name == self._BOUQUETS_LIST_NAME:
self.on_cut(view, ViewTarget.BOUQUET)
2018-09-18 14:40:24 +03:00
elif ctrl and key == Gdk.KEY_v or key == Gdk.KEY_V:
if model_name == self._FAV_LIST_NAME:
self.on_paste(view, ViewTarget.FAV)
elif model_name == self._BOUQUETS_LIST_NAME:
self.on_paste(view, ViewTarget.BOUQUET)
2018-09-18 14:40:24 +03:00
elif key == Gdk.KEY_Delete:
self.on_delete(view)
2017-11-09 19:01:09 +03:00
def on_tree_view_key_release(self, view, event):
2018-09-18 14:40:24 +03:00
""" Handling keystrokes on release """
2017-11-09 19:01:09 +03:00
key = event.keyval
2018-11-02 23:13:31 +03:00
map_key = KEY_MAP.get(key, None)
key = map_key if map_key else key
2017-11-09 19:01:09 +03:00
ctrl = event.state & Gdk.ModifierType.CONTROL_MASK
alt = event.state & Gdk.ModifierType.MOD1_MASK
2018-11-02 23:13:31 +03:00
model_name, model = get_model_data(view)
2017-11-09 19:01:09 +03:00
2018-09-18 14:40:24 +03:00
if ctrl and key in MOVE_KEYS:
2017-11-09 19:01:09 +03:00
self.move_items(key)
2018-01-24 13:39:11 +03:00
elif model_name == self._FAV_LIST_NAME and key == Gdk.KEY_Control_L or key == Gdk.KEY_Control_R:
self.update_fav_num_column(model)
2017-11-12 22:23:12 +03:00
self.update_bouquet_list()
elif ctrl and key == Gdk.KEY_Insert:
2017-11-09 19:01:09 +03:00
# Move items from app to fav list
2018-01-24 13:39:11 +03:00
if model_name == self._SERVICE_LIST_NAME:
self.on_to_fav_copy(view)
2018-01-24 13:39:11 +03:00
elif model_name == self._BOUQUETS_LIST_NAME:
2017-11-09 19:01:09 +03:00
self.on_new_bouquet(view)
elif ctrl and key == Gdk.KEY_BackSpace and model_name == self._SERVICE_LIST_NAME:
self.on_to_fav_end_copy(view)
2017-11-09 19:01:09 +03:00
elif ctrl and key == Gdk.KEY_s or key == Gdk.KEY_S:
self.on_data_save()
2017-11-27 13:58:33 +03:00
elif ctrl and key == Gdk.KEY_l or key == Gdk.KEY_L:
self.on_locked(None)
elif ctrl and key == Gdk.KEY_h or key == Gdk.KEY_H:
self.on_hide(None)
elif ctrl and key == Gdk.KEY_R or key == Gdk.KEY_r or key == Gdk.KEY_F2:
self.on_rename(view)
elif ctrl and key == Gdk.KEY_E or key == Gdk.KEY_e:
if model_name == self._BOUQUETS_LIST_NAME:
self.on_rename(view)
return
self.on_service_edit(view)
2018-01-25 21:43:48 +03:00
elif key == Gdk.KEY_Left or key == Gdk.KEY_Right:
view.do_unselect_all(view)
2018-08-01 11:05:29 +03:00
elif ctrl and model_name == self._FAV_LIST_NAME and key in (Gdk.KEY_P, Gdk.KEY_p):
self.on_play_stream()
2017-11-09 19:01:09 +03:00
def on_download(self, item):
DownloadDialog(transient=self._main_window,
properties=self._options.get(self._profile),
open_data_callback=self.open_data,
profile=Profile(self._profile)).show()
2017-11-09 19:01:09 +03:00
def on_view_focus(self, view, focus_event):
profile = Profile(self._profile)
2018-11-02 23:13:31 +03:00
model_name, model = get_model_data(view)
2017-11-16 01:24:16 +03:00
not_empty = len(model) > 0 # if > 0 model has items
2017-11-09 19:01:09 +03:00
2018-01-24 13:39:11 +03:00
if model_name == self._BOUQUETS_LIST_NAME:
for elem in self._tool_elements:
self._tool_elements[elem].set_sensitive(False)
2017-11-09 19:01:09 +03:00
for elem in self._BOUQUET_ELEMENTS:
self._tool_elements[elem].set_sensitive(not_empty)
if elem == "bouquets_paste_popup_item":
self._tool_elements[elem].set_sensitive(not_empty and self._bouquets_buffer)
2018-01-05 14:32:14 +03:00
if profile is Profile.NEUTRINO_MP:
for elem in self._LOCK_HIDE_ELEMENTS:
self._tool_elements[elem].set_sensitive(not_empty)
2017-11-09 19:01:09 +03:00
else:
2018-01-24 13:39:11 +03:00
is_service = model_name == self._SERVICE_LIST_NAME
2018-02-11 23:14:22 +03:00
bq_selected = False
if model_name == self._FAV_LIST_NAME:
bq_selected = self.get_selected_bouquet()
2018-02-11 23:14:22 +03:00
if profile is Profile.NEUTRINO_MP and bq_selected:
name, bq_type = bq_selected.split(":")
bq_selected = BqType(bq_type) is BqType.WEBTV
2017-11-09 19:01:09 +03:00
for elem in self._FAV_ELEMENTS:
2018-07-08 00:09:26 +03:00
if elem in ("paste_tool_button", "fav_paste_popup_item"):
self._tool_elements[elem].set_sensitive(not is_service and self._rows_buffer)
2018-02-11 23:14:22 +03:00
elif elem in self._FAV_ENIGMA_ELEMENTS:
2018-01-08 00:11:07 +03:00
if profile is Profile.ENIGMA_2:
self._tool_elements[elem].set_sensitive(bq_selected and not is_service)
elif elem in self._FAV_IPTV_ELEMENTS:
self._tool_elements[elem].set_sensitive(bq_selected and not is_service)
2017-11-16 01:24:16 +03:00
else:
self._tool_elements[elem].set_sensitive(not_empty and not is_service)
2017-11-09 19:01:09 +03:00
for elem in self._SERVICE_ELEMENTS:
self._tool_elements[elem].set_sensitive(not_empty and is_service)
2017-11-09 19:01:09 +03:00
for elem in self._BOUQUET_ELEMENTS:
self._tool_elements[elem].set_sensitive(False)
2017-11-26 20:40:22 +03:00
for elem in self._LOCK_HIDE_ELEMENTS:
self._tool_elements[elem].set_sensitive(not_empty and profile is Profile.ENIGMA_2)
2017-11-09 19:01:09 +03:00
2017-12-25 19:50:35 +03:00
for elem in self._COMMONS_ELEMENTS:
self._tool_elements[elem].set_sensitive(not_empty)
2017-11-09 19:01:09 +03:00
# self._tool_elements["services_add_new_popup_item"].set_sensitive(len(self._bouquets_model))
2017-11-23 16:59:21 +03:00
def on_hide(self, item):
self.set_service_flags(Flag.HIDE)
2017-11-23 16:59:21 +03:00
def on_locked(self, item):
self.set_service_flags(Flag.LOCK)
2017-11-23 16:59:21 +03:00
def set_service_flags(self, flag):
profile = Profile(self._profile)
bq_selected = self.get_selected_bouquet()
2018-01-05 14:32:14 +03:00
if profile is Profile.ENIGMA_2:
set_flags(flag, self._services_view, self._fav_view, self._services, self._blacklist)
2018-07-08 14:58:41 +03:00
elif profile is Profile.NEUTRINO_MP and bq_selected:
model, paths = self._bouquets_view.get_selection().get_selected_rows()
itr = model.get_iter(paths[0])
value = model.get_value(itr, 1 if flag is Flag.LOCK else 2)
2018-07-08 00:09:26 +03:00
value = None if value else LOCKED_ICON if flag is Flag.LOCK else HIDE_ICON
2018-07-08 14:58:41 +03:00
model.set_value(itr, 1 if flag is Flag.LOCK else 2, value)
2017-11-23 16:59:21 +03:00
2017-11-29 00:26:12 +03:00
@run_idle
def on_model_changed(self, model, path, itr=None):
model_name = model.get_name()
2018-01-24 13:39:11 +03:00
if model_name == self._FAV_LIST_NAME:
self._fav_count_label.set_text(str(len(model)))
2018-01-24 13:39:11 +03:00
elif model_name == self._SERVICE_LIST_NAME:
2017-11-30 00:45:52 +03:00
self.update_services_counts(len(model))
2018-01-24 13:39:11 +03:00
elif model_name == self._BOUQUETS_LIST_NAME:
self._bouquets_count_label.set_text(str(len(self._bouquets.keys())))
2017-11-29 00:26:12 +03:00
2017-11-30 00:45:52 +03:00
@lru_cache(maxsize=1)
def update_services_counts(self, size=0):
""" Updates counters for services. May be temporary! """
2017-11-29 00:26:12 +03:00
tv_count = 0
radio_count = 0
data_count = 0
for ch in self._services.values():
2017-11-29 00:26:12 +03:00
ch_type = ch.service_type
2018-03-03 20:55:08 +03:00
if ch_type in self._TV_TYPES:
2017-11-29 00:26:12 +03:00
tv_count += 1
elif ch_type == "Radio":
radio_count += 1
elif ch_type == "Data":
data_count += 1
self._tv_count_label.set_text(str(tv_count))
self._radio_count_label.set_text(str(radio_count))
self._data_count_label.set_text(str(data_count))
2017-11-29 00:26:12 +03:00
2017-12-19 22:57:04 +03:00
def on_insert_marker(self, view):
2017-12-20 16:46:15 +03:00
""" Inserts marker into bouquet services list. """
insert_marker(view, self._bouquets, self.get_selected_bouquet(), self._services, self._main_window)
self.update_fav_num_column(self._fav_model)
2017-12-19 22:57:04 +03:00
2018-04-29 01:44:28 +03:00
def on_fav_press(self, menu, event):
2018-08-01 11:05:29 +03:00
if event.get_event_type() == Gdk.EventType.DOUBLE_BUTTON_PRESS:
self.on_play_stream()
else:
return self.on_view_popup_menu(menu, event)
2018-04-29 15:36:35 +03:00
2018-09-01 00:49:11 +03:00
# ***************** IPTV *********************#
def on_iptv(self, item):
response = IptvDialog(self._main_window,
self._fav_view,
self._services,
self._bouquets.get(self.get_selected_bouquet(), None),
Profile(self._profile),
Action.ADD).show()
if response != Gtk.ResponseType.CANCEL:
self.update_fav_num_column(self._fav_model)
@run_idle
def on_iptv_list_configuration(self, item):
profile = Profile(self._profile)
if profile is Profile.NEUTRINO_MP:
show_dialog(DialogType.ERROR, transient=self._main_window, text="Neutrino at the moment not supported!")
return
iptv_rows = list(filter(lambda r: r[5] == BqServiceType.IPTV.value, self._fav_model))
if not iptv_rows:
2018-09-15 17:26:01 +03:00
show_dialog(DialogType.ERROR, self._main_window, "This list does not contains IPTV streams!")
return
bq_selected = self.get_selected_bouquet()
if not bq_selected:
return
bouquet = self._bouquets.get(bq_selected, [])
IptvListConfigurationDialog(self._main_window, self._services, iptv_rows, bouquet, profile).show()
@run_idle
def on_remove_all_unavailable(self, item):
iptv_rows = list(filter(lambda r: r[5] == BqServiceType.IPTV.value, self._fav_model))
if not iptv_rows:
2018-09-15 17:26:01 +03:00
show_dialog(DialogType.ERROR, self._main_window, "This list does not contains IPTV streams!")
return
bq_selected = self.get_selected_bouquet()
if not bq_selected:
return
if show_dialog(DialogType.QUESTION, self._main_window) == Gtk.ResponseType.CANCEL:
return
fav_bqt = self._bouquets.get(bq_selected, None)
prf = Profile(self._profile)
response = SearchUnavailableDialog(self._main_window, self._fav_model, fav_bqt, iptv_rows, prf).show()
if response:
next(self.remove_favs(response, self._fav_model), False)
2018-09-01 00:49:11 +03:00
def on_import_m3u(self, item):
""" Imports iptv from m3u files. """
response = get_chooser_dialog(self._main_window, self._options.get(self._profile), "*.m3u", "m3u files")
if response == Gtk.ResponseType.CANCEL:
return
if not str(response).endswith("m3u"):
show_dialog(DialogType.ERROR, self._main_window, text="No m3u file is selected!")
return
channels = parse_m3u(response, Profile(self._profile))
bq_selected = self.get_selected_bouquet()
if channels and bq_selected:
bq_services = self._bouquets.get(bq_selected)
self._fav_model.clear()
for ch in channels:
self._services[ch.fav_id] = ch
bq_services.append(ch.fav_id)
next(self.update_bouquet_services(self._fav_model, None, bq_selected), False)
2018-09-01 00:49:11 +03:00
2018-08-25 15:30:12 +03:00
# ***************** Player *********************#
2018-08-25 15:30:12 +03:00
def on_play_stream(self, item=None):
self.on_player_play()
@run_idle
2018-08-25 15:30:12 +03:00
def on_player_play(self, item=None):
url = self.get_stream_url()
self.update_player_buttons()
if not url:
return
self.play(url)
def play(self, url):
if not self._player:
try:
self._player = Player()
except (NameError, AttributeError):
show_dialog(DialogType.ERROR, self._main_window, "No VLC is found. Check that it is installed!")
return
else:
if self._drawing_area_xid:
self._player.set_xwindow(self._drawing_area_xid)
self._services_main_box.set_visible(False)
self._bouquets_main_box.set_visible(False)
w, h = self._main_window.get_size()
self._player_box.set_size_request(w * 0.6, -1)
self._player_box.set_visible(True)
if self._player.is_playing():
self.on_player_stop()
GLib.idle_add(self._player.play, url, priority=GLib.PRIORITY_LOW)
def get_stream_url(self):
2018-08-25 15:30:12 +03:00
path, column = self._fav_view.get_cursor()
if path:
row = self._fav_model[path][:]
2018-09-18 07:13:32 +03:00
if row[5] == BqServiceType.IPTV.name:
return get_iptv_url(row, Profile(self._profile))
2018-08-25 15:30:12 +03:00
def on_player_stop(self, item=None):
if self._player:
2018-09-18 07:13:32 +03:00
self._player.stop()
2018-08-25 15:30:12 +03:00
def on_player_previous(self, item):
if self._fav_view.do_move_cursor(self._fav_view, Gtk.MovementStep.DISPLAY_LINES, -1):
self.on_play_stream()
def on_player_next(self, item):
if self._fav_view.do_move_cursor(self._fav_view, Gtk.MovementStep.DISPLAY_LINES, 1):
self.on_play_stream()
def update_player_buttons(self):
if self._player:
path, column = self._fav_view.get_cursor()
current_index = path[0]
self._player_prev_button.set_sensitive(current_index != 0)
self._player_next_button.set_sensitive(len(self._fav_model) != current_index + 1)
2018-08-25 15:30:12 +03:00
def on_player_close(self, item=None):
2018-04-29 15:36:35 +03:00
if self._player:
2018-08-25 15:30:12 +03:00
self._player.release()
2018-08-25 23:53:53 +03:00
self._player = None
2018-09-30 23:16:30 +03:00
GLib.idle_add(self._player_box.set_visible, False, priority=GLib.PRIORITY_LOW)
2018-09-30 00:12:15 +03:00
GLib.idle_add(self._services_main_box.set_visible, True, priority=GLib.PRIORITY_LOW)
GLib.idle_add(self._bouquets_main_box.set_visible, True, priority=GLib.PRIORITY_LOW)
2018-08-25 15:30:12 +03:00
def on_drawing_area_realize(self, widget):
self._drawing_area_xid = widget.get_window().get_xid()
2018-09-18 07:13:32 +03:00
self._player.set_xwindow(self._drawing_area_xid)
2018-09-30 23:16:30 +03:00
def on_player_drawing_area_draw(self, widget, cr):
""" Used for black background drawing in the player drawing area.
Required for Gtk >= 3.20.
More info: https://developer.gnome.org/gtk3/stable/ch32s10.html,
https://developer.gnome.org/gtk3/stable/GtkStyleContext.html#gtk-render-background
"""
context = widget.get_style_context()
width = widget.get_allocated_width()
height = widget.get_allocated_height()
Gtk.render_background(context, cr, 0, 0, width, height)
r, g, b, a = 0, 0, 0, 1 # black color
cr.set_source_rgba(r, g, b, a)
cr.rectangle(0, 0, width, height)
cr.fill()
def on_player_press(self, area, event):
if event.button == Gdk.BUTTON_PRIMARY:
if event.type == Gdk.EventType.DOUBLE_BUTTON_PRESS:
self.on_full_screen()
def on_full_screen(self, item=None):
self._full_screen = not self._full_screen
self._main_window.fullscreen() if self._full_screen else self._main_window.unfullscreen()
def on_main_window_state(self, window, event):
2018-09-30 00:12:15 +03:00
full = not event.new_window_state & Gdk.WindowState.FULLSCREEN
self._main_data_box.set_visible(full)
self._status_bar_box.set_visible(full)
2018-09-30 23:16:30 +03:00
self._player_tool_bar.set_visible(full)
2018-09-01 00:49:11 +03:00
# ***************** Filter and search *********************#
2018-01-08 22:00:48 +03:00
2018-01-31 00:13:42 +03:00
def on_filter_toggled(self, toggle_button: Gtk.ToggleToolButton):
2018-07-08 00:09:26 +03:00
active = toggle_button.get_active()
if active:
self.update_filter_sat_positions()
2018-07-08 00:09:26 +03:00
self._filter_bar.set_search_mode(active)
2018-09-12 17:49:28 +03:00
self._filter_bar.set_visible(active)
2018-01-31 00:13:42 +03:00
def init_sat_positions(self):
self._sat_positions.clear()
first = (self._filter_sat_positions_model[0][0],)
self._filter_sat_positions_model.clear()
self._filter_sat_positions_model.append(first)
self._filter_sat_positions_box.set_active(0)
def update_sat_positions(self):
""" Updates positions values for the filtering function """
self._sat_positions.clear()
sat_positions = set()
terrestrial = False
for srv in self._services.values():
if srv.transponder_type == "s" and srv.pos:
sat_positions.add(float(srv.pos))
elif srv.transponder_type == "t":
terrestrial = True
if terrestrial:
self._sat_positions.append("T")
self._sat_positions.extend(map(str, sorted(sat_positions)))
if self._filter_bar.is_visible():
self.update_filter_sat_positions()
@run_idle
def update_filter_sat_positions(self):
model = self._filter_sat_positions_model
if len(model) < 2:
list(map(self._filter_sat_positions_model.append, map(lambda x: (str(x),), self._sat_positions)))
else:
selected = self._filter_sat_positions_box.get_active_id()
active = self._filter_sat_positions_box.get_active()
itrs = list(filter(lambda it: model[it][0] not in self._sat_positions, [row.iter for row in model][1:]))
list(map(model.remove, itrs))
if active != 0 and selected not in self._sat_positions:
self._filter_sat_positions_box.set_active(0)
@run_with_delay(1)
def on_filter_changed(self, item):
GLib.idle_add(self._services_model_filter.refilter, priority=GLib.PRIORITY_LOW)
def services_filter_function(self, model, iter, data):
if self._services_model_filter is None or self._services_model_filter == "None":
return True
else:
txt = self._filter_entry.get_text() in str(model.get(iter, 3, 6, 7, 10, 11, 12, 13, 14, 15, 16))
type_active = self._filter_types_box.get_active() > 0
pos_active = self._filter_sat_positions_box.get_active() > 0
free = not model.get(iter, 2)[0] if self._filter_only_free_button.get_active() else True
if type_active and pos_active:
return self._filter_types_box.get_active_id() == model.get(iter, 7)[
0] and self._filter_sat_positions_box.get_active_id() == model.get(iter, 16)[0] and txt and free
elif type_active:
return self._filter_types_box.get_active_id() == model.get(iter, 7)[0] and txt and free
elif pos_active:
return self._filter_sat_positions_box.get_active_id() == model.get(iter, 16)[0] and txt and free
return txt and free
2018-01-28 23:10:54 +03:00
2018-01-31 16:02:26 +03:00
def on_search_toggled(self, toggle_button: Gtk.ToggleToolButton):
2018-07-08 00:09:26 +03:00
self._search_bar.set_search_mode(toggle_button.get_active())
2018-01-31 16:02:26 +03:00
def on_search_down(self, item):
self._search_provider.on_search_down()
def on_search_up(self, item):
self._search_provider.on_search_up()
@run_with_delay(1)
2018-03-06 19:06:16 +03:00
def on_search(self, entry):
self._search_provider.search(entry.get_text())
2018-02-02 12:45:58 +03:00
2018-09-01 00:49:11 +03:00
# ***************** Editing *********************#
2018-02-15 15:16:34 +03:00
@run_idle
def on_service_edit(self, view):
model, paths = view.get_selection().get_selected_rows()
if is_only_one_item_selected(paths, self._main_window):
model_name = get_base_model(model).get_name()
if model_name == self._FAV_LIST_NAME:
srv_type = model.get_value(model.get_iter(paths), 5)
2018-03-11 21:52:10 +03:00
if srv_type == BqServiceType.MARKER.name:
return self.on_rename(view)
elif srv_type == BqServiceType.IPTV.name:
return IptvDialog(self._main_window,
self._fav_view,
self._services,
self._bouquets.get(self.get_selected_bouquet(), None),
Profile(self._profile),
2018-03-11 21:52:10 +03:00
Action.EDIT).show()
self.on_locate_in_services(view)
dialog = ServiceDetailsDialog(self._main_window,
self._options,
self._services_view,
2018-04-10 13:04:21 +03:00
self._fav_view,
self._services,
self._bouquets)
dialog.show()
2018-02-14 00:00:52 +03:00
def on_services_add_new(self, item):
dialog = ServiceDetailsDialog(self._main_window,
self._options,
self._services_view,
2018-04-10 13:04:21 +03:00
self._fav_view,
self._services,
self._bouquets,
action=Action.ADD)
dialog.show()
def on_bouquets_edit(self, view):
""" Rename bouquets """
bq_selected = self.get_selected_bouquet()
if not bq_selected:
show_dialog(DialogType.ERROR, self._main_window, "This item is not allowed to edit!")
return
model, paths = view.get_selection().get_selected_rows()
if paths:
itr = model.get_iter(paths[0])
bq_name, bq_type = model.get(itr, 0, 3)
response = show_dialog(DialogType.INPUT, self._main_window, bq_name)
if response == Gtk.ResponseType.CANCEL:
return
model.set_value(itr, 0, response)
self._bouquets["{}:{}".format(response, bq_type)] = self._bouquets.pop("{}:{}".format(bq_name, bq_type))
def on_rename(self, view):
2018-11-02 23:13:31 +03:00
name, model = get_model_data(view)
if name == self._BOUQUETS_LIST_NAME:
self.on_bouquets_edit(view)
elif name == self._FAV_LIST_NAME:
rename(view, self._main_window, ViewTarget.FAV, service_view=self._services_view,
2018-10-25 16:00:25 +03:00
services=self._services)
elif name == self._SERVICE_LIST_NAME:
2018-10-25 16:00:25 +03:00
rename(view, self._main_window, ViewTarget.SERVICES, fav_view=self._fav_view, services=self._services)
def on_rename_for_bouquet(self, item):
selection = get_selection(self._fav_view, self._main_window)
if not selection:
return
model, paths = selection
data = model[paths][:]
cur_name, srv_type, fav_id = data[2], data[5], data[7]
2018-10-26 19:28:40 +03:00
if srv_type == BqServiceType.IPTV.name or srv_type == BqServiceType.MARKER.name:
2018-10-26 20:00:11 +03:00
show_dialog(DialogType.ERROR, self._main_window, "Not allowed in this context!")
return
response = show_dialog(DialogType.INPUT, self._main_window, cur_name)
if response == Gtk.ResponseType.CANCEL:
return
srv = self._services.get(fav_id, None)
selected_bq = self.get_selected_bouquet()
ex_bq = self._extra_bouquets.get(selected_bq, None)
if srv.service == response and ex_bq:
ex_bq.pop(fav_id, None)
if not ex_bq:
self._extra_bouquets.pop(selected_bq, None)
else:
if ex_bq:
ex_bq[fav_id] = response
else:
self._extra_bouquets[selected_bq] = {fav_id: response}
model.set_value(model.get_iter(paths), 2, response)
def on_set_default_name_for_bouquet(self, item):
selection = get_selection(self._fav_view, self._main_window)
if not selection:
return
model, paths = selection
fav_id = model[paths][7]
srv = self._services.get(fav_id, None)
selected_bq = self.get_selected_bouquet()
ex_bq = self._extra_bouquets.get(selected_bq, None)
if not ex_bq:
show_dialog(DialogType.ERROR, self._main_window, "No changes required!")
return
else:
if not ex_bq.pop(fav_id, None):
show_dialog(DialogType.ERROR, self._main_window, "No changes required!")
return
if not ex_bq:
self._extra_bouquets.pop(selected_bq, None)
model.set_value(model.get_iter(paths), 2, srv.service)
2018-09-01 00:49:11 +03:00
def on_locate_in_services(self, view):
locate_in_services(view, self._services_view, self._main_window)
# ***************** Picons *********************#
@run_idle
def on_picons_loader_show(self, item):
ids = {}
if Profile(self._profile) is Profile.ENIGMA_2:
for r in self._services_model:
data = r[9].split("_")
ids["{}:{}:{}".format(data[3], data[5], data[6])] = r[9]
dialog = PiconsDialog(self._main_window, self._options, ids, self._sat_positions, Profile(self._profile))
2018-09-01 00:49:11 +03:00
dialog.show()
self.update_picons()
2018-08-18 10:21:40 +03:00
@run_task
2018-01-28 23:10:54 +03:00
def update_picons(self):
2018-08-18 10:21:40 +03:00
update_picons_data(self._options.get(self._profile).get("picons_dir_path"), self._picons)
append_picons(self._picons, self._services_model)
2018-01-23 16:18:28 +03:00
2018-01-30 12:37:04 +03:00
def on_assign_picon(self, view):
2018-02-01 21:43:44 +03:00
assign_picon(self.get_target_view(view),
self._services_view,
self._fav_view,
self._main_window,
self._picons,
self._options.get(self._profile),
self._services)
2018-01-29 18:07:47 +03:00
2018-01-30 12:37:04 +03:00
def on_remove_picon(self, view):
2018-02-01 13:10:06 +03:00
remove_picon(self.get_target_view(view),
self._services_view,
self._fav_view, self._picons,
self._options.get(self._profile))
2018-01-29 18:07:47 +03:00
2018-01-30 12:37:04 +03:00
def on_reference_picon(self, view):
""" Copying picon id to clipboard """
copy_picon_reference(self.get_target_view(view), view, self._services, self._clipboard, self._main_window)
2018-02-01 13:10:06 +03:00
def get_target_view(self, view):
return ViewTarget.SERVICES if Gtk.Buildable.get_name(view) == "services_tree_view" else ViewTarget.FAV
2018-01-29 18:07:47 +03:00
2018-09-01 00:49:11 +03:00
# ***************** Bouquets *********************#
def on_create_bouquet_for_current_satellite(self, item):
2018-04-03 22:47:29 +03:00
self.create_bouquets(BqGenType.SAT)
def on_create_bouquet_for_each_satellite(self, item):
2018-04-03 22:47:29 +03:00
self.create_bouquets(BqGenType.EACH_SAT)
def on_create_bouquet_for_current_package(self, item):
2018-04-03 22:47:29 +03:00
self.create_bouquets(BqGenType.PACKAGE)
def on_create_bouquet_for_each_package(self, item):
2018-04-03 22:47:29 +03:00
self.create_bouquets(BqGenType.EACH_PACKAGE)
2018-04-07 23:49:36 +03:00
def on_create_bouquet_for_current_type(self, item):
self.create_bouquets(BqGenType.TYPE)
def on_create_bouquet_for_each_type(self, item):
self.create_bouquets(BqGenType.EACH_TYPE)
2018-04-03 22:47:29 +03:00
def create_bouquets(self, g_type):
gen_bouquets(self._services_view, self._bouquets_view, self._main_window, g_type, self._TV_TYPES,
Profile(self._profile), self.append_bouquet)
2018-09-01 00:49:11 +03:00
# ***************** Profile label *********************#
2018-06-01 11:16:30 +03:00
def update_profile_label(self):
profile = Profile(self._profile)
if profile is Profile.ENIGMA_2:
2018-08-15 10:51:19 +03:00
self._header_bar.set_subtitle("{} Enigma2 v.{}".format(get_message("Profile:"), self.get_format_version()))
2018-06-01 11:16:30 +03:00
elif profile is Profile.NEUTRINO_MP:
2018-08-15 10:51:19 +03:00
self._header_bar.set_subtitle("{} Neutrino-MP".format(get_message("Profile:")))
2018-06-01 11:16:30 +03:00
def get_format_version(self):
return 5 if self._options.get(self._profile).get("v5_support", False) else 4
2017-11-09 19:01:09 +03:00
def start_app():
MainAppWindow()
Gtk.main()
def close_app():
Gtk.main_quit()
if __name__ == "__main__":
2017-11-10 13:38:03 +03:00
pass