diff --git a/api/_1_title.py b/api/_1_title.py index b6f2ab9..b3571d5 100644 --- a/api/_1_title.py +++ b/api/_1_title.py @@ -1,4 +1,5 @@ from _connection import _conn_dict as conn +from psycopg.rows import dict_row def have_title(name: str) -> bool: with conn[name].cursor() as cur: @@ -8,11 +9,19 @@ def have_title(name: str) -> bool: def set_title(name: str, value: str) -> None: if have_title(name): with conn[name].cursor() as cur: - cur.execute(f"UPDATE TITLE SET Value = {value}") + cur.execute(f"UPDATE TITLE SET Value = '{value}'") else: with conn[name].cursor() as cur: cur.execute(f"INSERT INTO TITLE (Value) VALUES ('{value}')") +def get_title(name: str) -> str: + with conn[name].cursor(row_factory=dict_row) as cur: + cur.execute(f"SELECT * FROM TITLE") + if cur.rowcount > 0: + return cur.fetchone()['value'] + else: + return "" + def unset_title(name: str) -> None: with conn[name].cursor() as cur: cur.execute(f"TRUNCATE TITLE") diff --git a/api/_1_title_test.py b/api/_1_title_test.py index 51c2011..bc0f9af 100644 --- a/api/_1_title_test.py +++ b/api/_1_title_test.py @@ -4,9 +4,26 @@ from _1_title import * open_project("net") print(have_title("net")) +print(get_title("net")) + set_title("net", "xxx") print(have_title("net")) +print(get_title("net")) + +set_title("net", "xxxx") +print(have_title("net")) +print(get_title("net")) + unset_title("net") print(have_title("net")) +print(get_title("net")) + +set_title("net", "xxxx") +print(have_title("net")) +print(get_title("net")) + +unset_title("net") +print(have_title("net")) +print(get_title("net")) close_project("net") \ No newline at end of file