@@ -132,12 +132,11 @@ def update_entry(
132132 """
133133 try :
134134 cursor = self .connection .cursor ()
135-
136- # TODO: check if username needs to be made dynamic
137- query = f"UPDATE { table_name } "
138- f"SET { column } = '{ data } ' "
139- f"WHERE username = '{ user } ';"
140-
135+ query = (
136+ f"UPDATE { table_name } "
137+ f"SET { column } = '{ data } ' "
138+ f"WHERE username = '{ user } ';"
139+ )
141140 cursor .execute (query )
142141 except Exception as e :
143142 self .connection .rollback ()
@@ -156,11 +155,10 @@ def search_entry(self, table_name: str, user: str, column: str):
156155 try :
157156 records = None
158157 cursor = self .connection .cursor ()
159-
160- query = f"""SELECT { column }
161- FROM { table_name }
162- WHERE username = '{ user } ';"""
163-
158+ query = (
159+ f"SELECT { column } FROM { table_name } "
160+ f"WHERE username = '{ user } ';"
161+ )
164162 cursor .execute (query )
165163 records = cursor .fetchall ()
166164
@@ -190,9 +188,13 @@ def append_entry(
190188 """
191189 try :
192190 cursor = self .connection .cursor ()
193- query = f"UPDATE { table_name } "
194- f"SET { column } = array_append({ column } ,'{ sender } ') "
195- f"WHERE username = '{ receiver } ';"
191+
192+ query = (
193+ f"UPDATE { table_name } "
194+ f"SET { column } = array_append({ column } ,'{ sender } ') "
195+ f"WHERE username = '{ receiver } ';"
196+ )
197+
196198 cursor .execute (query )
197199 self .connection .commit ()
198200 except Exception as e :
@@ -218,11 +220,13 @@ def remove_from_array(
218220 """
219221 try :
220222 cursor = self .connection .cursor ()
223+
221224 query = (
222225 f"UPDATE { table_name } "
223226 f"SET { column } = ARRAY_REMOVE({ column } , '{ value_to_remove } ')"
224227 f"WHERE username = '{ user } ';"
225228 )
229+
226230 cursor .execute (query )
227231 self .connection .commit ()
228232
@@ -245,7 +249,9 @@ def print_table(self, tablename: str):
245249 """
246250 try :
247251 cursor = self .connection .cursor ()
252+
248253 query = f"SELECT * FROM { tablename } ;"
254+
249255 cursor .execute (query )
250256 records = cursor .fetchall ()
251257
@@ -281,6 +287,7 @@ def search_user(self, table_name: str, user: str) -> bool:
281287 f"SELECT username FROM { table_name } "
282288 f"WHERE username = '{ user } ';"
283289 )
290+
284291 cursor .execute (query )
285292 record = list (cursor .fetchall ())
286293
0 commit comments