Merge pull request #151 from ashwek/fix

Update treat_names_here.py

Author: ionicc

Automation/src/treat_names_here/treat_names_here.py

@@ -1,30 +1,44 @@
-import os
-import sys
-
-""" 
+"""
 Renames the filenames within the same directory to:
 (1) Changes first letter to uppercase
 (2) Change the filename to uppercase
 (3) Changes the filename to lowercase
 Usage:
-python upper treat_names_here.py
-python lower treat_names_here.py
-python capitalize treat_names_here.py
+python treat_names_here.py upper
+python treat_names_here.py lower
+python treat_names_here.py capitalize
 """
 
+import os
+import sys
+
 path =  os.getcwd()
 filenames = os.listdir(path)
-opt = str(sys.argv[1])
+
+if(len(sys.argv) > 1):			# check if command-line argument is given
+	opt = str(sys.argv[1])
+else:			# if not given, print usage
+	print("""No command line argument was found
+	Usage:
+	  python treat_names_here.py upper
+	  python treat_names_here.py lower
+	  python treat_names_here.py capitalize
+	""")
+	filenames = ""		# and set filenames to empty string, to bypass for loop
 
 for filename in filenames:
 	if filename != os.path.basename(__file__):
 		if opt == 'lower':
 			os.rename(filename, filename.lower())
 
-		if opt == 'upper':
+		elif opt == 'upper':
 			os.rename(filename, filename.upper())
 
-		if opt == 'capitalize':
+		elif opt == 'capitalize':
 			os.rename(filename, filename.capitalize())
+		
+		else:
+			print("Unknown argument given")
+			break
 	else:
-		print('omitted actual file')
+		print('omitted actual file')