108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
# === Ajout de produit personnalisé ===
|
|
import streamlit as st
|
|
import networkx as nx
|
|
from utils.translations import _
|
|
from utils.persistance import get_champ_statut, maj_champ_statut, supprime_champ_statut
|
|
|
|
def ajouter_produit(G: nx.DiGraph) -> nx.DiGraph:
|
|
st.markdown(f"## {str(_('pages.personnalisation.add_new_product'))}")
|
|
|
|
# Restauration des produits personnalisés sauvegardés
|
|
if "pages.personnalisation.create_product.fait" not in st.session_state and get_champ_statut("pages.personnalisation.create_product.fait") == "oui":
|
|
index = 0
|
|
while True:
|
|
new_prod = get_champ_statut(f"pages.personnalisation.create_product.{index}.nom")
|
|
if new_prod == "":
|
|
break
|
|
|
|
G.add_node(new_prod, niveau="0", personnalisation="oui", label=new_prod)
|
|
|
|
sel_new_op = get_champ_statut(f"pages.personnalisation.create_product.{index}.edge")
|
|
if sel_new_op:
|
|
G.add_edge(new_prod, sel_new_op)
|
|
|
|
i = 0
|
|
while True:
|
|
comp = get_champ_statut(f"pages.personnalisation.create_product.{index}.composants.{i}")
|
|
if comp == "":
|
|
break
|
|
G.add_edge(new_prod, comp)
|
|
i += 1
|
|
|
|
index += 1
|
|
|
|
niveau1 = sorted([n for n, d in G.nodes(data=True) if d.get("niveau") == "1"])
|
|
ops_dispo = sorted([n for n, d in G.nodes(data=True) if d.get("niveau") == "assemblage"])
|
|
|
|
if "pages.personnalisation.add.new_product_name" not in st.session_state:
|
|
st.session_state["pages.personnalisation.add.new_product_name"] = get_champ_statut("pages.personnalisation.add.new_product_name")
|
|
|
|
new_prod = st.text_input(
|
|
str(_("pages.personnalisation.new_product_name")),
|
|
key="pages.personnalisation.add.new_product_name"
|
|
)
|
|
|
|
if new_prod:
|
|
if new_prod in G.nodes:
|
|
st.warning(str(_("pages.personnalisation.product_exists")))
|
|
return G
|
|
|
|
if "pages.personnalisation.add.assembly_operation" not in st.session_state:
|
|
st.session_state["pages.personnalisation.add.assembly_operation"] = get_champ_statut("pages.personnalisation.add.assembly_operation")
|
|
|
|
if "pages.personnalisation.add.components_to_link" not in st.session_state:
|
|
composants = []
|
|
i = 0
|
|
while True:
|
|
val = get_champ_statut(f"pages.personnalisation.add.components_to_link.{i}")
|
|
if val == "":
|
|
break
|
|
composants.append(val)
|
|
i += 1
|
|
st.session_state["pages.personnalisation.add.components_to_link"] = composants
|
|
|
|
ops_dispo = sorted([
|
|
n for n, d in G.nodes(data=True)
|
|
if d.get("niveau") == "10"
|
|
and any(G.has_edge(p, n) and G.nodes[p].get("niveau") == "0" for p in G.predecessors(n))
|
|
])
|
|
sel_new_op = st.selectbox(str(_("pages.personnalisation.assembly_operation")), ops_dispo,
|
|
key="pages.personnalisation.add.assembly_operation")
|
|
|
|
sel_comps = st.multiselect(
|
|
str(_("pages.personnalisation.components_to_link")),
|
|
options=niveau1,
|
|
key="pages.personnalisation.add.components_to_link"
|
|
)
|
|
|
|
if st.button(str(_("pages.personnalisation.create_product"))):
|
|
G.add_node(new_prod, niveau="0", personnalisation="oui", label=new_prod)
|
|
|
|
# Trouver le prochain index disponible
|
|
index = 0
|
|
while get_champ_statut(f"pages.personnalisation.create_product.{index}.nom") != "":
|
|
index += 1
|
|
|
|
maj_champ_statut(f"pages.personnalisation.create_product.{index}.nom", new_prod)
|
|
|
|
if sel_new_op != str(_("pages.personnalisation.none")):
|
|
G.add_edge(new_prod, sel_new_op)
|
|
maj_champ_statut(f"pages.personnalisation.create_product.{index}.edge", sel_new_op)
|
|
|
|
for j, comp in enumerate(sel_comps):
|
|
G.add_edge(new_prod, comp)
|
|
maj_champ_statut(f"pages.personnalisation.create_product.{index}.composants.{j}", comp)
|
|
|
|
maj_champ_statut("pages.personnalisation.create_product.fait", "oui")
|
|
|
|
# Nettoyage de session et des champs persistants
|
|
del st.session_state["pages.personnalisation.add.new_product_name"]
|
|
del st.session_state["pages.personnalisation.add.assembly_operation"]
|
|
del st.session_state["pages.personnalisation.add.components_to_link"]
|
|
supprime_champ_statut("pages.personnalisation.add")
|
|
|
|
st.success(f"{new_prod} {str(_('pages.personnalisation.added'))}")
|
|
st.rerun()
|
|
|
|
return G
|