Add sahkopiikki user create command

members/management/commands/createsahkopiikkiuser.py

@@ -0,0 +1,26 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth.models import User
+from rest_framework.authtoken.models import Token
+
+
+class Command(BaseCommand):
+    user_name = "sahkopiikki"
+    password = User.objects.make_random_password()
+
+    def handle(self, *args, **options):
+        if User.objects.filter(username=self.user_name).exists():
+            self.stdout.write("Sahkopiikki user already exists. Skipping.")
+            user = User.objects.get(username=self.user_name)
+            token = Token.objects.get(user=user)
+            self.stdout.write("Token: {}".format(token))
+            return
+
+        u = User(username=self.user_name)
+        u.set_password(self.password)
+        u.save()
+
+        token = Token.objects.create(user=u)
+
+        self.stdout.write("Created sahkopiikki user '{}' with password '{}' and token '{}'.".format(
+            self.user_name, self.password, token
+        ))

members/views/utils.py

@@ -13,9 +13,10 @@ import csv
 
 # REST framework
 from members.serializers import MemberSerializer
+from members.throttles import BurstRateThrottle, SustainedRateThrottle
+
 from rest_framework import generics
 from rest_framework import permissions
-from rest_framework.throttling import UserRateThrottle, AnonRateThrottle
 
 from members.models import Member, Request, Payment
 from members.forms import MemberForm, PaymentForm, ApplicationForm, CSVValidationError
@@ -29,7 +30,7 @@ class MemberDetail(generics.RetrieveAPIView):
     queryset = Member.objects.all()
     serializer_class = MemberSerializer
     permission_classes = (permissions.IsAdminUser, )
-    throttle_classes = (UserRateThrottle, AnonRateThrottle, )
+    throttle_classes = (BurstRateThrottle, SustainedRateThrottle, )
 
 
 def error_view(request, message):