From dca94271b0e48174c406d942596163f5a7e06c5e Mon Sep 17 00:00:00 2001
From: DYefremov <dmitry.v.yefremov@gmail.com>
Date: Sun, 1 Apr 2018 09:39:14 +0300
Subject: [PATCH] checking data in IPTV dialog

---
 app/ui/dialogs.glade |  5 +++--
 app/ui/iptv.py       | 22 ++++++++++++++++++++--
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/app/ui/dialogs.glade b/app/ui/dialogs.glade
index 604b09a5..622295b8 100644
--- a/app/ui/dialogs.glade
+++ b/app/ui/dialogs.glade
@@ -864,6 +864,7 @@ dmitry.v.yefremov@gmail.com
                     <property name="visible">True</property>
                     <property name="can_focus">True</property>
                     <property name="primary_icon_stock">gtk-edit</property>
+                    <signal name="changed" handler="on_url_changed" swapped="no"/>
                   </object>
                   <packing>
                     <property name="left_attach">0</property>
@@ -944,7 +945,7 @@ dmitry.v.yefremov@gmail.com
                       <object class="GtkLabel" id="label30">
                         <property name="visible">True</property>
                         <property name="can_focus">False</property>
-                        <property name="label" translatable="yes">Tr. ID</property>
+                        <property name="label" translatable="yes">TID</property>
                       </object>
                       <packing>
                         <property name="left_attach">2</property>
@@ -955,7 +956,7 @@ dmitry.v.yefremov@gmail.com
                       <object class="GtkLabel" id="label31">
                         <property name="visible">True</property>
                         <property name="can_focus">False</property>
-                        <property name="label" translatable="yes">Net. ID</property>
+                        <property name="label" translatable="yes">NID</property>
                       </object>
                       <packing>
                         <property name="left_attach">3</property>
diff --git a/app/ui/iptv.py b/app/ui/iptv.py
index 3e9fafa1..f2fd94d7 100644
--- a/app/ui/iptv.py
+++ b/app/ui/iptv.py
@@ -1,5 +1,6 @@
 import re
 
+from urllib.parse import urlparse
 from app.eparser.ecommons import BqServiceType, Service
 from app.eparser.iptv import NEUTRINO_FAV_ID_FORMAT, StreamType, ENIGMA2_FAV_ID_FORMAT
 from app.properties import Profile
@@ -9,8 +10,11 @@ from .main_helper import get_base_model
 
 
 class IptvDialog:
+    _DIGIT_ENTRY_NAME = "digit-entry"
+
     def __init__(self, transient, view, services, bouquet, profile=Profile.ENIGMA_2, action=Action.ADD):
         handlers = {"on_entry_changed": self.on_entry_changed,
+                    "on_url_changed": self.on_url_changed,
                     "on_save": self.on_save,
                     "on_stream_type_changed": self.on_stream_type_changed}
 
@@ -39,7 +43,6 @@ class IptvDialog:
         self._bouquet = bouquet
         self._services = services
         self._model, self._paths = view.get_selection().get_selected_rows()
-
         self._PATTERN = re.compile("(?:^[\s]*$|\D)")
         # style
         self._style_provider = Gtk.CssProvider()
@@ -71,6 +74,10 @@ class IptvDialog:
         self._dialog.destroy()
 
     def on_save(self, item):
+        if not self.is_data_correct():
+            show_dialog(DialogType.ERROR, self._dialog, "Error. Verify the data!")
+            return
+
         if show_dialog(DialogType.QUESTION, self._dialog) == Gtk.ResponseType.CANCEL:
             return
 
@@ -116,11 +123,15 @@ class IptvDialog:
 
     def on_entry_changed(self, entry):
         if self._PATTERN.search(entry.get_text()):
-            entry.set_name("digit-entry")
+            entry.set_name(self._DIGIT_ENTRY_NAME)
         else:
             entry.set_name("GtkEntry")
             self._update_reference_entry()
 
+    def on_url_changed(self, entry):
+        url = urlparse(entry.get_text())
+        entry.set_name("GtkEntry" if all([url.scheme, url.netloc, url.path]) else self._DIGIT_ENTRY_NAME)
+
     def on_stream_type_changed(self, item):
         self._update_reference_entry()
 
@@ -162,6 +173,13 @@ class IptvDialog:
             self._bouquet.insert(self._model.get_path(itr)[0], fav_id)
             self._services[fav_id] = Service(None, None, IPTV_ICON, name, *aggr[0:3], s_type, *aggr, fav_id, None)
 
+    def is_data_correct(self):
+        for elem in (self._srv_type_entry, self._sid_entry, self._tr_id_entry, self._net_id_entry,
+                     self._namespace_entry, self._url_entry):
+            if elem.get_name() == self._DIGIT_ENTRY_NAME:
+                return False
+        return True
+
 
 if __name__ == "__main__":
     pass