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()