Format code and add pycodestyle setup

Author: Olivier Roques

setup.cfg

@@ -0,0 +1,2 @@
+[pycodestyle]
+ignore = E701

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="sshcontroller",
-    version="1.2",
+    version="1.3",
     author="Olivier Roques",
     author_email="olivier@oroques.dev",
     description="A package to easily run SSH commands",

sshcontroller/sshcontroller.py

@@ -76,9 +76,7 @@ def __init__(
             else:
                 index = len(self.keys) if key_password is None else 0
                 self.keys.insert(index, key)
-
-            if not self.keys:
-                logging.error("No valid key found")
+            if not self.keys: logging.error("No valid key found")
 
     def connect(self):
         try:
@@ -123,8 +121,7 @@ def _run_until_event(
         channel = self.transport.open_session()
         channel.settimeout(2)
         channel.set_combine_stderr(combine_stderr)
-        if shell:
-            channel.get_pty()
+        if shell: channel.get_pty()
         channel.exec_command(command)
 
         if not display and not capture:
@@ -134,19 +131,13 @@ def _run_until_event(
                 try:
                     raw_data = channel.recv(self.nb_bytes)
                 except socket.timeout:
-                    if stop_event.is_set():
-                        break
+                    if stop_event.is_set(): break
                     continue
-
-                if not raw_data:
-                    break
+                if not raw_data: break
                 data = raw_data.decode("utf-8")
-                if display:
-                    print(data, end='')
-                if capture:
-                    output += data
-                if stop_event.is_set():
-                    break
+                if display: print(data, end='')
+                if capture: output += data
+                if stop_event.is_set(): break
 
         channel.close()
 
@@ -168,8 +159,7 @@ def _run_until_exit(
         channel = self.transport.open_session()
         channel.settimeout(timeout)
         channel.set_combine_stderr(combine_stderr)
-        if shell:
-            channel.get_pty()
+        if shell: channel.get_pty()
         channel.exec_command(command)
 
         try:
@@ -178,13 +168,10 @@ def _run_until_exit(
             else:
                 while True:
                     raw_data = channel.recv(self.nb_bytes)
-                    if not raw_data:
-                        break
+                    if not raw_data: break
                     data = raw_data.decode("utf-8")
-                    if display:
-                        print(data, end='')
-                    if capture:
-                        output += data
+                    if display: print(data, end='')
+                    if capture: output += data
         except socket.timeout:
             logging.warning(f"Timeout after {timeout}s")
             exit_code = 1
@@ -235,7 +222,6 @@ def wrapper(*args, **kwargs):
             if not self.transport.is_authenticated():
                 logging.error("SSH session is not ready")
                 return
-
             sftp_channel = SFTPController.from_transport(self.transport)
             r = getattr(sftp_channel, target)(*args, **kwargs)
             sftp_channel.close()