itkdb-gtk 0.0.3__py3-none-any.whl → 0.20.1__py3-none-any.whl

itkdb_gtk/findComponent.py

@@ -0,0 +1,242 @@
+#!/usr/bin/env python3
+"""GUI to upload tests."""
+import argparse
+import json
+import sys
+from pathlib import Path
+
+try:
+    import itkdb_gtk
+
+except ImportError:
+    cwd = Path(__file__).parent.parent
+    sys.path.append(cwd.as_posix())
+
+from itkdb_gtk import dbGtkUtils, ITkDBlogin, ITkDButils, QRScanner, findVTRx
+
+import gi
+gi.require_version("Gtk", "3.0")
+from gi.repository import Gtk, GLib
+
+class FindComponent(dbGtkUtils.ITkDBWindow):
+    """Read QR of bar code and retrieve information about component."""
+
+    def __init__(self, session, help_link=None):
+        """Initialization.
+
+        Args:
+            session: ITkDB session
+            help_link: link to help page.
+
+        """
+        super().__init__(session=session, title="Find Component", help_link=help_link)
+        self.scanner = QRScanner.QRScanner(self.get_qrcode)
+        self.ofd = None  # Output file descriptor
+        self.init_window()
+
+    def __del__(self):
+        """Destructor."""
+        if self.ofd is not None:
+            self.ofd.close()
+
+    def init_window(self):
+        """Create the Gtk window."""
+        # Initial tweaks
+        self.set_border_width(10)
+
+        # Prepare HeaderBar
+        self.hb.props.title = "Find Component"
+
+
+        # File name
+        box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=6)
+        self.mainBox.pack_start(box, False, False, 0)
+        self.open_btn = Gtk.Button(label="Output File")
+        self.open_btn.set_tooltip_text("Select output file to save scanned component information in CSV format.")
+        self.open_btn.connect("clicked", self.on_file_set)
+        box.pack_start(self.open_btn, False, False, 0)
+
+        self.close_btn = Gtk.Button(label="Close File")
+        self.close_btn.set_sensitive(False)
+        self.close_btn.set_tooltip_text("Close the output file.")
+        self.close_btn.connect("clicked", self.on_file_close)
+        box.pack_start(self.close_btn, False, False, 0)
+
+
+        self.file_name = Gtk.Label(label="No output file selected")
+        self.mainBox.pack_start(self.file_name, False, False, 5)
+
+        self.scanner_dev = Gtk.Label(label="No scanner found")
+        self.mainBox.pack_start(self.scanner_dev, False, False, 5)
+
+        # Object Data
+        lbl = Gtk.Label(label="Scan your QR or bar code. Information will appear below.")
+        self.mainBox.pack_start(lbl, False, False, 10)
+
+        
+        #btn = Gtk.Button(label="Test Button")
+        #btn.connect("clicked", self.test_qrcode)
+        #self.mainBox.pack_start(btn, True, True, 0)
+
+        # The text view
+        self.mainBox.pack_start(self.message_panel.frame, True, True, 0)
+
+        self.timer_id = GLib.timeout_add(500, self.find_scanner)
+
+        self.show_all()
+
+    def find_scanner(self, *args):
+        """Figure out if there is a scanner connected."""
+        if self.scanner.reader is None:
+            self.scanner_dev.set_text("No scanner found")
+        else:
+            self.scanner_dev.set_text("Found scanner in {}".format(self.scanner.reader.name))
+
+        return True
+    
+    def on_file_set(self, *args):
+        """File selected.
+
+            Opens the file for writing the information aobut the components
+            scanned.
+        """
+        fc = Gtk.FileChooserDialog(title="Save data file", action=Gtk.FileChooserAction.SAVE)
+        fc.set_current_name("scanned_components.csv")
+        fc.add_buttons(
+            Gtk.STOCK_CANCEL,
+            Gtk.ResponseType.CANCEL,
+            Gtk.STOCK_OPEN,
+            Gtk.ResponseType.OK,
+        )
+        response = fc.run()
+        if response != Gtk.ResponseType.OK:
+            fc.destroy()
+            return
+
+        filename = fc.get_filename()
+
+        if Path(filename).exists():
+            overwrite = dbGtkUtils.ask_for_confirmation("File already exists. Overwrite?", "{}".format(filename))
+            if not overwrite:
+                fc.destroy()
+                return
+
+        fc.destroy()
+        self.close_btn.set_sensitive(True)
+        self.open_btn.set_sensitive(False)
+        self.file_name.set_text("Output File: {}".format(filename))
+        self.ofd = open(filename, "w", encoding="utf-8")
+        self.ofd.write("SN,AltID,Type,Location,StageCode\n")
+        self.ofd.flush()
+
+    def on_file_close(self, *args):
+        """File closed."""
+        if self.ofd is not None:
+            self.ofd.close()
+            self.ofd = None
+            self.close_btn.set_sensitive(False)
+            self.open_btn.set_sensitive(True)
+            self.file_name.set_text("No output file selected")
+
+    def get_qrcode(self, txt):
+        """Gets data from QR scanner."""
+        is_vtrx = False
+        if findVTRx.is_vtrx(txt):
+            try:
+                SN = findVTRx.find_vtrx(self.session, txt)
+                is_vtrx = True
+            except ValueError as e:
+                self.write_message("Error: {}\n".format(e))
+                return
+        else:
+            SN = txt
+
+        obj = ITkDButils.get_DB_component(self.session, SN)
+        if obj is None:
+            self.write_message("Object {} not found in DB\n\n".format(SN))
+            return
+
+        if is_vtrx and obj["alternativeIdentifier"] is None:
+            obj["alternativeIdentifier"] = txt
+
+        msg = "\n{}\nObject SN: {}\nObject Alt. ID: {}\nObject Type: {}\nObject Loc. {}\nObject stage: {} - {}\n\n".format(
+            txt,
+            obj["serialNumber"],
+            obj["alternativeIdentifier"],
+            obj["componentType"]["name"],
+            obj["currentLocation"]["name"],
+            obj["currentStage"]["code"],
+            obj["currentStage"]["name"])
+        self.write_message(msg)
+
+        msg = "{}, {}, {}, {}, {}\n".format(
+            obj["serialNumber"],
+            obj["alternativeIdentifier"],
+            obj["componentType"]["name"],
+            obj["currentLocation"]["code"],
+            obj["currentStage"]["code"])
+
+        if self.ofd is not None:
+            self.ofd.write(msg)
+            self.ofd.flush()
+
+    def test_qrcode(self, *args):
+        """Gets data from QR scanner."""
+        txt = "a3c671bf38d3957dc053c6e5471aa27e"
+        self.write_message("{}\n".format(txt))
+
+        if findVTRx.is_vtrx(txt):
+            try:
+                SN = findVTRx.find_vtrx(self.session, txt)
+            except ValueError as e:
+                self.write_message("Error: {}\n".format(e))
+                return
+        else:
+            SN = txt
+
+        obj = ITkDButils.get_DB_component(self.session, SN)
+        if obj is None:
+            self.write_message("Object {} not found in DB\n\n".format(SN))
+            return
+
+
+        msg = "\n\nObject SN: {}\nObject Alt. ID: {}\nObject Type: {}\nObject Loc.: {}\nObject stage: {} - {}\n".format(
+            obj["serialNumber"],
+            obj["alternativeIdentifier"],
+            obj["componentType"]["name"],
+            obj["currentLocation"]["name"],
+            obj["currentStage"]["code"],
+            obj["currentStage"]["name"])
+
+        self.write_message(msg)
+        self.write_message("")
+
+def main():
+    """Main entry."""
+    HELP_LINK="https://itkdb-gtk.docs.cern.ch/uploadSingleTest.html"
+
+    # DB login
+    dlg = ITkDBlogin.ITkDBlogin()
+    client = dlg.get_client()
+    if client is None:
+        print("Could not connect to DB with provided credentials.")
+        dlg.die()
+        sys.exit()
+
+    client.user_gui = dlg
+
+    window = FindComponent(client, help_link=HELP_LINK)
+    window.set_accept_focus(True)
+    window.present()
+    window.connect("destroy", Gtk.main_quit)
+
+    try:
+        Gtk.main()
+
+    except KeyboardInterrupt:
+        print("Arrrgggg!!!")
+
+    dlg.die()
+
+if __name__ == "__main__":
+    main()

itkdb_gtk/findVTRx.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+"""Tells which VTRx correspods to the package serial number scanned."""
+
+import re
+
+__valid_sn__ = re.compile("J-(SD|MT)[0-9]+")
+
+def is_vtrx(SN):
+    """Checks if the SN corresponds to a VTRx."""
+    if __valid_sn__.match(SN) is None:
+        return False
+    return True
+
+def find_vtrx(client, SN):
+    """Searches VTRx."""
+    if not is_vtrx(SN):
+        return None
+    
+    payload = {
+        "filterMap": {
+            "project": "CE",
+            "componentType": ["VTRX"],
+            "propertyFilter": [{"code": "PACKAGE_SN", "operator": "=", "value": SN}],
+        }
+    }
+    out = client.get("listComponentsByProperty", json=payload)
+    vtrx = None
+    nitem = 0
+    for item in out:
+        vtrx = item["serialNumber"]
+        nitem += 1
+
+    if nitem > 1:
+        raise ValueError("Too many VTRx with same device SN.")
+
+    return vtrx

itkdb_gtk/readGoogleSheet.py

@@ -13,7 +13,7 @@ SCOPES = ['https://www.googleapis.com/auth/spreadsheets.readonly']
 
 
 # Get spreadsheet ID from share link
-re_sheet_id = re.compile(r"https://docs.google.com/spreadsheets/d/(?P<ID>\w+)", re.DOTALL)
+re_sheet_id = re.compile(r"https://docs.google.com/spreadsheets/d/(?P<ID>[\w-]+)", re.DOTALL)
 
 
 def get_spreadsheet_service():
@@ -55,7 +55,6 @@ def get_spreadsheet_data(url, data_range):
     """Get the data from the given range in teh google sheet.
 
     Args:
-    ----
         url: google sheet document url
         data_range: the data range, ej. inventory!A2:Z
 

itkdb_gtk/untrash_component.py

@@ -0,0 +1,35 @@
+#!/usr/bin/env python3
+"""Un trash a trashed component."""
+
+
+def untrash_component(client, SN, reason="Accidentally trashed"):
+    """Un trash given component.
+
+    Args:
+        SN (str): Serial number of component to recover.
+        reason (str): message for the DB
+    Returna:
+        dict: PDB response.
+
+    """
+    DTO = {
+        'component': SN,
+        'trashed': False,
+        'reason': reason
+    }
+
+    response = client.post('setComponentTrashed', json=DTO)
+    return response
+
+
+if __name__ == "__main__":
+    import sys
+    from itkdb_gtk import ITkDBlogin
+    dlg = ITkDBlogin.ITkDBlogin()
+    client = dlg.get_client()
+
+    try:
+        response = untrash_component(client, sys.argv[1])
+
+    except Exception as E:
+        print(str(E))

{itkdb_gtk-0.0.3.dist-info → itkdb_gtk-0.20.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
-Name: itkdb-gtk
-Version: 0.0.3
+Metadata-Version: 2.4
+Name: itkdb_gtk
+Version: 0.20.1
 Summary: A collection of Gtk based GUI to access ITkDB.
 Author-email: Carlos Lacasta <carlos.lacasta@cern.ch>
 Project-URL: Homepage, https://gitlab.cern.ch/atlas-itk/sw/db/itk-pdb-gtk-gui-utils
@@ -9,12 +9,14 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
-Requires-Dist: itkdb (>=0.4.0)
+Requires-Dist: itkdb>=0.4.0
 Requires-Dist: numpy
+Requires-Dist: matplotlib
 Requires-Dist: openpyxl
 Requires-Dist: pyserial
-Requires-Dist: python-dateutil
+Requires-Dist: python_dateutil
 Requires-Dist: requests
+Requires-Dist: PyGObject
 
 # Interaction with the ITk PDB.
 
@@ -42,25 +44,32 @@ Reads the AVS Production Sheet and FAT reports. With this information it creates
 the petal core in the DB, if not yet there, make the assembly of the components,
 and uploadas the test runs made at AVS.
 
-## uploadTest.py
+## UploadTest.py
 A GUI to upload the JSON files corresponding to a test and, also, to add
 attachmetns.
 
-## getShipments.py
+## UploadMultipleTests.py
+This will allow to select various test files, or even scan a whole directory to
+find them, and assign comments, defects or attachments to each of the tests found.
+
+## GetShipments.py
 Find all shipments to be received at a given site and list them. It handles a
 barcode reader that helps identifying the items actually received for the
 reception. It will finally make the DB aware of the items receptioned.
 
-## sendShipments.py
+## CreateShipments.py
 Create a new shipment. Allows to add items with the QR reader as well as from a
 GUI dialog. One can add comments and attachments to the shipment.
 
-## groundingTest.py
-Allows to enter valueos, comments and defects for those ot items in the gounding
+## GroundVITests.py
+Allows to upload and enter valueos, comments and defects for those items in the gounding
 and visual instpections tests of the petal core.
 
+## UploadModuleIV.py
+The idea behind this is that we messure the IV on a Ring module and on only one of the to Half modules. The IV of the remaining half modules is derived from the other 2. Eventually the IV test can be uploaded to the DB.
+
 ## dashBoard.py
 This is an launcher application from which we can start most of the other
 applications. It is a very good starting point. There is a Gnome desktop file (ITkDB.desktop)
-that needs to be copied to $HOME/.local/share/applications and an icon (ITkDB.svg) that
-needs to go to $HOME/.local/share/icons
+that needs to be copied to `$HOME/.local/share/applications` and an icon (ITkDB.svg) that
+needs to go to `$HOME/.local/share/icons`

itkdb_gtk-0.20.1.dist-info/RECORD

@@ -0,0 +1,30 @@
+itkdb_gtk/CreateShipments.py,sha256=zE6FZ5WJebgK4MX8ps0FdcIqE51jiL00-k4TdZfrH8E,13206
+itkdb_gtk/GetShipments.py,sha256=iiW4A7SQ8oMtUdwSdK_mOTaBr2xcc3Grl8edB4hcX-c,18728
+itkdb_gtk/GlueWeight.py,sha256=IHgvbN2DlT8Ou-DMi-n3Z-3xSrPr0tlrkkEH10AshrA,11587
+itkdb_gtk/ITkDB.desktop,sha256=v_K4mHsDxb912J1XGo6mOlbW2TkHvYNGrKmiOnsBQqM,172
+itkdb_gtk/ITkDB.svg,sha256=Ry702zrUkxvG61SqThbUNfXySyiLMqalwYpcM-b_KWo,24242
+itkdb_gtk/ITkDBlogin.py,sha256=40tipm_j5eUS4dnZnBT8VyL6Bu_8csuqS9TPWKxvKSY,10038
+itkdb_gtk/ITkDButils.py,sha256=mcvxxvg1Sqhw4cGe6BUi2Zq4tDNWFWD49izFPI8qhzo,21632
+itkdb_gtk/PanelVisualInspection.py,sha256=7IWrAOxn4T6S84BqeaVmwQg92_vi9ZPZqIMrDATEAww,20653
+itkdb_gtk/QRScanner.py,sha256=igSxIyaPSbWxChhd6du4MKuQ0m6Sd3irrAt374KZlB0,3480
+itkdb_gtk/SensorUtils.py,sha256=fYWF9TeutAbore53dLWNlZnVn9P3OsKYcFLNGOs8cnI,15426
+itkdb_gtk/ShowAttachments.py,sha256=KExxPCdbcb04XS8JSUkg5xF1McvlB8e9btwctDCKNXU,8498
+itkdb_gtk/ShowComments.py,sha256=OiMTFLnhGbbKRj5x61D517BYHAt-qY5Y1lvR3EQz3c0,3151
+itkdb_gtk/ShowDefects.py,sha256=aVAHeaE5IztmAPEuHwhi06KWo_pi9xX2J1fTLhKyAPI,3530
+itkdb_gtk/UploadModuleIV.py,sha256=sqh52bSxANBwlmVWZlDqFXpqRGf0rV0QsjJWC-tY_qI,17792
+itkdb_gtk/UploadMultipleTests.py,sha256=tFXGyeIC-c1Gh_Fyjr_AddDnKReGMrMphdotRzIsR-s,25301
+itkdb_gtk/UploadTest.py,sha256=ukgJ5-IG12bqa1QIp3bXIV8hkdXCv5UDxh1lQswN_ko,16832
+itkdb_gtk/VisualInspection.py,sha256=sTEhWrWW9Cez2ABLq1nggnvnncT2nZEsb3KWCSN6GCM,9798
+itkdb_gtk/WireBondGui.py,sha256=WFTLOw4l5JxSbvh4vZMxcF65fqlwvNw0fOyEru9ijqQ,39850
+itkdb_gtk/__init__.py,sha256=f7942SiDwfXYQACCkjz1NktKHYGwnxfwxd-nEYr4vvs,1269
+itkdb_gtk/dashBoard.py,sha256=1kNSJn-iJ-ME6Nhtz-TfjFQsdKIlBooExrD8U_LgyEA,12293
+itkdb_gtk/dbGtkUtils.py,sha256=UEdqVdw8OxAmDzVeCKncUq7mUJkCclV7tEZu0dadics,29629
+itkdb_gtk/findComponent.py,sha256=1AJc1a9USO7uFGf1q2Gtu6eroG_EFvSeCS-Jfew4yJw,7681
+itkdb_gtk/findVTRx.py,sha256=BYeVuX2oO_Oib1CTUef6SrFJpmOojSpHqWI2eyNApu0,874
+itkdb_gtk/readGoogleSheet.py,sha256=Lzm_oPWwDqZZzKoBUgsp277F9-wCfr_BA0X4VD2Eolo,2673
+itkdb_gtk/untrash_component.py,sha256=VrN46-f-kF7voOxtoh7OL-bZSWAaIFb7-Xbx6_WT7K8,757
+itkdb_gtk-0.20.1.dist-info/METADATA,sha256=696CpNzA97r8TdHxuspCZ31EsaaH4Dypu0JzlQqu8Pk,3149
+itkdb_gtk-0.20.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+itkdb_gtk-0.20.1.dist-info/entry_points.txt,sha256=vN0_mqIT6NwsnHirFsf3JHu-wq8CSHXE1IlhZzB5sbU,481
+itkdb_gtk-0.20.1.dist-info/top_level.txt,sha256=KVRrH4OS8ovzNR9bvADE0ABn5bNpSk987tuH0jCfkbU,10
+itkdb_gtk-0.20.1.dist-info/RECORD,,

{itkdb_gtk-0.0.3.dist-info → itkdb_gtk-0.20.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.40.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

itkdb_gtk-0.20.1.dist-info/entry_points.txt

@@ -0,0 +1,12 @@
+[gui_scripts]
+createShipments = itkdb_gtk:createShipments
+findDBComponent = itkdb_gtk:findDBComponent
+getShipments = itkdb_gtk:getShipments
+glueWeight = itkdb_gtk:glueWeight
+itkdb_dashBoard = itkdb_gtk:dash_board
+panelVisualInspection = itkdb_gtk:panelVisualInspection
+uploadModuleIV = itkdb_gtk:uploadModuleIV
+uploadMultipleTests = itkdb_gtk:uploadMultipleTests
+uploadTest = itkdb_gtk:uploadTest
+visualInspection = itkdb_gtk:visualInspection
+wirebondTest = itkdb_gtk:wirebondTest

itkdb_gtk/checkComponent.py

@@ -1,131 +0,0 @@
-#!/usr/bin/env python3
-"""Check childre of DB component."""
-
-try:
-    import readAVSdata
-    import ITkDBlogin
-    import ITkDButils
-
-    from dbGtkUtils import replace_in_container, complain, DictDialog, ask_for_confirmation
-
-except ModuleNotFoundError:
-    from itkdb_gtk import readAVSdata, ITkDBlogin, ITkDButils
-    from itkdb_gtk.dbGtkUtils import replace_in_container, complain, DictDialog, ask_for_confirmation
-
-import json
-
-
-def complain(main_msg, secondary_msg=None):
-    """Prints an error message
-
-    Args:
-    -----
-        main (): Main message
-        secondary (): Seconday message
-    """
-    print("**Error\n{}".format(main_msg))
-    if secondary_msg:
-        msg = secondary_msg.replace("\n", "\n\t")
-        print("\t{}".format(msg))
-
-
-def find_petal(session, SN, silent=False):
-    """Finds petal with given SN."""
-    try:
-        petal_core = ITkDButils.get_DB_component(session, SN)
-
-    except Exception as E:
-        if not silent:
-            complain("Could not find Petal Core in DB", str(E))
-
-        petal_core = None
-        return
-
-    try:
-        if petal_core["type"]["code"] != "CORE_AVS":
-            complain("Wrong component type", "This is not an AVS petal core")
-
-        # print(json.dumps(petal_core, indent=3))
-
-    except KeyError:
-        # Petal is not there
-        petal_core = None
-
-    return petal_core
-
-
-def get_type(child):
-    """Return object type
-
-    Args:
-    -----
-        child (): object
-
-    Returns
-    -------
-        str: object type
-        
-    """
-    if child["type"] is not None:
-        ctype = child["type"]["code"]
-    else:
-        ctype = child["componentType"]["code"]
-        
-    return ctype
-
-
-def main():
-    """Main entry point."""
-    # ITk_PB authentication
-    dlg = ITkDBlogin.ITkDBlogin()
-    session = dlg.get_client()
-
-    # find all cores
-    # Now all the objects
-    payload = {
-        "componentType": ["BT"],
-        "componentType": ["CORE_PETAL"],
-        "type": ["CORE_AVS"],
-    }
-    core_list = session.get("listComponents", json=payload)
-
-    for obj in core_list:
-        SN = obj["serialNumber"]
-        id = obj['alternativeIdentifier']
-        core = find_petal(session, SN)
-        if core is None:
-            print("SN: not a petal core.")
-            continue
-
-        location = core["currentLocation"]['code']
-        stage = core["currentStage"]['code']
-        if "children" not in core:
-            complain("{}[{}]".format(SN, id), "Not assembled")
-            continue
-
-        print("Petal {} [{}] - {}. {}".format(SN, id, stage, location))
-        clist = []
-        for child in core["children"]:
-            cstage = "Missing"
-            if child["component"] is None:
-                ctype = get_type(child)
-                clist.append((ctype, cstage, None, None))
-
-            else:
-                SN = child["component"]["serialNumber"]
-                ctype = get_type(child)
-                cobj = ITkDButils.get_DB_component(session, SN)
-                cstage = cobj["currentStage"]['code']
-                clocation = cobj["currentLocation"]['code']
-                clist.append((ctype, cstage, SN, clocation))
-
-        for item in clist:
-            print("\t{} [{}] - {} at {}".format(item[2], item[0], item[1], item[3]))
-            
-        print()
-
-    dlg.die()
-
-
-if __name__ == "__main__":
-    main()