Better looking set_path(wallet_path) method. Flattened function that's easier to read.

lib/wallet.py

@@ -354,19 +354,21 @@ class Wallet:
         """Set the path of the wallet."""
         if wallet_path is not None:
             self.path = wallet_path
-        else:
+            return
-            # backward compatibility: look for wallet file in the default data directory
+        # Look for wallet file in the default data directory.
+        # Keeps backwards compatibility.
         if "HOME" in os.environ:
-                wallet_dir = os.path.join( os.environ["HOME"], '.electrum')
+            wallet_dir = os.path.join(os.environ["HOME"], ".electrum")
         elif "LOCALAPPDATA" in os.environ:
-                wallet_dir = os.path.join( os.environ["LOCALAPPDATA"], 'Electrum' )
+            wallet_dir = os.path.join(os.environ["LOCALAPPDATA"], "Electrum")
         elif "APPDATA" in os.environ:
-                wallet_dir = os.path.join( os.environ["APPDATA"], 'Electrum' )
+            wallet_dir = os.path.join(os.environ["APPDATA"], "Electrum")
         else:
             raise BaseException("No home directory found in environment variables.")
-
+        # Make wallet directory if it does not yet exist.
-            if not os.path.exists( wallet_dir ): os.mkdir( wallet_dir )
+        if not os.path.exists(wallet_dir):
-            self.path = os.path.join( wallet_dir, 'electrum.dat' )
+            os.mkdir(wallet_dir)
+        self.path = os.path.join(wallet_dir, "electrum.dat")
 
     def import_key(self, keypair, password):
         address, key = keypair.split(':')