-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
23 lines (18 loc) · 813 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import sqlite3
class Database:
def __init__(self, db_file):
self.conn = sqlite3.connect(db_file)
self.cursor = self.conn.cursor()
def user_exists(self, user_id):
with self.conn:
result = self.cursor.execute("SELECT * FROM users WHERE user_id = ?", (user_id,)).fetchmany(1)
return bool(len(result))
def add_users(self, user_id):
with self.conn:
return self.cursor.execute("INSERT INTO users (user_id) VALUES (?)", (user_id,))
def set_active(self, user_id, active):
with self.conn:
return self.cursor.execute("UPDATE users SET active = ? WHERE user_id = ?", (active, user_id,))
def get_users(self):
with self.conn:
return self.cursor.execute("SELECT user_id FROM users").fetchall()