rDrama
/
rDrama
9
10
Fork 8
Code Issues Pull Requests Packages Releases Wiki Activity

ratelimits: turn off autocheck so we can init 

required app context globals
pull/1/head
justcool393 2022-11-29 19:29:06 -06:00
parent f2411415dd
commit db31fc17de
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
files/__main__.py
View File

@ -56,7 +56,8 @@ limiter = Limiter(
key_func=get_CF, key_func=get_CF,
default_limits=[DEFAULT_RATELIMIT], default_limits=[DEFAULT_RATELIMIT],
application_limits=["10/second;200/minute;5000/hour;10000/day"], application_limits=["10/second;200/minute;5000/hour;10000/day"],
storage_uri=environ.get("REDIS_URL", "redis://localhost") storage_uri=environ.get("REDIS_URL", "redis://localhost"),
auto_check=False,
) )
engine = create_engine(app.config['SQLALCHEMY_DATABASE_URL']) engine = create_engine(app.config['SQLALCHEMY_DATABASE_URL'])

8
files/routes/allroutes.py
View File

@ -4,7 +4,7 @@ from files.helpers.const import *
from files.helpers.settings import get_setting from files.helpers.settings import get_setting
from files.helpers.cloudflare import CLOUDFLARE_AVAILABLE from files.helpers.cloudflare import CLOUDFLARE_AVAILABLE
from files.routes.wrappers import * from files.routes.wrappers import *
from files.__main__ import app from files.__main__ import app, limiter
def session_init(): def session_init():
if not session.get("session_id"): if not session.get("session_id"):
@ -21,8 +21,7 @@ def before_request():
if not g.agent and request.path != '/kofi': if not g.agent and request.path != '/kofi':
return 'Please use a "User-Agent" header!', 403 return 'Please use a "User-Agent" header!', 403
ua = g.agent or '' ua = g.agent.lower()
ua = ua.lower()
if request.host != SITE: if request.host != SITE:
return {"error": "Unauthorized host provided"}, 403 return {"error": "Unauthorized host provided"}, 403
@ -31,7 +30,6 @@ def before_request():
if not get_setting('Bots') and request.headers.get("Authorization"): abort(403) if not get_setting('Bots') and request.headers.get("Authorization"): abort(403)
g.db = db_session()
g.webview = '; wv) ' in ua g.webview = '; wv) ' in ua
if ' firefox/' in ua: if ' firefox/' in ua:
@ -49,6 +47,8 @@ def before_request():
if not request.full_path: request.full_path = '/' if not request.full_path: request.full_path = '/'
session_init() session_init()
limiter.check()
g.db = db_session()
@app.after_request @app.after_request