-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdbFunctions.py
74 lines (57 loc) · 2.74 KB
/
dbFunctions.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
import mysql.connector
from mysql.connector import Error
import pandas as pd
def create_server_connection(host_name, user_name, user_password):
connection = None
try:
connection = mysql.connector.connect(
host=host_name,
user=user_name,
passwd=user_password
)
print("MySQL Database connection successful")
except Error as err:
print("Error: '{err}'")
return connection
def create_database(connection, query):
cursor = connection.cursor()
try:
cursor.execute(query)
print("Database created successfully")
except Error as err:
print("Error: '{err}'")
def init_database(connection):
cursor = connection.cursor()
try:
query="CREATE DATABASE IF NOT EXISTS twilioTextBookBalancer;"
cursor.execute(query)
print("Database created successfully")
except Error as err:
print("Error during database initialization, could not create db: '{err}'")
def init_table(connection):
cursor = connection.cursor()
try:
query="CREATE DATABASE IF NOT EXISTS twilioTextBookBalancer;"
cursor.execute(query)
query="USE twilioTextBookBalancer;"
cursor.execute(query)
query="CREATE TABLE IF NOT EXISTS messages (id INT NOT NULL AUTO_INCREMENT, fromNum CHAR(30) NOT NULL, cost DECIMAL(18,2) NOT NULL, details VARCHAR(250), dateTime DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, PRIMARY KEY (id));"
cursor.execute(query)
print("Table created successfully")
except Error as err:
print("Error during table initialization, could not create table: '{err}'")
def init_table2(connection):
#init two tables, a user table and a message table
cursor = connection.cursor()
try:
query="CREATE DATABASE IF NOT EXISTS twilioTextBookBalancer;"
cursor.execute(query)
query="USE twilioTextBookBalancer;"
cursor.execute(query)
query="CREATE TABLE IF NOT EXISTS users(userID INT NOT NULL AUTO_INCREMENT, userNumber CHAR(30) NOT NULL UNIQUE,startDate DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, active BOOLEAN NOT NULL DEFAULT 1, PRIMARY KEY(userID));"
cursor.execute(query)
query="CREATE TABLE IF NOT EXISTS messages (id INT NOT NULL AUTO_INCREMENT, fromNum CHAR(30) NOT NULL, cost DECIMAL(18,2) NOT NULL, details VARCHAR(250), dateTime DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, PRIMARY KEY (id), FOREIGN KEY(fromNum) REFERENCES users(userNumber) ON DELETE CASCADE ON UPDATE CASCADE);"
cursor.execute(query)
print("Table created successfully")
except Error as err:
print("Error during table initialization, could not create table: '{err}'")