Add LimitGroups and MachineLimit to Houdini Deadline settings

ynput · Oct 2, 2024 · 66dc0e3 · 66dc0e3
1 parent 4842bc5
commit 66dc0e3
Showing 1 changed file with 53 additions and 2 deletions.
diff --git a/server/settings/publish_plugins.py b/server/settings/publish_plugins.py
@@ -219,10 +219,40 @@ class HoudiniSubmitDeadlineModel(BaseSettingsModel):
     priority: int = SettingsField(title="Priority")
     chunk_size: int = SettingsField(title="Chunk Size")
     group: str = SettingsField(title="Group")
+    limits: str = SettingsField(
+        title="Limit Groups",
+        description=(
+            "Enter a comma separated list of limits.\n"
+            "Specifies the limit groups that this job is a member of (default = blank)."
+        )
+    )
+    machine_limit: int = SettingsField(
+        title="Machine Limit",
+        description=(
+            "Specifies the maximum number of machines this job can be"
+            " rendered on at the same time (default = 0, which means"
+            " unlimited)."
+        )
+    )
 
     export_priority: int = SettingsField(title="Export Priority")
     export_chunk_size: int = SettingsField(title="Export Chunk Size")
     export_group: str = SettingsField(title="Export Group")
+    export_limits: str = SettingsField(
+        title="Export Limit Groups",
+        description=(
+            "Enter a comma separated list of limits.\n"
+            "Specifies the limit groups that this job is a member of (default = blank)."
+        )
+    )
+    export_machine_limit: int = SettingsField(
+        title="Export Machine Limit",
+        description=(
+            "Specifies the maximum number of machines this job can be"
+            " rendered on at the same time (default = 0, which means"
+            " unlimited)."
+        )
+    )
 
 
 class HoudiniCacheSubmitDeadlineModel(BaseSettingsModel):
@@ -234,6 +264,21 @@ class HoudiniCacheSubmitDeadlineModel(BaseSettingsModel):
     priority: int = SettingsField(title="Priority")
     chunk_size: int = SettingsField(title="Chunk Size")
     group: str = SettingsField(title="Group")
+    limits: str = SettingsField(
+        title="Limit Groups",
+        description=(
+            "Enter a comma separated list of limits.\n"
+            "Specifies the limit groups that this job is a member of (default = blank)."
+        )
+    )
+    machine_limit: int = SettingsField(
+        title="Machine Limit",
+        description=(
+            "Specifies the maximum number of machines this job can be"
+            " rendered on at the same time (default = 0, which means"
+            " unlimited)."
+        )
+    )
 
 
 class AfterEffectsSubmitDeadlineModel(BaseSettingsModel):
@@ -452,7 +497,9 @@ class PublishPluginsModel(BaseSettingsModel):
         "active": True,
         "priority": 50,
         "chunk_size": 999999,
-        "group": ""
+        "group": "",
+        "limits": "",
+        "machine_limit": 0
     },
     "HoudiniSubmitDeadline": {
         "enabled": True,
@@ -461,9 +508,13 @@ class PublishPluginsModel(BaseSettingsModel):
         "priority": 50,
         "chunk_size": 1,
         "group": "",
+        "limits": "",
+        "machine_limit": 0,
         "export_priority": 50,
         "export_chunk_size": 10,
-        "export_group": ""
+        "export_group": "",
+        "export_limits": "",
+        "export_machine_limit": 0
     },
     "MaxSubmitDeadline": {
         "enabled": True,