forked from coder2020official/pyLivegramBot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconnection.py
46 lines (41 loc) · 1.21 KB
/
connection.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import sqlite3
def create_db_new():
db = sqlite3.connect('users.db', check_same_thread=False)
sql = db.cursor()
sql.execute('''CREATE TABLE IF NOT EXISTS USERS(
user_id INTEGER,
first_name VARCHAR,
messageid INT,
message VARCHAR)''')
sql.execute('''CREATE TABLE IF NOT EXISTS blocked(
user_id INT)''')
sql.execute('''CREATE TABLE IF NOT EXISTS user(
user_id INT)''')
sql.close()
db.close()
def database_query(query: str,spec_args):
"""Performs database commands.
:params:
query: str - this should be your command to be executed"""
db = sqlite3.connect('users.db', check_same_thread=False)
with db:
sql = db.cursor()
sql.execute(query,spec_args)
result = sql.fetchall()
if db:
db.commit()
sql.close()
return result
def database_query_spec(query: str):
"""Performs database commands.
:params:
query: str - this should be your command to be executed"""
db = sqlite3.connect('users.db', check_same_thread=False)
with db:
sql = db.cursor()
sql.execute(query)
result = sql.fetchall()
if db:
db.commit()
sql.close()
return result