itkdb-gtk 0.10.7__py3-none-any.whl → 0.10.9__py3-none-any.whl

itkdb_gtk/PetalReceptionTests.py

@@ -1,15 +1,16 @@
 #!/usr/bin/env python3
 """Test dashboard."""
-import json
 import sys
+import copy
+from pathlib import Path
 
 try:
     import itkdb_gtk
-    
+
 except ImportError:
-    from pathlib import Path
     cwd = Path(__file__).parent.parent
     sys.path.append(cwd.as_posix())
+    import itkdb_gtk
 
 from itkdb_gtk import dbGtkUtils, ITkDBlogin, ITkDButils
 
@@ -37,12 +38,12 @@ def find_children(W):
 class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
     """Petl Reception Test GUI."""
 
-    def __init__(self, session, help=None):
+    def __init__(self, session, help_link=None):
         """Initialization."""
         super().__init__(title="ITkDB Dashboard",
                          session=session,
                          show_search="Find object with given SN.",
-                         help=help)
+                         help_link=help_link)
 
         # Members
         self.dbObject = None
@@ -66,7 +67,7 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
 
         self.SN = itkdb_gtk.dbGtkUtils.TextEntry()
         self.SN.connect("text-changed", self.on_SN_changed)
-        
+
         #self.SN = Gtk.Entry()
         #self.SN.connect("focus-in-event", self.on_sn_enter)
         #self.SN.connect("focus-out-event", self.on_sn_leave)
@@ -107,6 +108,8 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
         self.create_test_box("Visual Inspection", "VISUAL_INSPECTION")
         self.create_test_box("Grounding", "GROUNDING_CHECK")
         self.create_test_box("Pipe bending", "BENDING120")
+        self.create_test_box("X-rays", "XRAYIMAGING")
+
 
         # The text view
         self.mainBox.pack_end(self.message_panel.frame, True, True, 5)
@@ -117,7 +120,7 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
         """New SN given. Ask in PDB,"""
         if len(value) <= 0:
             return None
-        
+
         self.query_db()
         current_location = self.dbObject["currentLocation"]["code"]
         dbGtkUtils.set_combo_iter(self.inst_cmb, current_location, 0)
@@ -134,7 +137,7 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
         for i in range(npages):
             page = self.notebook.get_nth_page(i)
             page.dict_dialog.factory_reset()
-        
+
 
     def create_test_box(self, label, test_name, institute="IFIC"):
         """Create and add to notebook a test dialog.
@@ -213,6 +216,27 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
                 page.dict_dialog.values["institution"] = self.institute
                 page.dict_dialog.refresh()
 
+    def upload_this_test(self, values):
+        """Upload a single test."""
+        # print(json.dumps(values, indent=2))
+
+        attachments = []
+        if values["testType"] == "XRAYIMAGING":
+            fnam = values["results"]["IMAGELINK"]
+            if fnam is not None and len(fnam)>0:
+                P = Path(fnam).expanduser().resolve()
+                if P.exists():
+                    A = ITkDButils.Attachment(path=P.as_posix(), title=P.name, desc="X-ray image")
+                    values["results"]["IMAGELINK"] = P.name
+                    attachments.append(A)
+
+        rc = ITkDButils.upload_test(self.session, values, attachments=attachments, check_runNumber=True)
+        if rc is not None:
+            dbGtkUtils.complain("Could not upload test", rc)
+
+        else:
+            self.write_message("Test uploaded. {} - {}\n".format(values["component"], values["testType"]))
+
     def upload_single_test(self, *args):
         """Upload the current test."""
         SN = self.SN.get_text()
@@ -225,16 +249,9 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
         if dctD is None:
             return
 
-        values = dctD.values
+        values = copy.deepcopy(dctD.values)
         values["component"] = SN
-        # print(json.dumps(values, indent=2))
-        rc = ITkDButils.upload_test(self.session, values)
-        if rc is not None:
-            dbGtkUtils.complain("Could not upload test", rc)
-
-        else:
-            dbGtkUtils.ask_for_confirmation("Test uploaded.",
-                                            "{} - {}".format(values["component"], values["testType"]))
+        self.upload_this_test(values)
 
     def upload_tests(self, *args):
         """Upload the current test."""
@@ -251,14 +268,7 @@ class PetalReceptionTests(dbGtkUtils.ITkDBWindow):
 
             values = dctD.values
             values["component"] = SN
-            # print(json.dumps(values, indent=2))
-            rc = ITkDButils.upload_test(self.session, values)
-            if rc is not None:
-                dbGtkUtils.complain("Could not upload test", rc)
-
-            else:
-                self.write_message("Test uploaded. {} - {}\n".format(values["component"], values["testType"]))
-
+            self.upload_this_test(values)
 
 def main():
     """Main entry."""

itkdb_gtk/SensorUtils.py

@@ -1,10 +1,8 @@
 #!/usr/bin/env python3
 """A collection of utilities for sensor data."""
-import io
 from pathlib import Path
-import numpy as np
 import warnings
-import tempfile
+import numpy as np
 from itkdb_gtk import ITkDButils
 
 #
@@ -210,7 +208,7 @@ def LocateMicroDischarge(
         V_BDV_ind = V_BDV_ind[ind]
         if V_avg_BD.shape == (0,):
             return NO_BD_CONST
-        
+
     return NO_BD_CONST
 
 
@@ -260,8 +258,12 @@ def sensor_data_to_json(session, mdata, mod_type, logger=None):
     passed = True
 
     # Find Current @ 500V
-    indx = np.where(V == 500)[0]
-    i_500 = I[indx][0] / area
+    try:
+        indx = np.where(V == 500)[0]
+        i_500 = I[indx][0] / area
+    except IndexError:
+        i_500 = 999
+
     if logger:
         logger.write_message("I @ 500V = {:.2f} nA/cm2\n".format(i_500))
 
@@ -296,7 +298,7 @@ def sensor_data_to_json(session, mdata, mod_type, logger=None):
             )
             if logger:
                 logger.write_message("...{}. FAILED\n".format(txt))
-                
+
             passed = False
     else:
         if MicroDischargeV == NO_BD_CONST:
@@ -388,7 +390,7 @@ def save_sensor_data(fnam, mdata, name=None):
     Args:
         fnam: file name or file object.
         mdata (dict): data as returned by :function:`read_sensor_file`
-        
+
     """
     if hasattr(fnam, "write") and callable(fnam.write):
         data_out = fnam
@@ -396,17 +398,17 @@ def save_sensor_data(fnam, mdata, name=None):
             fnam = name
         else:
             fnam = build_file_name(mdata)
-            
+
     else:
         data_out = open(fnam, 'w', encoding="utf-8")
         fnam = Path(fnam).name
-        
+
     is_module = "Module_SN" in mdata
     if is_module:
         SN = mdata["Module_SN"]
     else:
         SN = mdata["Component"]
-        
+
     if is_module:
         items = [
             "Type",
@@ -446,7 +448,7 @@ def save_sensor_data(fnam, mdata, name=None):
             "Humidity",
             "Comments",
         ]
-    
+
     data_out.write("{}\n".format(fnam))
     for key in items:
         if key == "Module_SN" or key == "Component":
@@ -469,7 +471,7 @@ def save_sensor_data(fnam, mdata, name=None):
     V = np.abs(mdata["curve"]["V"])
     I = np.abs(mdata["curve"]["I"])
     S = np.abs(mdata["curve"]["S"])
-    
+
     ndata = len(V)
     for i in range(ndata):
         data_out.write("{:10.2f}\t{:10.2f}\t{:10.2f}\n".format(V[i], I[i], S[i]))
@@ -487,4 +489,4 @@ def build_file_name(mdata):
         SN = mdata["Component"]
         fnam = "{}-W{}_IV_{}".format(mdata["Batch"], mdata["Wafer"], mdata["RunNumber"])
 
-    return fnam
+    return fnam

itkdb_gtk/ShowAttachments.py

@@ -232,7 +232,9 @@ class ShowAttachments(Gtk.Dialog):
         rc = dlg.run()
         if rc == Gtk.ResponseType.OK:
             path = Path(dlg.fC.get_filename())
-            A = ITkDButils.Attachment(path, dlg.att_title.get_text().strip(), dlg.att_desc.get_text().strip())
+            A = ITkDButils.Attachment(path=path, 
+                                      title=dlg.att_title.get_text().strip(), 
+                                      desc=dlg.att_desc.get_text().strip())
             self.append_attachment_to_view(A)
             self.attachments.append(A)
 

itkdb_gtk/UploadModuleIV.py

@@ -16,7 +16,6 @@ import json
 import tempfile
 import copy
 from pathlib import Path
-import shutil
 
 from matplotlib.backends.backend_gtk3agg import FigureCanvasGTK3Agg as FigureCanvas
 from matplotlib.backends.backend_gtk3 import NavigationToolbar2GTK3 as NavigationToolbar
@@ -56,11 +55,12 @@ def remove_files(W, flist):
 class IVwindow(dbGtkUtils.ITkDBWindow):
     """GUI for IV file handling."""
 
-    def __init__(self, session, title="IV window", options=None, help=None):
+    def __init__(self, session, title="IV window", options=None, help_link=None):
         """Initialization."""
-        super().__init__(
-            session=session, title=title, show_search=None, gtk_runs=gtk_runs, help=help
-        )
+        super().__init__(session=session, title=title,
+                         show_search=None, gtk_runs=gtk_runs,
+                         help_link=help_link
+                         )
         self.mdata = {}
         self.mod_type = {}
         self.mod_SN = {}
@@ -179,7 +179,7 @@ class IVwindow(dbGtkUtils.ITkDBWindow):
         js_out.write(json.dumps(test, indent=3, cls=dbGtkUtils.MyEncoder))
         js_out.close()
 
-        attachment = ITkDButils.Attachment(data_out.name, "resultsFile", fnam)
+        attachment = ITkDButils.Attachment(path=data_out.name, title="resultsFile", desc=fnam)
         uploadW = UploadTest.UploadTest(self.session, js_out.name, attachment)
         uploadW.connect("destroy", remove_files, [data_out.name, js_out.name])
 
@@ -247,7 +247,7 @@ class IVwindow(dbGtkUtils.ITkDBWindow):
             rc = fc.run()
             if rc == Gtk.ResponseType.OK:
                 SensorUtils.save_sensor_data(fc.get_filename(), mdata)
-                
+
             fc.hide()
             fc.destroy()
 
@@ -256,7 +256,7 @@ class IVwindow(dbGtkUtils.ITkDBWindow):
 
         if do_both.get_active():
             self.upload_test(self.mdata["single"], self.mod_type["single"])
-            
+
         dlg.destroy()
 
     def on_refresh(self, *args):

itkdb_gtk/UploadMultipleTests.py

@@ -11,7 +11,6 @@ try:
     import itkdb_gtk
 
 except ImportError:
-    from pathlib import Path
     cwd = Path(__file__).parent.parent
     sys.path.append(cwd.as_posix())
 
@@ -103,15 +102,17 @@ def check_data(data):
 class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
     """Collects information to upload a test and its attachments."""
 
-    def __init__(self, session, help=help):
+    def __init__(self, session, help_link=None):
         """Initialization.
 
         Args:
             session: ITkDB session
 
         """
-        super().__init__(session=session, title="Upload Tests", gtk_runs=gtk_runs, help=help)
+        super().__init__(session=session, title="Upload Tests", gtk_runs=gtk_runs, help_link=help_link)
         self.tests = []
+        self.data = None
+        self.tree = None
 
         self.init_window()
 
@@ -224,12 +225,12 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
         # double click shows attachments
         if event.button == 1 and event.type == Gdk.EventType.DOUBLE_BUTTON_PRESS:
             select = self.tree.get_selection()
-            model, iter = select.get_selected()
-            if not iter:
+            model, lv_iter = select.get_selected()
+            if not lv_iter:
                 return
 
-            self.on_show_json(None, (model, iter, model[iter]))
-            # self.on_show_attachments(None, (model, iter, model[iter]))
+            self.on_show_json(None, (model, lv_iter, model[lv_iter]))
+            # self.on_show_attachments(None, (model, lv_iter, model[lv_iter]))
             return
 
         if event.button != 3:
@@ -237,44 +238,44 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
 
         # Create popup menu
         select = self.tree.get_selection()
-        model, iter = select.get_selected()
+        model, lv_iter = select.get_selected()
         values = None
-        if iter:
-            values = model[iter]
+        if lv_iter:
+            values = model[lv_iter]
 
-        if not iter:
+        if not lv_iter:
             P = tree.get_path_at_pos(event.x, event.y)
             if P:
                 print(P[0].to_string())
-                iter = model.get_iter(P[0])
-                values = model[iter]
+                lv_iter = model.get_iter(P[0])
+                values = model[lv_iter]
 
         if not values:
             return
 
         menu = Gtk.Menu()
         item_show = Gtk.MenuItem(label="Show JSOn")
-        item_show.connect("activate", self.on_show_json, (model, iter, values))
+        item_show.connect("activate", self.on_show_json, (model, lv_iter, values))
         menu.append(item_show)
 
         item_set_stage = Gtk.MenuItem(label="Set Stage")
-        item_set_stage.connect("activate", self.on_set_stage, (model, iter, values))
+        item_set_stage.connect("activate", self.on_set_stage, (model, lv_iter, values))
         menu.append(item_set_stage)
 
         item_show_att = Gtk.MenuItem(label="Edit Attachments")
-        item_show_att.connect("activate", self.on_show_attachments, (model, iter, values))
+        item_show_att.connect("activate", self.on_show_attachments, (model, lv_iter, values))
         menu.append(item_show_att)
 
         item_show_com = Gtk.MenuItem(label="Edit Comments")
-        item_show_com.connect("activate", self.on_show_comments, (model, iter, values))
+        item_show_com.connect("activate", self.on_show_comments, (model, lv_iter, values))
         menu.append(item_show_com)
 
         item_show_def = Gtk.MenuItem(label="Edit Defects")
-        item_show_def.connect("activate", self.on_show_defects, (model, iter, values))
+        item_show_def.connect("activate", self.on_show_defects, (model, lv_iter, values))
         menu.append(item_show_def)
 
         item_del = Gtk.MenuItem(label="Delete")
-        item_del.connect("activate", self.on_delete_tests, (model, iter, values))
+        item_del.connect("activate", self.on_delete_tests, (model, lv_iter, values))
         menu.append(item_del)
         menu.show_all()
 
@@ -282,17 +283,17 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
 
     def on_show_json(self, item, data):
         """Test JSon."""
-        model, iter, val = data
+        model, lv_iter, val = data
         payload = val[TestList.Json]
         value, dlg = create_json_data_editor(payload)
         rc = dlg.run()
         if rc == Gtk.ResponseType.OK:
             payload = value.values
-            model.set_value(iter, TestList.Json, payload)
-            model.set_value(iter, TestList.SN, payload["component"])
-            model.set_value(iter, TestList.RunNumber, payload["runNumber"])
-            model.set_value(iter, TestList.Date, handle_test_date(payload["date"]))
-            model.set_value(iter, TestList.Institute, handle_test_date(payload["institution"]))
+            model.set_value(lv_iter, TestList.Json, payload)
+            model.set_value(lv_iter, TestList.SN, payload["component"])
+            model.set_value(lv_iter, TestList.RunNumber, payload["runNumber"])
+            model.set_value(lv_iter, TestList.Date, handle_test_date(payload["date"]))
+            model.set_value(lv_iter, TestList.Institute, handle_test_date(payload["institution"]))
 
 
         dlg.hide()
@@ -300,48 +301,48 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
 
     def on_show_attachments(self, item, data):
         """Show the attachmetns."""
-        model, iter, val = data
+        model, lv_iter, val = data
 
         SA = ShowAttachments("Test Attachments", self.session, val[TestList.Attachments], parent=self)
         response = SA.run()
         if response == Gtk.ResponseType.OK:
-            model.set_value(iter, TestList.Attachments, SA.attachments)
-            model.set_value(iter, TestList.Nattch, len(SA.attachments))
+            model.set_value(lv_iter, TestList.Attachments, SA.attachments)
+            model.set_value(lv_iter, TestList.Nattch, len(SA.attachments))
 
         SA.hide()
         SA.destroy()
 
     def on_show_comments(self, item, data):
         """Show comments"""
-        model, iter, val = data
+        model, lv_iter, val = data
         SC = ShowComments("Test Comments", val[TestList.Comments], self)
         rc = SC.run()
         if rc == Gtk.ResponseType.OK:
-            model.set_value(iter, TestList.Comments, SC.comments)
-            model.set_value(iter, TestList.Ncomm, len(SC.comments))
+            model.set_value(lv_iter, TestList.Comments, SC.comments)
+            model.set_value(lv_iter, TestList.Ncomm, len(SC.comments))
 
         SC.hide()
         SC.destroy()
 
     def on_show_defects(self, item, data):
         """Show comments"""
-        model, iter, val = data
+        model, lv_iter, val = data
         SD = ShowDefects("Test Defects", val[TestList.Defects], self)
         rc = SD.run()
         if rc == Gtk.ResponseType.OK:
-            model.set_value(iter, TestList.Defects, SD.defects)
-            model.set_value(iter, TestList.Ndef, len(SD.defects))
+            model.set_value(lv_iter, TestList.Defects, SD.defects)
+            model.set_value(lv_iter, TestList.Ndef, len(SD.defects))
 
         SD.hide()
         SD.destroy()
 
     def on_delete_tests(self, item, data):
         """Test edit."""
-        model, iter, val = data
+        model, lv_iter, val = data
         rc = dbGtkUtils.ask_for_confirmation("Remove this test?",
                                              "{} - {}".format(val[TestList.SN], val[TestList.TestType]))
         if rc:
-            model.remove(iter)
+            model.remove(lv_iter)
 
     def get_test_institute(self):
         """Select an institue."""
@@ -373,7 +374,7 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
 
     def on_set_stage(self, item, data):
         """Set the test stage."""
-        model, iter, val = data
+        model, lv_iter, val = data
         SN = val[TestList.SN]
         combo, currentStage = self.get_component_stages(SN)
 
@@ -392,9 +393,9 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
         rc = dlg.run()
         if rc == Gtk.ResponseType.OK:
             new_stage = combo.get_active_text()
-            model[iter][TestList.Stage] = new_stage
-            data = model[iter][TestList.Json]
-            changed = (new_stage != model[iter][TestList.currentStage])
+            model[lv_iter][TestList.Stage] = new_stage
+            data = model[lv_iter][TestList.Json]
+            changed = (new_stage != model[lv_iter][TestList.currentStage])
             data["isRetroactive"] = changed
             if changed:
                 data["stage"] = new_stage
@@ -402,7 +403,7 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
                 if "stage" in data:
                     del data["stage"]
 
-            model[iter][TestList.Json] = data
+            model[lv_iter][TestList.Json] = data
 
 
         dlg.hide()
@@ -437,9 +438,9 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
         for ifile in files:
             try:
                 has_errors = False
-                data = json.loads(open(ifile).read())
+                data = json.loads(open(ifile, 'r', encoding="UTF-8").read())
                 errors, missing = check_data(data)
-                if len(missing):
+                if len(missing) > 0:
                     self.write_message("{}\n".format(Path(ifile).name))
                     self.write_message("Some keys are missing in the JSon file.\n")
                     self.write_message("{}\n".format("\n".join(['\t'+line for line in missing])))
@@ -469,10 +470,12 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
                             if path.exists():
                                 path = path.expanduser().resolve()
                             else:
-                                path = folder / path
+                                path = folder / path.name
 
                             if path.exists():
-                                attachments.append(ITkDButils.Attachment(path, att["title"], att["description"]))
+                                attachments.append(ITkDButils.Attachment(path=path, 
+                                                                         title=att["title"], 
+                                                                         desc=att["description"]))
                             else:
                                 self.write_message("Ignoring atachment {}".format(data["path"]))
 
@@ -596,15 +599,15 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
     def upload_test(self):
         """Uploads tests and attachments."""
         model = self.tree.get_model()
-        iter = model.get_iter_first()
-        while iter:
+        lv_iter = model.get_iter_first()
+        while lv_iter:
             past_iter = None
-            values = model[iter]
+            values = model[lv_iter]
             payload = values[TestList.Json]
             payload["comments"] = values[TestList.Comments]
             payload["defects"] = values[TestList.Defects]
 
-            rc = ITkDButils.upload_test(self.session, payload, values[TestList.Attachments])
+            rc = ITkDButils.upload_test(self.session, payload, values[TestList.Attachments], check_runNumber=True)
             if rc:
                 ipos = rc.find("The following details may help:")
                 msg = rc[ipos:]
@@ -613,9 +616,9 @@ class UploadMultipleTests(dbGtkUtils.ITkDBWindow):
 
             else:
                 self.write_message("Upload {}-{} successfull\n".format(payload["component"], payload["testType"]))
-                past_iter = iter
+                past_iter = lv_iter
 
-            iter = model.iter_next(iter)
+            lv_iter = model.iter_next(lv_iter)
             if past_iter:
                 model.remove(past_iter)
 

itkdb_gtk/UploadPetalInformation.py

@@ -16,7 +16,6 @@ try:
     import itkdb_gtk
 
 except ImportError:
-    from pathlib import Path
     cwd = Path(__file__).parent.parent
     sys.path.append(cwd.as_posix())
 
@@ -124,11 +123,11 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
         grid.attach(Gtk.Label(label="Serial No."), 0, 0, 1, 1)
         grid.attach(self.SN, 1, 0, 1, 1)
         grid.attach(self.btn_state, 2, 0, 1, 1)
-        
+
 
         grid.attach(Gtk.Label(label="Prod. Sheet"), 0, 1, 1, 1)
         grid.attach(self.btnPSF, 1, 1, 1, 1)
-        
+
         btn = Gtk.Button(label="Reset")
         btn.connect("clicked", self.on_reset)
         grid.attach(btn, 2, 1, 1, 1)
@@ -209,7 +208,7 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
 
         btn = Gtk.Button(label="Upload Tests")
         btn.set_tooltip_text("Upload all tests")
-        
+
         btn.connect("clicked", self.on_upload)
         btnBox.add(btn)
 
@@ -361,7 +360,7 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
 
         # Check if we need to assemble the module
         self.check_assembly(self.DESY_comp)
-        
+
         self.check_components()
 
     def on_fat_set(self, *args):
@@ -394,7 +393,7 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
         self.update_scroll_window("delamination", j_del)
         self.update_scroll_window("grounding", j_gnd)
         self.update_scroll_window("metrology", j_mtr)
-        
+
         self.check_tests(True)
 
     def read_avs_files(self, widgets):
@@ -473,7 +472,7 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
     def on_check_components(self, *args):
         """Button clicked."""
         self.check_components()
-        
+
     def check_components(self):
         """Check that components are in DB."""
         for cmp, cmp_SN in self.DESY_comp.items():
@@ -558,7 +557,7 @@ class AVSPanel(dbGtkUtils.ITkDBWindow):
                         if rc is None:
                             print("Could not set final stage of {}".format(cSN))
 
-            if len(error_txt):
+            if len(error_txt)>0:
                 dbGtkUtils.complain("Assembly of {} could not be completeed:".format(this_petal),
                          "\n".join(error_txt))
 
@@ -708,4 +707,4 @@ def main():
     sys.exit()
 
 if __name__ == "__main__":
-    main()
+    main()