itkdb-gtk 0.9.0__py3-none-any.whl → 0.10.0__py3-none-any.whl

itkdb_gtk/dashBoard.py

@@ -13,7 +13,7 @@ except ImportError:
     
 from itkdb_gtk import dbGtkUtils
 from itkdb_gtk import GetShipments
-from itkdb_gtk import GroundVITests
+from itkdb_gtk import PetalReceptionTests
 from itkdb_gtk import ITkDBlogin
 from itkdb_gtk import CreateShipments
 from itkdb_gtk import UploadTest
@@ -27,6 +27,7 @@ import gi
 gi.require_version("Gtk", "3.0")
 from gi.repository import Gtk
 
+HELP_LINK="https://itkdb-gtk.docs.cern.ch"
 
 class DashWindow(dbGtkUtils.ITkDBWindow):
     """Dashboard class."""
@@ -41,7 +42,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
     
     def __init__(self, session):
         """Initialization."""
-        super().__init__(title="ITkDB Dashboard", session=session)
+        super().__init__(title="ITkDB Dashboard", session=session, help=HELP_LINK)
         self.mask = 0
 
         # set border width
@@ -71,7 +72,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
         grid.attach(btnTest, 1, irow, 1, 1)
 
         irow += 1
-        btnGnd = Gtk.Button(label="Petal VI/GND")
+        btnGnd = Gtk.Button(label="Petal Reception")
         btnGnd.connect("clicked", self.petal_gnd)
         grid.attach(btnGnd, 0, irow, 1, 1)
 
@@ -80,7 +81,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
         grid.attach(btnWeight, 1, irow, 1, 1)
 
         irow += 1
-        btnModIV = Gtk.Button(label="Module IV")
+        btnModIV = Gtk.Button(label="Sensor/Module IV")
         btnModIV.connect("clicked", self.module_IV)
         grid.attach(btnModIV, 0, irow, 1, 1)
 
@@ -120,7 +121,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = UploadTest.UploadTest(self.session)
+        W = UploadTest.UploadTest(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def upload_multiple_tests(self, *args):
@@ -131,7 +132,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = UploadMultipleTests.UploadMultipleTests(self.session)
+        W = UploadMultipleTests.UploadMultipleTests(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def create_shipment(self, *args):
@@ -142,7 +143,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = CreateShipments.CreateShipments(self.session)
+        W = CreateShipments.CreateShipments(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def receive_shipment(self, *args):
@@ -153,7 +154,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = GetShipments.ReceiveShipments(self.session)
+        W = GetShipments.ReceiveShipments(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def petal_gnd(self, *args):
@@ -164,7 +165,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = GroundVITests.GroundingTest(self.session)
+        W = PetalReceptionTests.PetalReceptionTests(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def glue_weight(self, *args):
@@ -175,7 +176,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = GlueWeight.GlueWeight(self.session)
+        W = GlueWeight.GlueWeight(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def module_IV(self, *args):
@@ -186,7 +187,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = UploadModuleIV.IVwindow(self.session)
+        W = UploadModuleIV.IVwindow(self.session, help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def wire_bond(self, *args):
@@ -197,7 +198,7 @@ class DashWindow(dbGtkUtils.ITkDBWindow):
             return
 
         self.mask |= bt
-        W = WireBondGui.WireBond(session=self.session, title="Wirebond")
+        W = WireBondGui.WireBond(session=self.session, title="Wirebond", help=HELP_LINK)
         W.connect("destroy", self.app_closed, bitn)
 
     def app_closed(self, *args):

itkdb_gtk/dbGtkUtils.py

@@ -1,18 +1,51 @@
 """A set of GTK utilities for DB scripts."""
 import json
 import time
+import pathlib
+
 from collections.abc import Iterable
 from copy import deepcopy
 from datetime import datetime
+import webbrowser
+import serial
 
 import dateutil.parser
-import gi
 import numpy as np
 
 
+import gi
 gi.require_version("Gtk", "3.0")
 from gi.repository import Gtk, GObject, Gio, GLib
 
+
+def setup_scanner(callback):
+    """Setup scanner and callback function."""
+    for fnam in ["/dev/ttyACM0", "/dev/cu.usbmodemS_N_G19F204881"]:
+        P = pathlib.Path(fnam)
+        if P.exists():
+            rdr = serial.Serial(fnam, 9600)
+            GLib.unix_fd_add_full(
+                GLib.PRIORITY_DEFAULT,
+                rdr.fileno(),
+                GLib.IOCondition.IN,
+                callback,
+                rdr,
+            )
+            break
+
+def scanner_get_line(reader):
+    """Reads a line from serial."""
+    available = reader.in_waiting
+    while True:
+        delta = reader.in_waiting - available
+        if not delta:
+            break
+
+    # Get data from serial device passed in via
+    data = reader.read_until(expected='\r', size=reader.in_waiting).strip()
+    txt = data.decode('utf-8')
+    return txt
+
 def parse_date(txt):
     """Parse a date."""
     try:
@@ -20,6 +53,7 @@ def parse_date(txt):
 
     except Exception:
         return None
+
 def parse_date_as_string(txt):
     """Parse data and return DB compatible string."""
     D = parse_date(txt)
@@ -62,6 +96,14 @@ def set_entry_style(container):
     style_context.add_provider(provider, Gtk.STYLE_PROVIDER_PRIORITY_SETTINGS)
     return container    
 
+def set_button_color(btn, bg_color, fg_color="white"):
+    """Set button color"""
+    css = "#{} {{background-image: none; background-color: {}; color: {}}}".format(btn.get_name(), bg_color, fg_color)
+    provider = Gtk.CssProvider()
+    provider.load_from_data(css.encode())
+    style_context = btn.get_style_context()
+    style_context.add_provider(provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)
+    
 def set_combo_iter(combo, txt, col=0):
     """Set scombo active iter to that containing txt in column col."""
     model = combo.get_model()
@@ -422,7 +464,7 @@ class MessagePanel(object):
         end = self.textbuffer.get_end_iter()
         self.text_view.scroll_to_iter(end, 0, False, 0, 0)
 
-    def write_message(self, text):
+    def write_message(self, text, write_date=True):
         """Writes text to Text Viewer."""
         nlines = self.textbuffer.get_line_count()
         if nlines > 100:
@@ -431,15 +473,23 @@ class MessagePanel(object):
             self.textbuffer.delete(start, end)
 
         end = self.textbuffer.get_end_iter()
-        msg = "[{}]  {}".format(time.strftime("%d/%m/%y %T"), text)
+        if write_date:
+            msg = "[{}]  {}".format(time.strftime("%d/%m/%y %T"), text)
+        else:
+            msg = text
+
         self.textbuffer.insert(end, msg)
         GLib.idle_add(self.scroll_to_end)
+        
+    def write(self, txt):
+        """A write method."""
+        self.write_message(txt, write_date=False)
 
 
 class ITkDBWindow(Gtk.Window):
     """Base class for GUI main windows."""
 
-    def __init__(self, title="", session=None, show_search=None, gtk_runs=True, panel_size=100):
+    def __init__(self, title="", session=None, show_search=None, help=None, gtk_runs=True, panel_size=100):
         """Initialization.
 
         Args:
@@ -455,6 +505,7 @@ class ITkDBWindow(Gtk.Window):
         self.inst2code = {}
         self.code2inst = {}
         self.message_panel = None
+        self.help = help
 
         if gtk_runs:
             super().__init__(title=title)
@@ -482,6 +533,15 @@ class ITkDBWindow(Gtk.Window):
             button.set_tooltip_text(show_search)
             button.connect("clicked", self.query_db)
             self.hb.pack_end(button)
+            
+        if self.help:
+            button = Gtk.Button()
+            icon = Gio.ThemedIcon(name="help-browser-symbolic")
+            image = Gtk.Image.new_from_gicon(icon, Gtk.IconSize.BUTTON)
+            button.add(image)
+            button.connect("clicked", self.show_help)
+            self.hb.pack_end(button)
+            
 
         # Create main content box
         self.mainBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
@@ -507,6 +567,10 @@ class ITkDBWindow(Gtk.Window):
         self.hide()
         self.destroy()
 
+    def show_help(self, *args):
+        """Show help"""
+        webbrowser.open(self.help)
+    
     def query_db(self, *args):
         """Search button clicked."""
         pass
@@ -516,7 +580,9 @@ class ITkDBWindow(Gtk.Window):
         if msg == "<OK>":
             if hasattr(self.session, "user_gui"):
                 self.session = self.session.user_gui.get_client()
-            self.userLabel.get_child().set_text(self.session.user.name)
+                
+            if self.userLabel.get_child():
+                self.userLabel.get_child().set_text(self.session.user.name)
 
         else:
             self.write_message("Could not login.\n{}".format(msg))
@@ -746,6 +812,28 @@ class DictDialog(Gtk.Grid):
         self.show_values()
         self.show_all()
         self.queue_draw()
+        
+    @staticmethod
+    def create_json_data_editor(data):
+        """Create a dialog to show the JSon file."""
+        dlg = Gtk.Dialog(title="Test Data")
+        dlg.add_buttons(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
+                        Gtk.STOCK_OK, Gtk.ResponseType.OK)
+
+        dlg.set_property("height-request", 500)
+        box = dlg.get_content_area()
+        value = DictDialog(data)
+        scrolled = Gtk.ScrolledWindow()
+        scrolled.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
+        scrolled.add(value)
+        box.pack_start(scrolled, True, True, 10)
+
+        dlg.show_all()
+        rc = dlg.run()
+        dlg.hide()
+        dlg.destroy()
+        
+        return value.values, rc
 
 
 def create_scrolled_dictdialog(the_dict, hidden=("component", "testType")):