-
Notifications
You must be signed in to change notification settings - Fork 0
/
bot.py
69 lines (57 loc) · 2.08 KB
/
bot.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from vkbottle.bot import Bot
from vkbottle import load_blueprints_from_package
from data.config import BOT_TOKEN, group_id, host, db_name, user, password
import pymysql
def mysql_connect():
connection = pymysql.connect(
host=host,
port=3306,
user=user,
password=password,
database=db_name
)
return connection
if __name__ == "__main__":
bot = Bot(BOT_TOKEN, group_id)
bot.labeler.vbml_ignore_case = True
for bp in load_blueprints_from_package("commands"):
bp.load(bot)
try:
connection = mysql_connect()
cursor = connection.cursor()
cursor.execute("""CREATE TABLE IF NOT EXISTS users(
id int AUTO_INCREMENT,
name varchar(32),
phone varchar(32),
address varchar(500),
num_orders int(15),
status varchar(32),
banned BOOLEAN,
PRIMARY KEY(id)
)""")
connection.commit()
cursor.execute("""CREATE TABLE IF NOT EXISTS orders(
order_id int AUTO_INCREMENT,
user_id int,
name varchar(80),
address varchar(500),
phone varchar(32),
date varchar(32),
order_list varchar(1000),
completed BOOLEAN,
PRIMARY KEY(order_id)
)""")
connection.commit()
cursor.execute("""CREATE TABLE IF NOT EXISTS questions(
ques_id int AUTO_INCREMENT,
user_id int,
text varchar(1000),
completed BOOLEAN,
date varchar(32),
PRIMARY KEY(ques_id)
)""")
connection.commit()
print("[+] Database has been connected")
except Exception as ex:
print(ex)
bot.run_forever()