format(): support Format(Enum) as an argument

1 files changed, 4 insertions, 2 deletions

diff --git a/inverterd/inverterd.py b/inverterd/inverterd.py
index 5b9541e..b2814d5 100644
--- a/inverterd/inverterd.py
+++ b/inverterd/inverterd.py
@@ -45,8 +45,10 @@ class Client:
         self._write(f'v {v}')
         return self._read()
 
-    def format(self, format):
-        self._write(f'format {format}')
+    def format(self, fmt):
+        if isinstance(fmt, Format):
+            fmt = fmt.value
+        self._write(f'format {fmt}')
         return self._read()
 
     def exec(self, command, arguments=()):