From 62698bcddef3e13cdd26b63b03f8dcf6d94422a7 Mon Sep 17 00:00:00 2001
From: Yuta Hayashibe <yuta@hayashibe.jp>
Date: Tue, 27 Sep 2022 12:27:22 +0900
Subject: [PATCH] Refactoring

---
 whispering/cli.py | 40 +++++++++++++++++++++-------------------
 1 file changed, 21 insertions(+), 19 deletions(-)

diff --git a/whispering/cli.py b/whispering/cli.py
index a387608..863c5c7 100644
--- a/whispering/cli.py
+++ b/whispering/cli.py
@@ -120,8 +120,19 @@ def get_opts() -> argparse.Namespace:
         "--show-devices",
         action="store_true",
     )
+    opts = parser.parse_args()
 
-    return parser.parse_args()
+    if opts.beam_size <= 0:
+        opts.beam_size = None
+    if len(opts.temperature) == 0:
+        opts.temperature = [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]
+    opts.temperature = sorted(set(opts.temperature))
+
+    try:
+        opts.mic = int(opts.mic)
+    except Exception:
+        pass
+    return opts
 
 
 def get_wshiper(*, opts):
@@ -139,32 +150,23 @@ def get_wshiper(*, opts):
     return wsp
 
 
+def show_devices():
+    devices = sd.query_devices()
+    for i, device in enumerate(devices):
+        if device["max_input_channels"] > 0:
+            print(f"{i}: {device['name']}")
+
+
 def main() -> None:
     opts = get_opts()
 
-    if opts.show_devices:
-        devices = sd.query_devices()
-        for i, device in enumerate(devices):
-            if device["max_input_channels"] > 0:
-                print(f"{i}: {device['name']}")
-
-        return
-
     basicConfig(
         level=DEBUG if opts.debug else INFO,
         format="[%(asctime)s] %(module)s.%(funcName)s:%(lineno)d %(levelname)s -> %(message)s",
     )
 
-    if opts.beam_size <= 0:
-        opts.beam_size = None
-    if len(opts.temperature) == 0:
-        opts.temperature = [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]
-    opts.temperature = sorted(set(opts.temperature))
-
-    try:
-        opts.mic = int(opts.mic)
-    except Exception:
-        pass
+    if opts.show_devices:
+        return show_devices()
 
     if opts.host is not None and opts.port is not None:
         if opts.mode == "client":