From 43f9d78f8c00e13732b809519f0529ac293fd5ab Mon Sep 17 00:00:00 2001 From: Mitsuo Tokumori Date: Sun, 29 Oct 2023 07:41:04 -0500 Subject: Rename to ustayml (u-stayML) --- flaskr/db.py | 49 ------------------------------------------------- 1 file changed, 49 deletions(-) delete mode 100644 flaskr/db.py (limited to 'flaskr/db.py') 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')) - - -@click.command('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 -- cgit v1.2.3