21 lines
534 B
Python
21 lines
534 B
Python
from flask import g
|
|
from database.sqlite import SQLite
|
|
import json
|
|
|
|
def get_config():
|
|
config = getattr(g, "_config", None)
|
|
|
|
if config is None:
|
|
with open("../config.json") as config_file:
|
|
config = g._config = json.loads(config_file.read())
|
|
|
|
return config
|
|
|
|
def get_db():
|
|
db = getattr(g, "_database", None)
|
|
|
|
if db is None:
|
|
if str(get_config()["database"]["type"]).casefold() == "SQLite".casefold():
|
|
db = g._database = SQLite(get_config()["database"]["name"])
|
|
|
|
return db |