def initialise_db(db_name):
"""Creates a new database if an existing one cannot be found
:param db_name: Filename of the database
:type db_name: str
"""
if db_name not in glob('*'):
logger.info('Creating new database \'%s\'', db_name)
db.create_all()
def database_esistente(self):
"""
Controllo che il database esista altrimenti lo creo
"""
path = os.getcwd()
if DB_NAME in os.listdir(path):
self.write_to_log("WARNING: Database esistente!")
else:
self.write_to_log("INFO: Creazione database...")
db.create_all()
from flask import Flask, url_for, redirect, render_template, request
from flask_login import login_required, current_user, login_user, logout_user
from dashboards import dashboard_list
from login_manager import login_manager, authenticate_user
from db_models import User, Wine, db
from dummy_data import create_dummy_user, load_wine_data
app = Flask(__name__)
app.secret_key = 'SuperDuperSecretSecret'
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///./app.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
with app.app_context():
db.init_app(app)
db.create_all()
if User.query.first() is None:
create_dummy_user()
if Wine.query.first() is None:
load_wine_data()
for app_type in dashboard_list:
try:
app_type.create_dash_app(app)
except:
traceback.print_exc()
for view_func in app.view_functions:
if view_func.startswith('/dashboards'):
app.view_functions[view_func] = login_required(
app.view_functions[view_func])
def setup():
db.create_all()
db_inserts()
user = current_user
if request.method == 'POST' and user.is_authenticated():
instance = user.Gamma.order_by('-id').first()
instance.comments = request.form.get("comments", None)
db.session.commit()
return redirect(url_for('index'))
@app.route('/delete/<id>', methods=['GET', 'POST'])
@login_required
def delete_post(id):
id = int(id)
user = current_user
if user.is_authenticated():
if id == -1:
instances = user.Gamma.delete()
else:
try:
instance = user.Gamma.filter_by(id=id).first()
db.session.delete(instance)
except:
pass
db.session.commit()
return redirect(url_for('old'))
if __name__ == '__main__':
if not os.path.isfile(os.path.join(os.path.dirname(__file__), 'sqlite.db')):
db.create_all()
app.run(debug=True)