Now the database is made if necessary, not manually. Should be done a long time ago.
This commit is contained in:
parent
ff2f4e3407
commit
0d84d59ca2
18
egon.py
18
egon.py
|
@ -18,7 +18,7 @@ from pysqlite2 import dbapi2 as sqlite
|
|||
from qrc_resources import *
|
||||
|
||||
|
||||
__version__ = "1.0.0"
|
||||
__version__ = "1.0.1"
|
||||
|
||||
|
||||
main = None
|
||||
|
@ -1520,12 +1520,15 @@ class BookModel():
|
|||
## Connect to the database and return the cursor and connection
|
||||
def initDB():
|
||||
conn = sqlite.connect('egon.db')
|
||||
curs = conn.cursor()
|
||||
return curs, conn
|
||||
cursor = conn.cursor()
|
||||
|
||||
## Initialize the tables
|
||||
def initNewDB():
|
||||
cursor, conn = initDB()
|
||||
cursor.execute('''
|
||||
SELECT name FROM sqlite_master
|
||||
WHERE type='table'
|
||||
ORDER BY name
|
||||
''')
|
||||
|
||||
if cursor.fetchall() == []:
|
||||
initSemesterDB(cursor)
|
||||
initAssignmentDB(cursor)
|
||||
initReadingDB(cursor)
|
||||
|
@ -1533,6 +1536,9 @@ def initNewDB():
|
|||
initBookDB(cursor)
|
||||
initCourseDB(cursor)
|
||||
initCourseUsesBook(cursor)
|
||||
|
||||
return cursor, conn
|
||||
|
||||
exitDB(conn)
|
||||
|
||||
# Create the database
|
||||
|
|
Reference in New Issue