· 4 years ago · Mar 16, 2021, 10:26 AM
1from sqlite3 import connect
2
3class DataBase():
4 # Data base initialization
5 def __init__(self):
6 self.db = connect('main_data_base.db')
7 self.cursor = self.db.cursor()
8
9 self.cursor.execute("""CREATE TABLE IF NOT EXISTS sellers (
10 seller_name TEXT,
11 seller_chat_name TEXT,
12 chat_code INT,
13 seller_chat_id INT)""")
14
15 self.cursor.execute("""CREATE TABLE IF NOT EXISTS users (
16 user_name TEXT,
17 user_location TEXT,
18 current_seller TEXT,
19 user_rate_quality INT,
20 user_rate_time INT,
21 user_rate_service INT,
22 user_comment TEXT)""")
23
24 self.db.commit()
25 # -----------------------
26
27 # Create new record in table "sellers"
28 def new_rec_sellers(self, seller_name, seller_chat_name, chat_code=None, seller_chat_id=None):
29 self.cursor.execute('INSERT INTO sellers VALUES(?, ?, ?, ?)', (seller_name, seller_chat_name, chat_code, seller_chat_id))
30 self.db.commit()
31 # ------------------------
32
33 # Create new record in table "users"
34 def new_rec_users(self, user_name, current_seller=None, user_location=None, user_rate_quality=None, user_rate_time=None, user_rate_service=None, user_comment=None):
35 self.cursor.execute('INSERT INTO users VALUES(?, ?, ?, ?, ?, ?, ?)', (user_name, current_seller, user_location, user_rate_quality, user_rate_time, user_rate_service, user_comment))
36 self.db.commit()
37 # ------------------------
38
39 def in_list(self, column='*', table='sellers', one_element=False):
40 self.cursor.execute(f'SELECT {column} FROM {table}')
41 self.db_list = self.cursor.fetchall()
42
43 if one_element:
44 for i in range(len(self.db_list)):
45 self.db_list[i] = self.db_list[i][0]
46
47 return self.db_list
48
49
50 def in_list_where(self, column='*', where=None, table='sellers', one_element=False):
51 self.cursor.execute(f'SELECT {column} FROM {table} {where}')
52 self.db_list_where = self.cursor.fetchall()
53
54 if one_element:
55 for i in range(len(self.db_list_where)):
56 self.db_list_where[i] = self.db_list_where[i][0]
57
58 return self.db_list_where
59
60 def update(self, set_data, where_data, table='sellers'):
61 self.cursor.execute(f'UPDATE {table} SET {set_data[0]} = ? WHERE {where_data[0]} = ?', (set_data[1], where_data[1]))
62
63 self.db.commit()
64
65