summaryrefslogtreecommitdiffstats
path: root/flaskr/db.py
diff options
context:
space:
mode:
authorMitsuo Tokumori <[email protected]>2023-10-29 07:41:04 -0500
committerMitsuo Tokumori <[email protected]>2023-10-29 07:41:04 -0500
commit43f9d78f8c00e13732b809519f0529ac293fd5ab (patch)
treed9779a6d4b511dd27cfe80b3f8d0ad2e5ef5d7da /flaskr/db.py
parentde8b3d760a37fc54645a2588b062f85963146070 (diff)
downloadustayml-43f9d78f8c00e13732b809519f0529ac293fd5ab.tar.gz
ustayml-43f9d78f8c00e13732b809519f0529ac293fd5ab.tar.bz2
ustayml-43f9d78f8c00e13732b809519f0529ac293fd5ab.zip
Rename to ustayml (u-stayML)
Diffstat (limited to 'flaskr/db.py')
-rw-r--r--flaskr/db.py49
1 files changed, 0 insertions, 49 deletions
diff --git a/flaskr/db.py b/flaskr/db.py
deleted file mode 100644
index eb4e8b4..0000000
--- a/flaskr/db.py
+++ /dev/null
@@ -1,49 +0,0 @@
-import sqlite3
-
-import click
-from flask import current_app, g
-
-
-def get_db():
- """Returns database connection
-
- sqlite3: https://docs.python.org/3/library/sqlite3.html
- """
- if 'db' not in g:
- g.db = sqlite3.connect(
- current_app.config['DATABASE'],
- detect_types=sqlite3.PARSE_DECLTYPES
- )
- # Return rows that behave like dicts
- g.db.row_factory = sqlite3.Row
-
- return g.db
-
-
-def close_db(e=None):
- db = g.pop('db', None)
-
- if db:
- db.close()
-
-# CLI:
-# https://flask.palletsprojects.com/en/3.0.x/cli/
-
-def init_db():
- db = get_db()
-
- with current_app.open_resource('schema.sql') as f:
- db.executescript(f.read().decode('utf8'))
-
-
-def init_db_command():
- """Clear the existing data and create new tables."""
- init_db()
- click.echo('Initialized the database.')
-
-# Register function with application
-
-def init_app(app):
- app.teardown_appcontext(close_db) # callback after returning response
- app.cli.add_command(init_db_command) \ No newline at end of file