From 1c5d0984bf60311f3c29e3387e7f16f897e398c3 Mon Sep 17 00:00:00 2001
From: wqy <wqyuyu@outlook.com>
Date: Thu, 1 Sep 2022 22:38:01 +0800
Subject: [PATCH] Get title

---
 api/_1_title.py      | 11 ++++++++++-
 api/_1_title_test.py | 17 +++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)

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