format file

organize.py

@@ -12,30 +12,31 @@ EXT_NAME = "ext.py"
 def organize_files(drive):
     # check os
     operating_system = platform.system()
-    if operating_system == 'Darwin':  # Mac
-        drive = os.path.expanduser(drive)
+    if operating_system == "Windows":  # Windows
+        drive = drive + "\\"
         if not os.path.exists(drive):
             print(f"ERROR! {drive} is not a valid location")
             return
         files = os.listdir(drive)
-        extns = {os.path.splitext(file)[1].strip('.') for file in files}
+        extns = {os.path.splitext(file)[1].strip(".") for file in files}
+
         # Create Folders
         for ext in extns:
             folder = foldername(ext)
-            if folder and not os.path.exists(os.path.join(drive, folder)):
-                os.makedirs(os.path.join(drive, folder))
+            if folder and not os.path.exists(drive + folder):
+                os.makedirs(drive + folder)
 
         # Move Files To Folders
         for file in files:
             if file in [FILE_NAME, EXT_NAME]:
                 continue
-            ext = os.path.splitext(file)[1].strip('.')
+            ext = os.path.splitext(file)[1].strip(".")
             folder = foldername(ext)
             if not folder:
                 continue
 
-            src = os.path.join(drive, file)
-            dest = os.path.join(drive, folder, file)
+            src = drive + file
+            dest = drive + folder + "/" + file
 
             if not os.path.exists(dest):
                 shutil.move(src, dest)
@@ -43,31 +44,30 @@ def organize_files(drive):
 
         print(f"\nSUCCESS! All files organized in {drive}")
 
-    elif operating_system == 'Windows':  # Windows
-        drive = drive + '\\'
+    elif operating_system == "Darwin":  # Mac
+        drive = os.path.expanduser(drive)
         if not os.path.exists(drive):
             print(f"ERROR! {drive} is not a valid location")
             return
         files = os.listdir(drive)
-        extns = {os.path.splitext(file)[1].strip('.') for file in files}
-
+        extns = {os.path.splitext(file)[1].strip(".") for file in files}
         # Create Folders
         for ext in extns:
             folder = foldername(ext)
-            if folder and not os.path.exists(drive + folder):
-                os.makedirs(drive + folder)
+            if folder and not os.path.exists(os.path.join(drive, folder)):
+                os.makedirs(os.path.join(drive, folder))
 
         # Move Files To Folders
         for file in files:
             if file in [FILE_NAME, EXT_NAME]:
                 continue
-            ext = os.path.splitext(file)[1].strip('.')
+            ext = os.path.splitext(file)[1].strip(".")
             folder = foldername(ext)
             if not folder:
                 continue
 
-            src = drive + file
-            dest = drive + folder + '/' + file
+            src = os.path.join(drive, file)
+            dest = os.path.join(drive, folder, file)
 
             if not os.path.exists(dest):
                 shutil.move(src, dest)
@@ -75,7 +75,7 @@ def organize_files(drive):
 
         print(f"\nSUCCESS! All files organized in {drive}")
     else:
-        print(f'{operating_system} not currently supported')
+        print(f"{operating_system} not currently supported")
 
 
 if __name__ == "__main__":