get_connection: Error on invalid deployment_type

Without this, you get "UnboundLocalError: local variable 'conn' referenced before assignment" Signed-off-by: Zack Cerza <[email protected]>
ceph · Jan 31, 2020 · cc9877f · cc9877f
1 parent f54afc5
commit cc9877f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ceph_medic/connection.py b/ceph_medic/connection.py
@@ -39,6 +39,9 @@ def get_connection(hostname, username=None, threads=5, use_sudo=None, detect_sud
                 threads=threads,
                 detect_sudo=detect_sudo,
             )
+        else:
+            raise RuntimeError(
+                    'Invalid deployment_type: %s' % deployment_type)
         # Set a timeout value in seconds to disconnect and move on
         # if no data is sent back.
         conn.global_timeout = 300