style corrections
This commit is contained in:
parent
7da002eb00
commit
f1a57c16e4
|
@ -5,9 +5,10 @@ from apscheduler.schedulers.background import BackgroundScheduler
|
|||
from tlapbot.db import get_db
|
||||
from tlapbot.owncast_requests import is_stream_live, give_points_to_chat
|
||||
|
||||
|
||||
def create_app(test_config=None):
|
||||
app = Flask(__name__, instance_relative_config=True)
|
||||
|
||||
|
||||
# ensure the instance folder exists
|
||||
try:
|
||||
os.makedirs(app.instance_path)
|
||||
|
@ -29,19 +30,18 @@ def create_app(test_config=None):
|
|||
gunicorn_logger = logging.getLogger('gunicorn.error')
|
||||
app.logger.handlers = gunicorn_logger.handlers
|
||||
app.logger.setLevel(gunicorn_logger.level)
|
||||
|
||||
|
||||
# Check for wrong config that would break Tlapbot
|
||||
if len(app.config['PREFIX']) != 1:
|
||||
raise RuntimeError("Prefix is >1 character. "
|
||||
"Change your config to set 1-character prefix.")
|
||||
|
||||
|
||||
# Check for spaces in redeems (they won't work)
|
||||
for redeem in app.config['REDEEMS']:
|
||||
if ' ' in redeem:
|
||||
app.logger.warning(f"Redeem '{redeem}' has spaces in its name.")
|
||||
app.logger.warning("Redeems with spaces are impossible to redeem.")
|
||||
|
||||
|
||||
# prepare webhooks and redeem dashboard blueprints
|
||||
from . import owncast_webhooks
|
||||
from . import tlapbot_dashboard
|
||||
|
@ -57,7 +57,7 @@ def create_app(test_config=None):
|
|||
app.cli.add_command(db.refresh_milestones_command)
|
||||
app.cli.add_command(db.reset_milestone_command)
|
||||
app.cli.add_command(db.hard_reset_milestone_command)
|
||||
|
||||
|
||||
# scheduler job for giving points to users
|
||||
def proxy_job():
|
||||
with app.app_context():
|
||||
|
|
|
@ -6,6 +6,7 @@ from flask.cli import with_appcontext
|
|||
|
||||
from tlapbot.redeems import milestone_complete
|
||||
|
||||
|
||||
def get_db():
|
||||
if 'db' not in g:
|
||||
g.db = sqlite3.connect(
|
||||
|
@ -127,7 +128,7 @@ def refresh_milestones():
|
|||
|
||||
|
||||
def reset_milestone(milestone):
|
||||
if not milestone in current_app.config['REDEEMS']:
|
||||
if milestone not in current_app.config['REDEEMS']:
|
||||
print(f"Failed resetting milestone, {milestone} not in redeems file.")
|
||||
return False
|
||||
try:
|
||||
|
@ -147,7 +148,6 @@ def reset_milestone(milestone):
|
|||
return False
|
||||
|
||||
|
||||
|
||||
@click.command('init-db')
|
||||
@with_appcontext
|
||||
def init_db_command():
|
||||
|
@ -184,7 +184,7 @@ def refresh_and_clear_command():
|
|||
@click.command('refresh-milestones')
|
||||
@with_appcontext
|
||||
def refresh_milestones_command():
|
||||
"""Initialize all milestones from the redeems file,
|
||||
"""Initialize all milestones from the redeems file,
|
||||
delete milestones not in redeem file."""
|
||||
if refresh_milestones():
|
||||
click.echo('Refreshed milestones.')
|
||||
|
|
|
@ -2,6 +2,7 @@ from flask import current_app
|
|||
from sqlite3 import Error
|
||||
from re import sub
|
||||
|
||||
|
||||
# # # db stuff # # #
|
||||
def read_users_points(db, user_id):
|
||||
"""Errors out if user doesn't exist."""
|
||||
|
|
|
@ -45,4 +45,3 @@ def send_chat(message):
|
|||
current_app.logger.error(f"Check owncast instance url and access key.")
|
||||
return
|
||||
return r.json()
|
||||
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
from flask import current_app
|
||||
from tlapbot.db import get_db
|
||||
from tlapbot.owncast_requests import send_chat
|
||||
from tlapbot.redeems import (add_to_redeem_queue, add_to_counter, add_to_milestone,
|
||||
from tlapbot.redeems import (add_to_redeem_queue, add_to_counter, add_to_milestone,
|
||||
check_apply_milestone_completion, milestone_complete, is_redeem_active)
|
||||
from tlapbot.owncast_helpers import use_points, read_users_points
|
||||
|
||||
|
@ -44,7 +44,7 @@ def handle_redeem(message, user_id):
|
|||
else:
|
||||
send_chat(f"Redeeming milestone {redeem} failed.")
|
||||
return
|
||||
|
||||
|
||||
# handle redeems with price argument
|
||||
price = current_app.config['REDEEMS'][redeem]["price"]
|
||||
if not points or points < price:
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
from flask import render_template, Blueprint, request, current_app
|
||||
from tlapbot.db import get_db
|
||||
from tlapbot.redeems import all_active_counters, all_active_milestones, all_active_redeems, pretty_redeem_queue
|
||||
from tlapbot.owncast_helpers import read_all_users_with_username
|
||||
from tlapbot.owncast_helpers import read_all_users_with_username
|
||||
from datetime import timezone
|
||||
|
||||
bp = Blueprint('redeem_dashboard', __name__)
|
||||
|
|
Loading…
Reference in New Issue