mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Add scheduling_enabled parameter for scripts

Browse Source
This commit is contained in:
jeremystretch
2023-04-17 11:53:05 -04:00
committed by Jeremy Stretch
parent 014a5d10d1
commit 197c6a1cbf
7 changed files with 76 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
netbox/extras/api/serializers.py
View File

@@ -478,6 +478,16 @@ class ScriptInputSerializer(serializers.Serializer):
schedule_at = serializers.DateTimeField(required=False, allow_null=True)
interval = serializers.IntegerField(required=False, allow_null=True)
def validate_schedule_at(self, value):
if value and not self.context['script'].scheduling_enabled:
raise serializers.ValidationError("Scheduling is not enabled for this script.")
return value
def validate_interval(self, value):
if value and not self.context['script'].scheduling_enabled:
raise serializers.ValidationError("Scheduling is not enabled for this script.")
return value
class ScriptLogMessageSerializer(serializers.Serializer):
status = serializers.SerializerMethodField(read_only=True)

5
netbox/extras/api/views.py
View File

@@ -329,7 +329,10 @@ class ScriptViewSet(ViewSet):
raise PermissionDenied("This user does not have permission to run scripts.")
module, script = self._get_script(pk)
input_serializer = serializers.ScriptInputSerializer(data=request.data)
input_serializer = serializers.ScriptInputSerializer(
data=request.data,
context={'script': script}
)
# Check that at least one RQ worker is running
if not Worker.count(get_connection('default')):