dani/django-ram
1
0
Fork 0
mirror of https://github.com/daniviga/django-ram.git synced 2025-08-04 13:17:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add an option to completely disable driver

Browse Source
This commit is contained in:
Daniele Viganò
2022-07-11 20:56:31 +02:00
parent 49d28b176e
commit 47b4b2915b
4 changed files with 47 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
ram/driver/views.py
View File

@@ -34,9 +34,22 @@ def addresschecker(f):
return addresslookup
class IsEnabled(BasePermission):
def has_permission(self, request, view):
config = DriverConfiguration.get_solo()
# if driver is disabled, block all connections
if not config.enabled:
raise Http404
return True
class Firewall(BasePermission):
def has_permission(self, request, view):
config = DriverConfiguration.get_solo()
# if network is not configured, accept only read ops
if not config.network:
return request.method in SAFE_METHODS
@@ -61,7 +74,7 @@ class Test(APIView):
"""
parser_classes = [PlainTextParser]
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def get(self, request):
response = Connector().passthrough("<s>")
@@ -76,7 +89,7 @@ class SendCommand(APIView):
"""
parser_classes = [PlainTextParser]
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def put(self, request):
data = request.data
@@ -101,7 +114,7 @@ class Function(APIView):
Send "Function" commands to a valid DCC address
"""
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def put(self, request, address):
serializer = FunctionSerializer(data=request.data)
@@ -118,7 +131,7 @@ class Cab(APIView):
Send "Cab" commands to a valid DCC address
"""
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def put(self, request, address):
serializer = CabSerializer(data=request.data)
@@ -134,7 +147,7 @@ class Infra(APIView):
Send "Infra" commands to a valid DCC address
"""
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def put(self, request):
serializer = InfraSerializer(data=request.data)
@@ -150,7 +163,7 @@ class Emergency(APIView):
Send an "Emergency" stop, no matter the HTTP method used
"""
permission_classes = [IsAuthenticated | Firewall]
permission_classes = [IsEnabled & IsAuthenticated | Firewall]
def put(self, request):
Connector().emergency()