xevion/v1.xevion.dev
1
0
Fork 0
mirror of https://github.com/Xevion/v1.xevion.dev.git synced 2025-12-10 22:09:02 -06:00
Code Issues Packages Projects Releases Wiki Activity

add key_func to disable warning, simplify custom download limiter

Browse Source
This commit is contained in:
Xevion
2019-12-25 02:42:41 -06:00
parent 248a2efb6c
commit b4d52fd7df
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/__init__.py
View File

@@ -6,6 +6,7 @@ from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate from flask_migrate import Migrate
from flask_login import LoginManager from flask_login import LoginManager
from flask_limiter import Limiter from flask_limiter import Limiter
from flask_limiter.util import get_remote_address
# App & App config setup # App & App config setup
app = Flask(__name__) app = Flask(__name__)
@@ -15,7 +16,7 @@ login = LoginManager(app)
login.login_view = 'login' login.login_view = 'login'
db = SQLAlchemy(app) db = SQLAlchemy(app)
migrate = Migrate(app, db) migrate = Migrate(app, db)
limiter = Limiter(app, default_limits=["10 per minute"]) limiter = Limiter(app, key_func=get_remote_address, default_limits=["10 per minute"])
from app import models from app import models
from app import routes, simple_routes, hidden, dashboard from app import routes, simple_routes, hidden, dashboard

5
app/sound.py
View File

@@ -34,9 +34,8 @@ def get_youtube(mediaid):
# Under the request context, it grabs the same args needed to decide whether the stream has been downloaded previously # Under the request context, it grabs the same args needed to decide whether the stream has been downloaded previously
# It applies rate limiting differently based on service, and whether the stream has been accessed previously # It applies rate limiting differently based on service, and whether the stream has been accessed previously
def downloadLimiter(): def downloadLimiter():
service, mediaid = request.view_args['service'], request.view_args['mediaid'] if request.view_args['service'] == 'youtube':
if service == 'youtube': if YouTubeAudio.query.get(request.view_args['mediaid']) is not None:
if YouTubeAudio.query.get(mediaid) is not None:
return '5/minute' return '5/minute'
else: else:
return '1/30seconds' return '1/30seconds'