add main()

Author: HidekiHokuto

dir_test.py

@@ -7,21 +7,27 @@
 
 # Description			: Tests to see if the directory testdir exists, if not it will create the directory for you
 from __future__ import print_function
-import os #Import the OS Module
+import os  # Import the OS Module
 import sys
 
-if sys.version_info.major >= 3:
-	input_func = input
-else:
-	input_func = raw_input
 
-CheckDir = input_func("Enter the name of the directory to check : ")
-print()
+def main():
+    if sys.version_info.major >= 3:
+        input_func = input
+    else:
+        input_func = raw_input
 
-if os.path.exists(CheckDir):#Checks if the dir exists
-	print("The directory exists")
-else:
-	print("No directory found for "+CheckDir) #Output if no directory
-	print()
-	os.makedirs(CheckDir)#Creates a new dir for the given name
-	print("Directory created for "+CheckDir)
+    CheckDir = input_func("Enter the name of the directory to check : ")
+    print()
+
+    if os.path.exists(CheckDir):  # Checks if the dir exists
+        print("The directory exists")
+    else:
+        print("No directory found for " + CheckDir)  # Output if no directory
+        print()
+        os.makedirs(CheckDir)  # Creates a new dir for the given name
+        print("Directory created for " + CheckDir)
+
+
+if __name__ == '__main__':
+    main()