Move monkey-patching to dedicated file

main.py

@@ -1,38 +1,10 @@
 import logging
 import paramiko
-import sshim
-import pylxd
+import sshim_patch as sshim
+import lxd_interface
 import os
 import re
 
-# monkey patching
-
-
-def check_auth_none(self, username):
-    return paramiko.AUTH_PARTIALLY_SUCCESSFUL
-
-
-def check_auth_password(self, username, password):
-    if username == os.environ["ssh-username"] and password == os.environ["ssh-password"]:
-        return paramiko.AUTH_SUCCESSFUL
-    return paramiko.AUTH_FAILED
-
-
-def check_auth_publickey(self, username, key):
-    return paramiko.AUTH_FAILED
-
-
-def enable_auth_gssapi(self):
-    return paramiko.AUTH_FAILED
-
-
-sshim.Handler.check_auth_none = check_auth_none
-sshim.Handler.check_auth_password = check_auth_password
-sshim.Handler.check_auth_publickey = check_auth_publickey
-sshim.Handler.enable_auth_gssapi = enable_auth_gssapi
-
-# monkey patching complete
-
 logging.basicConfig(level='DEBUG')
 logger = logging.getLogger()
 

sshim_patch.py

@@ -0,0 +1,28 @@
+from sshim import *
+import paramiko
+import os
+
+
+def check_auth_none(self, username):
+    return paramiko.AUTH_PARTIALLY_SUCCESSFUL
+
+
+def check_auth_password(self, username, password):
+    if username == os.environ["ssh-username"] and password == os.environ["ssh-password"]:
+        return paramiko.AUTH_SUCCESSFUL
+    return paramiko.AUTH_FAILED
+
+
+def check_auth_publickey(self, username, key):
+    return paramiko.AUTH_FAILED
+
+
+def enable_auth_gssapi(self):
+    return paramiko.AUTH_FAILED
+
+
+Handler.check_auth_none = check_auth_none
+Handler.check_auth_password = check_auth_password
+Handler.check_auth_publickey = check_auth_publickey
+Handler.enable_auth_gssapi = enable_auth_gssapi
+