53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
import streamlit as st
|
|
import json
|
|
|
|
def importer_exporter_graph(G):
|
|
st.markdown("## Sauvegarder ou restaurer la configuration")
|
|
if st.button("Exporter configuration"):
|
|
nodes = [n for n, d in G.nodes(data=True) if d.get("personnalisation") == "oui"]
|
|
edges = [(u, v) for u, v in G.edges() if u in nodes]
|
|
conf = {"nodes": nodes, "edges": edges}
|
|
json_str = json.dumps(conf, ensure_ascii=False)
|
|
st.download_button(
|
|
label="Télécharger (JSON)",
|
|
data=json_str,
|
|
file_name="config_personnalisation.json",
|
|
mime="application/json"
|
|
)
|
|
|
|
uploaded = st.file_uploader("Importer une configuration JSON (max 100 Ko)", type=["json"])
|
|
if uploaded:
|
|
if uploaded.size > 100 * 1024:
|
|
st.error("Fichier trop volumineux (max 100 Ko).")
|
|
else:
|
|
try:
|
|
conf = json.loads(uploaded.read().decode("utf-8"))
|
|
all_nodes = conf.get("nodes", [])
|
|
all_edges = conf.get("edges", [])
|
|
|
|
if not all_nodes:
|
|
st.warning("Aucun produit trouvé dans le fichier.")
|
|
else:
|
|
st.markdown("### Sélection des produits à restaurer")
|
|
sel_nodes = st.multiselect(
|
|
"Produits à restaurer",
|
|
options=all_nodes,
|
|
default=all_nodes,
|
|
key="restaurer_selection"
|
|
)
|
|
|
|
if st.button("Restaurer les éléments sélectionnés", type="primary"):
|
|
for node in sel_nodes:
|
|
if not G.has_node(node):
|
|
G.add_node(node, niveau="0", personnalisation="oui", label=node)
|
|
|
|
for u, v in all_edges:
|
|
if u in sel_nodes and v in sel_nodes + list(G.nodes()) and not G.has_edge(u, v):
|
|
G.add_edge(u, v)
|
|
|
|
st.success("Configuration partielle restaurée avec succès.")
|
|
except Exception as e:
|
|
st.error(f"Erreur d'import : {e}")
|
|
|
|
return G
|