Get base
This commit is contained in:
@@ -1,4 +1,5 @@
|
|||||||
from _connection import _conn_dict as conn
|
from _connection import _conn_dict as conn
|
||||||
|
from psycopg.rows import dict_row
|
||||||
|
|
||||||
_NODE = "_NODE"
|
_NODE = "_NODE"
|
||||||
_LINK = "_LINK"
|
_LINK = "_LINK"
|
||||||
@@ -52,6 +53,28 @@ def have_pattern(name: str, id: str) -> bool:
|
|||||||
return _have_impl(name, id, _PATTERN)
|
return _have_impl(name, id, _PATTERN)
|
||||||
|
|
||||||
|
|
||||||
|
# get
|
||||||
|
def _get_impl(name: str, id: str, table: str) -> dict[str, str]:
|
||||||
|
with conn[name].cursor(row_factory=dict_row) as cur:
|
||||||
|
cur.execute(f"SELECT * FROM {table} WHERE ID = '{id}'")
|
||||||
|
if cur.rowcount > 0:
|
||||||
|
return cur.fetchone()
|
||||||
|
else:
|
||||||
|
return {}
|
||||||
|
|
||||||
|
def get_node(name: str, id: str) -> dict[str, str]:
|
||||||
|
return _get_impl(name, id, _NODE)
|
||||||
|
|
||||||
|
def get_link(name: str, id: str) -> dict[str, str]:
|
||||||
|
return _get_impl(name, id, _LINK)
|
||||||
|
|
||||||
|
def get_curve(name: str, id: str) -> dict[str, str]:
|
||||||
|
return _get_impl(name, id, _CURVE)
|
||||||
|
|
||||||
|
def get_pattern(name: str, id: str) -> dict[str, str]:
|
||||||
|
return _get_impl(name, id, _PATTERN)
|
||||||
|
|
||||||
|
|
||||||
# delete
|
# delete
|
||||||
|
|
||||||
def _delete_impl(name: str, id: str, table: str) -> None:
|
def _delete_impl(name: str, id: str, table: str) -> None:
|
||||||
|
|||||||
@@ -4,21 +4,25 @@ from _0_base import *
|
|||||||
open_project("net")
|
open_project("net")
|
||||||
|
|
||||||
add_node("net", "n-1", JUNCTION)
|
add_node("net", "n-1", JUNCTION)
|
||||||
|
print(get_node("net", "n-1"))
|
||||||
if have_node("net", "n-1"):
|
if have_node("net", "n-1"):
|
||||||
delete_node("net", "n-1")
|
delete_node("net", "n-1")
|
||||||
print(have_node("net", "n-1"))
|
print(have_node("net", "n-1"))
|
||||||
|
|
||||||
add_link("net", "l-1", PIPE)
|
add_link("net", "l-1", PIPE)
|
||||||
|
print(get_link("net", "l-1"))
|
||||||
if have_link("net", "l-1"):
|
if have_link("net", "l-1"):
|
||||||
delete_link("net", "l-1")
|
delete_link("net", "l-1")
|
||||||
print(have_link("net", "l-1"))
|
print(have_link("net", "l-1"))
|
||||||
|
|
||||||
add_curve("net", "c-1")
|
add_curve("net", "c-1")
|
||||||
|
print(get_curve("net", "c-1"))
|
||||||
if have_curve("net", "c-1"):
|
if have_curve("net", "c-1"):
|
||||||
delete_curve("net", "c-1")
|
delete_curve("net", "c-1")
|
||||||
print(have_curve("net", "c-1"))
|
print(have_curve("net", "c-1"))
|
||||||
|
|
||||||
add_pattern("net", "p-1")
|
add_pattern("net", "p-1")
|
||||||
|
print(get_pattern("net", "p-1"))
|
||||||
if have_pattern("net", "p-1"):
|
if have_pattern("net", "p-1"):
|
||||||
delete_pattern("net", "p-1")
|
delete_pattern("net", "p-1")
|
||||||
print(have_pattern("net", "p-1"))
|
print(have_pattern("net", "p-1"))
|
||||||
|
|||||||
Reference in New Issue
Block a user