Add hibernate functionality #103

Add hibernate functionality Merge pull request #103 from padawarmik/main
dougiteixeira · Jul 21, 2023 · c6a6cdc · c6a6cdc
2 parents a784735 + 508890a
commit c6a6cdc
Show file tree

Hide file tree

Showing 6 changed files with 25 additions and 5 deletions.
diff --git a/custom_components/proxmoxve/__init__.py b/custom_components/proxmoxve/__init__.py
@@ -610,9 +610,14 @@ def call_api_post_status(
         elif api_category is ProxmoxType.Node:
             result = proxmox(["nodes", node, "status"]).post(command=command)
         else:
-            result = proxmox(
-                ["nodes", node, api_category, vm_id, "status", command]
-            ).post()
+            if command == ProxmoxCommand.HIBERNATE:
+                result = proxmox(
+                    ["nodes", node, api_category, vm_id, "status", ProxmoxCommand.SUSPEND]
+                ).post(todisk=1)
+            else:
+                result = proxmox(
+                    ["nodes", node, api_category, vm_id, "status", command]
+                ).post()
 
     except (ResourceException, ConnectTimeout) as err:
         raise ValueError(

diff --git a/custom_components/proxmoxve/button.py b/custom_components/proxmoxve/button.py
@@ -118,6 +118,14 @@ class ProxmoxButtonEntityDescription(ProxmoxEntityDescription, ButtonEntityDescr
         entity_registry_enabled_default=False,
         translation_key="suspend",
     ),
+    ProxmoxButtonEntityDescription(
+        key=ProxmoxCommand.HIBERNATE,
+        icon="mdi:bed",
+        name="Hibernate",
+        api_category=ProxmoxType.QEMU,
+        entity_registry_enabled_default=False,
+        translation_key="hibernate",
+    ),
     ProxmoxButtonEntityDescription(
         key=ProxmoxCommand.RESET,
         icon="mdi:restart-alert",

diff --git a/custom_components/proxmoxve/const.py b/custom_components/proxmoxve/const.py
@@ -55,6 +55,7 @@ class ProxmoxCommand(StrEnum):
     RESET = "reset"
     START_ALL = "startall"
     STOP_ALL = "stopall"
+    HIBERNATE = "hibernate"
 
 
 class ProxmoxKeyAPIParse(StrEnum):

diff --git a/custom_components/proxmoxve/manifest.json b/custom_components/proxmoxve/manifest.json
@@ -8,5 +8,5 @@
   "issue_tracker": "https://github.com/dougiteixeira/proxmoxve/issues",
   "loggers": ["proxmoxer"],
   "requirements": ["proxmoxer==2.0.1"],
-  "version": "2.0.3"
+  "version": "2.0.4"
 }
diff --git a/custom_components/proxmoxve/translations/en.json b/custom_components/proxmoxve/translations/en.json
@@ -78,6 +78,9 @@
             },
             "suspend": {
                 "name": "Suspend"
+            },
+            "hibernate": {
+                "name": "Hibernate"
             }
         },
         "sensor": {

diff --git a/custom_components/proxmoxve/translations/pt-BR.json b/custom_components/proxmoxve/translations/pt-BR.json
@@ -78,6 +78,9 @@
           },
           "reset": {
             "name": "Resetar"
+          },
+          "hibernate": {
+             "name": "Hibernar"
           }
       },
       "sensor": {
@@ -224,4 +227,4 @@
           }
       }
   }
-}
+}