You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.6 KiB
94 lines
2.6 KiB
import click
|
|
from flask import Flask
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_wtf.csrf import CSRFProtect
|
|
from flask_session import Session
|
|
from flask_bcrypt import Bcrypt
|
|
from flask_login import LoginManager
|
|
from redis import Redis
|
|
from datetime import datetime
|
|
from wowstash import config
|
|
|
|
|
|
db = SQLAlchemy()
|
|
bcrypt = Bcrypt()
|
|
|
|
def _setup_db(app: Flask):
|
|
uri = 'postgresql+psycopg2://{user}:{pw}@{host}:{port}/{db}'.format(
|
|
user=config.DB_USER,
|
|
pw=config.DB_PASS,
|
|
host=config.DB_HOST,
|
|
port=config.DB_PORT,
|
|
db=config.DB_NAME
|
|
)
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = uri
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
db = SQLAlchemy(app)
|
|
import wowstash.models
|
|
db.create_all()
|
|
|
|
def create_app():
|
|
app = Flask(__name__)
|
|
app.config.from_envvar('FLASK_SECRETS')
|
|
|
|
# Setup backends
|
|
_setup_db(app)
|
|
bcrypt = Bcrypt(app)
|
|
login_manager = LoginManager(app)
|
|
|
|
with app.app_context():
|
|
|
|
# Login manager
|
|
login_manager.login_view = 'auth.login'
|
|
login_manager.logout_view = 'auth.logout'
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
from wowstash.models import User
|
|
user = User.query.get(user_id)
|
|
return user
|
|
|
|
# Template filters
|
|
@app.template_filter('datestamp')
|
|
def datestamp(s):
|
|
d = datetime.fromtimestamp(s)
|
|
return d.strftime('%Y-%m-%d %H:%M:%S')
|
|
|
|
@app.template_filter('from_atomic')
|
|
def from_atomic(a):
|
|
from wowstash.library.jsonrpc import from_atomic
|
|
atomic = from_atomic(a)
|
|
if atomic == 0:
|
|
return 0
|
|
else:
|
|
return float(atomic)
|
|
|
|
# CLI
|
|
@app.cli.command('clean_containers')
|
|
def clean_containers():
|
|
from wowstash.library.docker import docker
|
|
docker.cleanup()
|
|
|
|
@app.cli.command('reset_wallet')
|
|
@click.argument('user_id')
|
|
def reset_wallet(user_id):
|
|
from wowstash.models import User
|
|
user = User.query.get(user_id)
|
|
user.clear_wallet_data()
|
|
print(f'Wallet data cleared for user {user.id}')
|
|
|
|
@app.cli.command('init')
|
|
def init():
|
|
import wowstash.models
|
|
db.create_all()
|
|
|
|
# Routes/blueprints
|
|
from wowstash.blueprints.auth import auth_bp
|
|
from wowstash.blueprints.wallet import wallet_bp
|
|
from wowstash.blueprints.meta import meta_bp
|
|
app.register_blueprint(meta_bp)
|
|
app.register_blueprint(auth_bp)
|
|
app.register_blueprint(wallet_bp)
|
|
|
|
return app
|