385 lines
20 KiB
Python
Executable File
385 lines
20 KiB
Python
Executable File
import csv
|
|
import os
|
|
import re
|
|
import sys
|
|
import weakref
|
|
from glob import glob
|
|
|
|
from lib.db import Recipes, Steps, Users
|
|
from PyQt5.QtCore import QTimer, pyqtSignal
|
|
from PyQt5.QtGui import QKeySequence
|
|
from PyQt5.QtWidgets import QFileDialog, QShortcut
|
|
from ui.crud import Crud, Json_External_Dialog_Editor_Cell_Widget
|
|
from ui.helpers import replace_widget
|
|
from ui.recipe_spec_and_step_editor import Recipe_Spec_And_Step_Editor
|
|
from ui.widget import Widget
|
|
|
|
|
|
class Noner:
|
|
def __getitem__(self, key):
|
|
return None
|
|
|
|
|
|
class Recipe_Selection(Widget):
|
|
ok = pyqtSignal(Recipes)
|
|
|
|
def __init__(self, config, unsupported_steps=None):
|
|
super().__init__()
|
|
self.config = config
|
|
session = Users.get_session()
|
|
if session.is_admin:
|
|
# readonly = ["id"]
|
|
readonly = False
|
|
crud_aliases = {
|
|
# "id": "Id",
|
|
"name": "Ricetta",
|
|
"client": "Cliente",
|
|
"part_number": "N° disegno",
|
|
"spec": "Specifica",
|
|
"description": "Descrizione",
|
|
# "archived": "Archiviata",
|
|
}
|
|
filters = None
|
|
else:
|
|
readonly = True
|
|
crud_aliases = {
|
|
"name": "Ricetta",
|
|
"client": "Cliente",
|
|
"part_number": "N° disegno",
|
|
"spec": "Specifica",
|
|
"description": "Descrizione",
|
|
# "archived": "Archiviata",
|
|
}
|
|
filters = {"archived": False}
|
|
with open("config/machine_settings/recipes_defaults.csv", "r") as f:
|
|
self.defaults = dict(next(csv.DictReader(f)))
|
|
step_defaults = self.read_steps(self.defaults, Noner())
|
|
self.crud = Crud(
|
|
"recipes",
|
|
display_name="SELEZIONE RICETTA",
|
|
readonly=readonly,
|
|
select=list(crud_aliases.keys()),
|
|
filters=filters,
|
|
fields_aliases=crud_aliases,
|
|
# autocomplete={
|
|
# "spec": {
|
|
# "step_editors": step_defaults.update({
|
|
# "vision": {
|
|
# # "recipe": sorted(glob("*.ini", root_dir="./config/vision/recipes/")), # only in python3.10
|
|
# "recipe": sorted(map(os.path.basename, glob("./config/vision/recipes/*.ini"))),
|
|
# },
|
|
# "print": {
|
|
# # "template": sorted(glob("*.prn", root_dir="./config/label_templates/")), # only in python3.10
|
|
# "template": sorted(map(os.path.basename, glob("./config/label_templates/*.prn"))),
|
|
# },
|
|
# }),
|
|
# },
|
|
# "archived": False,
|
|
# },
|
|
widget_classes={"spec": lambda *args, **kwargs: Json_External_Dialog_Editor_Cell_Widget(
|
|
Recipe_Spec_And_Step_Editor,
|
|
*args,
|
|
**kwargs,
|
|
unsupported_steps=unsupported_steps
|
|
), },
|
|
pagination=25,
|
|
)
|
|
self.crud.delete_b.setEnabled(False)
|
|
self.crud.delete_b.setVisible(False)
|
|
replace_widget(self, "crud_w", self.crud)
|
|
self.crud_modified = None
|
|
self.selected = None
|
|
self.select_b.setEnabled(False)
|
|
self.select_b.clicked.connect(self.select)
|
|
QShortcut(QKeySequence("Return"), self).activated.connect(self.select_b.click)
|
|
QShortcut(QKeySequence("Enter"), self).activated.connect(self.select_b.click)
|
|
self.crud.modified.connect(self.check_modified)
|
|
self.crud.selected.connect(self.check_selected)
|
|
self.crud.emit()
|
|
self.crud.db_tw.setColumnWidth(0, 200)
|
|
self.crud.db_tw.setColumnWidth(1, 200)
|
|
self.crud.db_tw.setColumnWidth(2, 200)
|
|
self.crud.db_tw.setColumnWidth(3, 200)
|
|
self.crud.db_tw.setColumnWidth(4, 200)
|
|
self.crud.db_tw.setColumnWidth(5, 400)
|
|
self.import_b.clicked.connect(lambda checked, self=weakref.ref(self): self().import_recipes())
|
|
self.export_b.clicked.connect(lambda checked, self=weakref.ref(self): self().export_recipes())
|
|
# TESTING
|
|
if "--auto-select" in sys.argv or "--test" in sys.argv:
|
|
recipe = "TEST"
|
|
cn = self.crud.select_index["name"]
|
|
self.crud.db_tw.clearSelection()
|
|
for rn in range(1, self.crud.db_tw.rowCount()):
|
|
if self.crud.db_tw.cellWidget(rn, cn).text() == recipe:
|
|
selection = self.crud.db_tw.model().index(rn, cn)
|
|
self.crud.db_tw.setCurrentIndex(selection)
|
|
break
|
|
self.test_timer = QTimer()
|
|
self.test_timer.setSingleShot(True)
|
|
self.test_timer.timeout.connect(self.select_b.clicked.emit)
|
|
self.test_timer.start(500)
|
|
# /TESTING
|
|
|
|
def check_modified(self, modified):
|
|
self.crud_modified = modified
|
|
self.check(self.crud_modified, self.selected)
|
|
|
|
def check_selected(self, selected=None):
|
|
if selected is not None and len(selected) == 1:
|
|
selected = selected[0] - 1 # - 1 because rn starts from 1 (filters line)
|
|
if selected >= 0 and selected < len(self.crud.data_index):
|
|
selected = self.crud.data_index[selected]
|
|
if selected is not None:
|
|
self.selected = selected
|
|
else:
|
|
self.selected = None
|
|
else:
|
|
self.selected = None
|
|
self.check(self.crud_modified, self.selected)
|
|
|
|
def check(self, modified, selected):
|
|
self.select_b.setEnabled(modified is False and selected is not None)
|
|
|
|
def select(self):
|
|
if self.selected is not None:
|
|
self.ok.emit(self.crud.db.table_model.get_by_id(self.selected))
|
|
|
|
def read_steps(self, row, defaults=None):
|
|
if defaults is None:
|
|
defaults = self.defaults
|
|
return {
|
|
"connector": {
|
|
"connector": row.get("connettore", defaults["connettore"])
|
|
},
|
|
"barcodes": {
|
|
"serial": row.get(self.config.get("recipe", {}).get("barcode_serial_field", "codice_a_barre").strip(), defaults[self.config.get("recipe", {}).get("barcode_serial_field", "codice_a_barre").strip()]),
|
|
},
|
|
"resistance": {
|
|
"scale": float(row.get("scala_resistenza", defaults["scala_resistenza"])),
|
|
"expected": float(row.get("resistenza_attesa", defaults["resistenza_attesa"])),
|
|
"tolerance": float(row.get("tolleranza_resistenza", defaults["tolleranza_resistenza"]))
|
|
},
|
|
"leak_1": {
|
|
"pre_filling_time": int(row.get("tempo_pre_riempimento", defaults["tempo_pre_riempimento"])),
|
|
"pre_filling_pressure": int(row.get("pressione_pre_riempimento", defaults["pressione_pre_riempimento"])),
|
|
"filling_time": int(row.get("tempo_riempimento", defaults["tempo_riempimento"])),
|
|
"settling_time": int(row.get("tempo_assestamento", defaults["tempo_assestamento"])),
|
|
"settling_pressure_min_percent": int(row.get("percentuale_minima_pressione_assestamento", defaults["percentuale_minima_pressione_assestamento"])),
|
|
"settling_pressure_max_percent": int(row.get("percentuale_massima_pressione_assestamento", defaults["percentuale_massima_pressione_assestamento"])),
|
|
"test_time": int(row.get("tempo_di_test", defaults["tempo_di_test"])),
|
|
"test_pressure_min_delta": int(row.get("pressione_di_test_delta_minimo", defaults["pressione_di_test_delta_minimo"])),
|
|
"test_pressure": int(row.get("pressione_di_test", defaults["pressione_di_test"])),
|
|
"test_pressure_max_delta": int(row.get("pressione_di_test_delta_massimo", defaults["pressione_di_test_delta_massimo"])),
|
|
"flush_time": int(row.get("tempo_svuotamento", defaults["tempo_svuotamento"])),
|
|
"flush_pressure": int(row.get("pressione_svuotmento", defaults["pressione_svuotmento"]))
|
|
},
|
|
"leak_2": {
|
|
"pre_filling_time": int(row.get("tempo_pre_riempimento_2", defaults["tempo_pre_riempimento_2"])),
|
|
"pre_filling_pressure": int(row.get("pressione_pre_riempimento_2", defaults["pressione_pre_riempimento_2"])),
|
|
"filling_time": int(row.get("tempo_riempimento_2", defaults["tempo_riempimento_2"])),
|
|
"settling_time": int(row.get("tempo_assestamento_2", defaults["tempo_assestamento_2"])),
|
|
"settling_pressure_min_percent": int(row.get("percentuale_minima_pressione_assestamento_2", defaults["percentuale_minima_pressione_assestamento_2"])),
|
|
"settling_pressure_max_percent": int(row.get("percentuale_massima_pressione_assestamento_2", defaults["percentuale_massima_pressione_assestamento_2"])),
|
|
"test_time": int(row.get("tempo_di_test_2", defaults["tempo_di_test_2"])),
|
|
"test_pressure_min_delta": int(row.get("pressione_di_test_delta_minimo_2", defaults["pressione_di_test_delta_minimo_2"])),
|
|
"test_pressure": int(row.get("pressione_di_test_2", defaults["pressione_di_test_2"])),
|
|
"test_pressure_max_delta": int(row.get("pressione_di_test_delta_massimo_2", defaults["pressione_di_test_delta_massimo_2"])),
|
|
"flush_time": int(row.get("tempo_svuotamento_2", defaults["tempo_svuotamento_2"])),
|
|
"flush_pressure": int(row.get("pressione_svuotmento_2", defaults["pressione_svuotmento_2"]))
|
|
},
|
|
"vision": {
|
|
"recipe": row.get("ricetta_visione", defaults["ricetta_visione"]),
|
|
},
|
|
"print": {
|
|
"template": row.get(self.config.get("recipe", {}).get("label_template_field", "modello_etichetta").strip(), defaults[self.config.get("recipe", {}).get("label_template_field", "modello_etichetta").strip()]),
|
|
},
|
|
}
|
|
|
|
def import_recipes(self, csv_path=None, defaults=None):
|
|
if defaults is None:
|
|
defaults = self.defaults
|
|
if csv_path is None:
|
|
csv_path, _ = QFileDialog.getOpenFileName(
|
|
None,
|
|
"Esportazione ricette",
|
|
"ricette.csv",
|
|
"CSV data (*.csv);;All Files (*)",
|
|
)
|
|
csv_path = str(csv_path)
|
|
if not len(csv_path):
|
|
return
|
|
self.log.info(f"recipes: importing recipes from {csv_path}")
|
|
with open(csv_path, "r") as f:
|
|
reader = csv.DictReader(f)
|
|
count = 0
|
|
for row in reader:
|
|
recipe_name = row.get(self.config.get("recipe", {}).get("recipe_name_field", "codice_ricetta").strip(), row.get(defaults.get(self.config.get("recipe", {}).get("recipe_name_field", "codice_ricetta").strip(), "name")))
|
|
try:
|
|
recipe = Recipes.get_by_id(recipe_name)
|
|
recipe_is_new = False
|
|
except Recipes.DoesNotExist:
|
|
recipe = Recipes(name=recipe_name)
|
|
recipe_is_new = True
|
|
steps = {}
|
|
for step_type, step_spec in self.read_steps(row, defaults=defaults).items():
|
|
try:
|
|
if recipe.spec is None:
|
|
raise Steps.DoesNotExist()
|
|
step = Steps.get_by_id(recipe.spec.get("available_steps", {}).get(step_type, None))
|
|
step_is_new = False
|
|
except Steps.DoesNotExist:
|
|
step = Steps()
|
|
step_is_new = True
|
|
step.type = re.sub(r"^(.*)_[0-9]+$", r"\1", step_type)
|
|
step.spec = step_spec
|
|
if step_is_new:
|
|
step.save(force_insert=True)
|
|
else:
|
|
step.save()
|
|
steps[step_type] = step
|
|
recipe.client = row.get("cliente", defaults["cliente"])
|
|
recipe.part_number = row.get("part_number", defaults["part_number"])
|
|
recipe.spec = {
|
|
"connector": bool(len(row.get("verifica_connettore_abilitata", defaults["verifica_connettore_abilitata"]))),
|
|
"barcodes": bool(len(row.get("verifica_codice_a_barre_abilitata", defaults["verifica_codice_a_barre_abilitata"]))),
|
|
"resistance": bool(len(row.get("verifica_resistenza_connettore_abilitata", defaults["verifica_resistenza_connettore_abilitata"]))),
|
|
"leak_1": bool(len(row.get("prova_tenuta_abilitata", defaults["prova_tenuta_abilitata"]))),
|
|
"leak_2": bool(len(row.get("prova_tenuta_abilitata_2", defaults["prova_tenuta_abilitata_2"]))),
|
|
"vision": bool(len(row.get("test_visione_abilitato", defaults["test_visione_abilitato"]))),
|
|
"print": bool(len(row.get("stampa_etichetta_abilitata", defaults["stampa_etichetta_abilitata"]))),
|
|
"steps": [], # should be pks of the enabled steps
|
|
"available_steps": {
|
|
"connector": steps["connector"].name,
|
|
"barcodes": steps["barcodes"].name,
|
|
"resistance": steps["resistance"].name,
|
|
"leak_1": steps["leak_1"].name,
|
|
"leak_2": steps["leak_2"].name,
|
|
"vision": steps["vision"].name,
|
|
"print": steps["print"].name,
|
|
},
|
|
}
|
|
for step_name, step in recipe.spec["available_steps"].items():
|
|
if recipe.spec[step_name]:
|
|
recipe.spec["steps"].append(step)
|
|
if recipe_is_new:
|
|
recipe.save(force_insert=True)
|
|
else:
|
|
recipe.save()
|
|
count += 1
|
|
self.log.info(f"recipes: imported {count} rows.")
|
|
self.crud.refresh()
|
|
|
|
def export_recipes(self, csv_path=None):
|
|
if csv_path is None:
|
|
csv_path, _ = QFileDialog.getSaveFileName(
|
|
None,
|
|
"Esportazione ricette",
|
|
"ricette.csv",
|
|
"CSV data (*.csv);;All Files (*)",
|
|
)
|
|
csv_path = str(csv_path)
|
|
if not len(csv_path):
|
|
return
|
|
if not csv_path.lower().endswith(".csv"):
|
|
csv_path += ".csv"
|
|
csv_dir = os.path.dirname(csv_path)
|
|
if len(csv_dir):
|
|
os.makedirs(csv_dir, exist_ok=True)
|
|
data = []
|
|
fieldnames = [
|
|
self.config.get("recipe", {}).get("recipe_name_field", "codice_ricetta").strip(),
|
|
"cliente",
|
|
"part_number",
|
|
"verifica_connettore_abilitata",
|
|
"connettore",
|
|
"verifica_codice_a_barre_abilitata",
|
|
self.config.get("recipe", {}).get("barcode_serial_field", "codice_a_barre").strip(),
|
|
"verifica_resistenza_connettore_abilitata",
|
|
"scala_resistenza",
|
|
"resistenza_attesa",
|
|
"tolleranza_resistenza",
|
|
"prova_tenuta_abilitata",
|
|
"tempo_pre_riempimento",
|
|
"pressione_pre_riempimento",
|
|
"tempo_riempimento",
|
|
"tempo_assestamento",
|
|
"percentuale_minima_pressione_assestamento",
|
|
"percentuale_massima_pressione_assestamento",
|
|
"tempo_di_test",
|
|
"pressione_di_test_delta_minimo",
|
|
"pressione_di_test",
|
|
"pressione_di_test_delta_massimo",
|
|
"tempo_svuotamento",
|
|
"pressione_svuotmento",
|
|
"prova_tenuta_abilitata_2",
|
|
"tempo_pre_riempimento_2",
|
|
"pressione_pre_riempimento_2",
|
|
"tempo_riempimento_2",
|
|
"tempo_assestamento_2",
|
|
"percentuale_minima_pressione_assestamento_2",
|
|
"percentuale_massima_pressione_assestamento_2",
|
|
"tempo_di_test_2",
|
|
"pressione_di_test_delta_minimo_2",
|
|
"pressione_di_test_2",
|
|
"pressione_di_test_delta_massimo_2",
|
|
"tempo_svuotamento_2",
|
|
"pressione_svuotmento_2",
|
|
"test_visione_abilitato",
|
|
"ricetta_visione",
|
|
"stampa_etichetta_abilitata",
|
|
self.config.get("recipe", {}).get("label_template_field", "modello_etichetta").strip(),
|
|
]
|
|
for recipe in Recipes.select():
|
|
steps = recipe.get_steps_map()
|
|
exportable = {
|
|
self.config.get("recipe", {}).get("recipe_name_field", "codice_ricetta").strip(): recipe.name,
|
|
"cliente": recipe.client,
|
|
"part_number": recipe.part_number,
|
|
"verifica_connettore_abilitata": "x" if recipe.spec["connector"] else "",
|
|
"connettore": steps["connector"].spec["connector"],
|
|
"verifica_codice_a_barre_abilitata": "x" if recipe.spec["barcodes"] else "",
|
|
self.config.get("recipe", {}).get("barcode_serial_field", "codice_a_barre").strip(): steps["barcodes"].spec["serial"],
|
|
"verifica_resistenza_connettore_abilitata": "x" if recipe.spec["resistance"] else "",
|
|
"scala_resistenza": steps["resistance"].spec["scale"],
|
|
"resistenza_attesa": steps["resistance"].spec["expected"],
|
|
"tolleranza_resistenza": steps["resistance"].spec["tolerance"],
|
|
"prova_tenuta_abilitata": "x" if recipe.spec["leak_1"] else "",
|
|
"tempo_pre_riempimento": steps["leak_1"].spec["pre_filling_time"],
|
|
"pressione_pre_riempimento": steps["leak_1"].spec["pre_filling_pressure"],
|
|
"tempo_riempimento": steps["leak_1"].spec["filling_time"],
|
|
"tempo_assestamento": steps["leak_1"].spec["settling_time"],
|
|
"percentuale_minima_pressione_assestamento": steps["leak_1"].spec["settling_pressure_min_percent"],
|
|
"percentuale_massima_pressione_assestamento": steps["leak_1"].spec["settling_pressure_max_percent"],
|
|
"tempo_di_test": steps["leak_1"].spec["test_time"],
|
|
"pressione_di_test_delta_minimo": steps["leak_1"].spec["test_pressure_min_delta"],
|
|
"pressione_di_test": steps["leak_1"].spec["test_pressure"],
|
|
"pressione_di_test_delta_massimo": steps["leak_1"].spec["test_pressure_max_delta"],
|
|
"tempo_svuotamento": steps["leak_1"].spec["flush_time"],
|
|
"pressione_svuotmento": steps["leak_1"].spec["flush_pressure"],
|
|
"prova_tenuta_abilitata_2": "x" if recipe.spec["leak_2"] else "",
|
|
"tempo_pre_riempimento_2": steps["leak_2"].spec["pre_filling_time"],
|
|
"pressione_pre_riempimento_2": steps["leak_2"].spec["pre_filling_pressure"],
|
|
"tempo_riempimento_2": steps["leak_2"].spec["filling_time"],
|
|
"tempo_assestamento_2": steps["leak_2"].spec["settling_time"],
|
|
"percentuale_minima_pressione_assestamento_2": steps["leak_2"].spec["settling_pressure_min_percent"],
|
|
"percentuale_massima_pressione_assestamento_2": steps["leak_2"].spec["settling_pressure_max_percent"],
|
|
"tempo_di_test_2": steps["leak_2"].spec["test_time"],
|
|
"pressione_di_test_delta_minimo_2": steps["leak_2"].spec["test_pressure_min_delta"],
|
|
"pressione_di_test_2": steps["leak_2"].spec["test_pressure"],
|
|
"pressione_di_test_delta_massimo_2": steps["leak_2"].spec["test_pressure_max_delta"],
|
|
"tempo_svuotamento_2": steps["leak_2"].spec["flush_time"],
|
|
"pressione_svuotmento_2": steps["leak_2"].spec["flush_pressure"],
|
|
"test_visione_abilitato": recipe.spec["vision"],
|
|
"ricetta_visione": steps["vision"].spec["recipe"],
|
|
"stampa_etichetta_abilitata": "x" if recipe.spec["print"] else "",
|
|
self.config.get("recipe", {}).get("label_template_field", "modello_etichetta").strip(): steps["print"].spec["template"],
|
|
}
|
|
data.append(exportable)
|
|
if len(data):
|
|
self.log.info(f"recipes: exporting recipes to {csv_path}")
|
|
with open(csv_path, "w", newline="") as f:
|
|
w = csv.DictWriter(f, fieldnames, extrasaction="ignore")
|
|
w.writeheader()
|
|
w.writerows(data)
|
|
self.log.info(f"recipes: exported {len(data)} rows.")
|