diff options
| author | Mitsuo Tokumori <[email protected]> | 2023-10-29 07:41:04 -0500 |
|---|---|---|
| committer | Mitsuo Tokumori <[email protected]> | 2023-10-29 07:41:04 -0500 |
| commit | 43f9d78f8c00e13732b809519f0529ac293fd5ab (patch) | |
| tree | d9779a6d4b511dd27cfe80b3f8d0ad2e5ef5d7da /flaskr/db.py | |
| parent | de8b3d760a37fc54645a2588b062f85963146070 (diff) | |
| download | ustayml-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.py | 49 |
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')) - - [email protected]('init-db') -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 |
