mirror of
https://github.com/partitioncloud/partitioncloud-server.git
synced 2025-01-23 17:26:26 +01:00
29 lines
620 B
Python
29 lines
620 B
Python
|
import sqlite3
|
||
|
|
|
||
|
|
from flask import current_app
|
||
|
|
from flask import g
|
||
|
|
|
||
|
|
|
||
|
|
def get_db():
|
||
|
|
"""Connect to the application's configured database. The connection
|
||
|
|
is unique for each request and will be reused if this is called
|
||
|
|
again.
|
||
|
|
"""
|
||
|
|
if "db" not in g:
|
||
|
|
g.db = sqlite3.connect(
|
||
|
|
current_app.config["DATABASE"], detect_types=sqlite3.PARSE_DECLTYPES
|
||
|
|
)
|
||
|
|
g.db.row_factory = sqlite3.Row
|
||
|
|
|
||
|
|
return g.db
|
||
|
|
|
||
|
|
|
||
|
|
def close_db(e=None):
|
||
|
|
"""If this request connected to the database, close the
|
||
|
|
connection.
|
||
|
|
"""
|
||
|
|
db = g.pop("db", None)
|
||
|
|
|
||
|
|
if db is not None:
|
||
|
|
db.close()
|