def cluster_connect(conn):
n_units = st.number_input("Number of nodes", value=2)
units = dict()
for i in range(n_units):
units[i] = ui.known_field_input(conn, f"Source {i+1}:", offset=i)
add = st.button("Connect")
if add:
for node_1, node_2 in combinations(units.values(), 2):
db.write_edge(conn, node_1, node_2)
st.write(f"Added: {node_1}:{node_2}")
conn.commit()
ui.confirm()
def trail_node_entry(conn):
nodes = db.list_nodes(conn)
node_1 = ui.known_field_input(conn, "Existing", offset=0)
node_2 = st.text_input('New')
nonn_button = st.button("Add and connect")
if node_1 and node_2 and nonn_button:
db.write_node(conn, node_2)
db.write_edge(conn, node_1, node_2)
conn.commit()
ui.confirm()
return None
def edge_entry(conn):
node_1 = ui.known_field_input(conn, "Source", offset=0)
node_2 = ui.known_field_input(conn, "Target", offset=1)
add_button, del_button = ui.add_del()
if node_1 and node_2 and add_button:
db.write_edge(conn, node_1, node_2)
conn.commit()
ui.confirm()
if node_1 and node_2 and del_button:
found = True
while found:
found = db.del_edge(conn, node_1, node_2)
st.write("(Edge deleted.)")
conn.commit()
ui.confirm()
return None
def dyad_entry(conn):
node_1 = st.text_input('Enter new node 1 name')
node_2 = st.text_input('Enter new node 2 name')
add_button, del_button = ui.add_del()
if node_1 and node_2 and add_button:
db.write_node(conn, node_1)
db.write_node(conn, node_2)
db.write_edge(conn, node_1, node_2)
conn.commit()
ui.confirm()
if node_1 and node_2 and del_button:
db.del_edge(conn, node_1, node_2)
db.del_node(conn, node_1)
db.del_node(conn, node_2)
conn.commit()
ui.confirm()
return None
def triad_entry(conn):
parent = st.text_input('Enter head node name')
left = st.text_input('Enter left child name')
right = st.text_input('Enter right child name')
add_button, del_button = ui.add_del()
if parent and left and right and add_button:
db.write_node(conn, parent)
db.write_node(conn, left)
db.write_node(conn, right)
db.write_edge(conn, parent, left)
db.write_edge(conn, parent, right)
conn.commit()
ui.confirm()
if parent and left and right and del_button:
db.del_node(conn, parent)
db.del_node(conn, left)
db.del_node(conn, right)
db.del_edge(conn, parent, left)
db.del_edge(conn, parent, right)
conn.commit()
ui.confirm()
return None