1
0

first commit

This commit is contained in:
2021-01-16 22:15:10 +08:00
commit f539b7e11f
30 changed files with 2381 additions and 0 deletions

51
src/database.py Normal file
View File

@@ -0,0 +1,51 @@
import config
import sqlite3
import json
class CalendarDatabase(object):
def __init__(self):
self.db = None
def open(self):
if (self.is_database_valid()):
raise Exception('Databade is opened')
if config.CustomConfig['database-type'] == 'sqlite':
self.db = sqlite3.connect(config.CustomConfig['database-config']['url'])
elif config.CustomConfig['database-type'] == 'mysql':
raise Exception('Not implemented database')
else:
raise Exception('Unknow database type')
def init(self, username, password):
if (self.is_database_valid()):
raise Exception('Databade is opened')
self.open()
with open('sql/sqlite.sql', 'r', encoding='utf-8') as fsql:
cursor = self.db.cursor()
cursor.executescript(fsql.read())
cursor.close()
self.db.commit()
#todo: finish init
def close(self):
self.check_database()
self.db.close()
self.db = None
def check_database(self):
if (not self.is_database_valid()):
raise Exception('Databade is None')
def is_database_valid(self):
return not (self.db == None)
# operation function
def login_step1(self, username):
self.check_database()
def login_step2(self, username, password):
self.check_database()