async def startup(): await fa_users_db.connect() init_db() scheduler = AsyncIOScheduler(timezone=pytz.timezone('America/Los_Angeles')) scheduler.start() scheduler.add_job(habitica.cron, "cron", hour="*") scheduler.add_job(run_influencers, "cron", hour="*") scheduler.add_job(cloud_up_maybe, "cron", second="*/5") scheduler.add_job(M.Machine.prune, "cron", minute="*") scheduler.add_job(M.Job.prune, "cron", minute="*")
db_session.query(User).filter(User.name==u.name).delete() db_session.commit() flash('Selected User was successfully deleted') return redirect(url_for('show_entries')) @app.route('/login', methods=['GET', 'POST']) def login(): error = None if request.method == 'POST': if request.form['username'] != app.config['USERNAME']: error = 'Invalid username' elif request.form['password'] != app.config['PASSWORD']: error = 'Invalid password' else: session['logged_in'] = True flash('You were logged in') return redirect(url_for('show_entries')) return render_template('login.html', error=error) @app.route('/logout') def logout(): session.pop('logged_in', None) flash('You were logged out') return redirect(url_for('show_entries')) if __name__ == '__main__': init_db() app.run()
async def startup(): await fa_users_db.connect() init_db()
with D.session() as sess: # drop & re-create since we'll be futzing with models/fields for t in """ users cache_users entries cache_entries fields influencers jobs machines """.split(): sess.execute(f"drop table if exists {t} cascade") sess.commit() D.init_db() @pytest.fixture(scope='session') def db(): with D.session() as sess: yield sess @pytest.fixture(scope='session') def entries(): return fixtures.entries @pytest.fixture(scope='session') def users():
from flask import Flask from flask_restful import Api, Resource from flask_migrate import Migrate from common import database, marshmollow app = Flask(__name__) app.config["DEBUG"] = True app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///book_storm.db' api = Api(app) database.init_db(app) marshmollow.init_ma(app) from api import book_store_api, book_api, user_api if __name__ == "__main__": migrate = Migrate(app, database.db) migrate.db.create_all() app.run()
def wipe(url, and_init=False): engine(url).execute(DROP_SQL) if and_init: import common.models # sets up Base.metadata init_db() shutdown_db()