2017-10-20 20:59:21 +08:00
|
|
|
import os
|
2017-09-16 08:49:46 +08:00
|
|
|
import sqlite3
|
|
|
|
|
|
|
|
# Check if database exist
|
2017-10-22 08:02:25 +08:00
|
|
|
if os.path.exists(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db')) == True:
|
2017-09-16 08:49:46 +08:00
|
|
|
pass
|
|
|
|
else:
|
2017-10-22 10:51:15 +08:00
|
|
|
# Create data directory tree
|
2017-10-22 11:01:24 +08:00
|
|
|
try:
|
|
|
|
os.mkdir(os.path.join(os.path.dirname(__file__), 'data'))
|
|
|
|
except OSError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
try:
|
|
|
|
os.mkdir(os.path.join(os.path.dirname(__file__), 'data/cache'))
|
|
|
|
except OSError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
try:
|
|
|
|
os.mkdir(os.path.join(os.path.dirname(__file__), 'data/db'))
|
|
|
|
except OSError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
try:
|
|
|
|
os.mkdir(os.path.join(os.path.dirname(__file__), 'data/log'))
|
|
|
|
except OSError:
|
|
|
|
pass
|
2017-10-22 10:51:15 +08:00
|
|
|
|
2017-09-16 08:49:46 +08:00
|
|
|
# Get SQL script from file
|
2017-10-22 08:02:25 +08:00
|
|
|
fd = open(os.path.join(os.path.dirname(__file__), 'create_db.sql'), 'r')
|
2017-09-16 08:49:46 +08:00
|
|
|
script = fd.read()
|
|
|
|
|
|
|
|
# Open database connection
|
2017-10-22 08:02:25 +08:00
|
|
|
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'))
|
2017-09-16 08:49:46 +08:00
|
|
|
c = db.cursor()
|
|
|
|
|
|
|
|
# Execute script and commit change to database
|
|
|
|
c.executescript(script)
|
|
|
|
|
|
|
|
# Close database connection
|
|
|
|
db.close()
|
|
|
|
|
|
|
|
# Close SQL script file
|
|
|
|
fd.close()
|