-
Notifications
You must be signed in to change notification settings - Fork 3
/
set_up_db.py
44 lines (39 loc) · 1.31 KB
/
set_up_db.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
#!/usr/bin/env python
import json
import MySQLdb
import pymysql.cursors
import sys
if __name__ == '__main__':
# load json config
try:
with open('config.json') as json_config:
config = json.load(json_config)
except Exception:
sys.exit('Failed to open config.json')
else:
db_name = config['db_name']
table_name = config['table_name']
# create database
try:
cursor = MySQLdb.connect(host='localhost', user='root').cursor()
cursor.execute('CREATE DATABASE `%s`;' % db_name)
except Exception as e:
sys.exit('Error creating database')
finally:
cursor.close()
# create table
try:
# connect to db
connection = pymysql.connect(host='localhost',
user='root',
db=db_name,
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor)
with connection.cursor() as cursor:
sql = "CREATE TABLE `%s` (celeb VARCHAR(30) NOT NULL PRIMARY KEY, author VARCHAR(30), authored DATE)" % table_name
cursor.execute(sql)
connection.commit()
except Exception as e:
sys.exit('Error creating table')
finally:
connection.close()