LW00BMDZ

1import sqlite3
2class DBHelper:
3 def __init__(self, dbname="botdb.db"):
4        self.dbname = dbname
5        self.conn = sqlite3.connect(dbname,check_same_thread=False)
6
7 def setup(self):
8     print("creating table")
9     stmt = "CREATE TABLE IF NOT EXISTS list_db (User_id int,List_item  text,item_index int AUTO_INCREMENT)"
10     self.conn.execute(stmt)
11     self.conn.commit()
12
13 def add_item(self, user_id, text):
14     stmt = "INSERT INTO list_db (User_id, List_item ) VALUES (?, ?)"
15     args = (user_id, text)
16     self.conn.execute(stmt, args)
17     self.conn.commit()
18
19 def delete_item(self, user_id,messageid):
20     stmt = "DELETE FROM list_db WHERE User_id = (?) AND item_index = (?)"
21     args = (user_id, messageid)
22     self.conn.execute(stmt, args)
23     self.conn.commit()
24
25 def get_index(self,userid):
26     stmt = "SELECT item_index, FROM list_db WHERE User_id =(?) AND item_index = (?)"
27     args = (userid,)
28     return [x[0] for x in self.conn.execute(stmt, args)]
29
30 def get_values(self,userid):
31     stmt = "SELECT List_item FROM list_db WHERE User_id =(?)"
32     args = (userid,)
33     return [x[0] for x in self.conn.execute(stmt, args)]
34
35 def get_values2(self,userid):
36     stmt = "SELECT List_item FROM list_db WHERE User_id =(?)"
37     args = (userid,)
38     return [x[0] for x in self.conn.execute(stmt, args)]