Get title

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")

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")