From 3685cfc86c8400392bac8e206a39f7793db85894 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Arkadiusz=20Dzi=C4=99giel?= <arkadiusz.dziegiel@glorpen.pl>
Date: Mon, 5 Dec 2022 15:24:10 +0100
Subject: [PATCH] translate reboot names

---
 hetzner/reset.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/hetzner/reset.py b/hetzner/reset.py
index 4afc310..f8b37f0 100644
--- a/hetzner/reset.py
+++ b/hetzner/reset.py
@@ -5,6 +5,13 @@
 
 
 class Reset(object):
+    _modes = {
+        'manual': 'man',
+        'hard': 'hw',
+        'soft': 'sw',
+        'power': 'power',
+    }
+
     def __init__(self, server):
         self.server = server
         self.conn = server.conn
@@ -42,7 +49,8 @@ def reset_types(self):
         """
         if self._reset_types is None:
             self._update_status()
-        return self._reset_types
+
+        return tuple(key for key, value in self._modes.items() if value in self._reset_types)
 
     def check_ssh(self, port=22, timeout=5):
         """
@@ -117,13 +125,7 @@ def reboot(self, mode='soft'):
         a poor devil from the data center to go to your server and press the
         power button.
         """
-        modes = {
-            'manual': 'man',
-            'hard': 'hw',
-            'soft': 'sw',
-            'power': 'power',
-        }
-
-        modekey = modes.get(mode, modes['soft'])
+
+        modekey = self._modes.get(mode, self._modes['soft'])
         return self.conn.post('/reset/{0}'.format(self.server.number),
                               {'type': modekey})