summaryrefslogtreecommitdiffstats
path: root/stator
diff options
context:
space:
mode:
Diffstat (limited to 'stator')
-rw-r--r--stator/runner.py6
-rw-r--r--stator/views.py12
2 files changed, 12 insertions, 6 deletions
diff --git a/stator/runner.py b/stator/runner.py
index 187aa47..bb1b009 100644
--- a/stator/runner.py
+++ b/stator/runner.py
@@ -19,9 +19,9 @@ class StatorRunner:
def __init__(
self,
models: List[Type[StatorModel]],
- concurrency: int = 30,
- concurrency_per_model: int = 5,
- run_period: int = 30,
+ concurrency: int = 50,
+ concurrency_per_model: int = 10,
+ run_period: int = 60,
wait_period: int = 30,
):
self.models = models
diff --git a/stator/views.py b/stator/views.py
index ef09b8e..9d2e154 100644
--- a/stator/views.py
+++ b/stator/views.py
@@ -1,8 +1,9 @@
-from django.http import HttpResponse
+from django.conf import settings
+from django.http import HttpResponse, HttpResponseForbidden
from django.views import View
+from stator.models import StatorModel
from stator.runner import StatorRunner
-from users.models import Follow
class RequestRunner(View):
@@ -12,6 +13,11 @@ class RequestRunner(View):
"""
async def get(self, request):
- runner = StatorRunner([Follow])
+ # Check the token, if supplied
+ if settings.STATOR_TOKEN:
+ if request.GET.get("token") != settings.STATOR_TOKEN:
+ return HttpResponseForbidden()
+ # Run on all models
+ runner = StatorRunner(StatorModel.subclasses)
handled = await runner.run()
return HttpResponse(f"Handled {handled}")