step-4 and all scenarios convered. Solution achieved

Author: sambitr

Coding Challenge#33[linux head]/README.md

@@ -103,3 +103,40 @@ Hello, World
 
 C:\Users\kumar\Documents\Python Scripts\Coding-Challenges\Coding Challenge#33[linux head]>
 <br>########## ==================<br>
+
+C:\Users\kumar\Documents\Python Scripts\Coding-Challenges\Coding Challenge#33[linux head]>python head.py -n 10 text.txt text2.txt
+==> text.txt <==
+The Project Gutenberg eBook of The Art of War
+
+
+
+This ebook is for the use of anyone anywhere in the United States and
+
+most other parts of the world at no cost and with almost no restrictions
+
+whatsoever. You may copy it, give it away or re-use it under the terms
+
+of the Project Gutenberg License included with this ebook or online
+
+at www.gutenberg.org. If you are not located in the United States,
+
+you will have to check the laws of the country where you are located
+
+before using this eBook.
+
+
+
+==> text2.txt <==
+Hello, World
+
+
+
+
+
+
+
+
+
+
+C:\Users\kumar\Documents\Python Scripts\Coding-Challenges\Coding Challenge#33[linux head]>python head.py -n 10 text.txt
+<br>########## ================== <br>

Coding Challenge#33[linux head]/head.py

@@ -1,15 +1,11 @@
 import sys
 
-#print(sys.argv[0], sys.argv[1], sys.argv[2], sys.argv[3])
 if len(sys.argv) == 2: ## first item is the script name and second item is the file name sent as argument
-    ## take the file name from the command line and print the contents of it
     file_name = sys.argv[1]
-    #print(f"file name is: {file_name}")
 
     file_handle = open(file_name, 'r', encoding="utf-8")
     print(file_handle.read())
     file_handle.close()
-    ######################
 else:
     ## no file name. Take option from command line and print it. Shold exit ater 10 lines.
     if len(sys.argv) == 1 :
@@ -20,13 +16,27 @@
 
     elif sys.argv[1] == "-n":
         line_to_print = sys.argv[2]
-        file_name_to_work = sys.argv[3]
-        #file_operation(file_name_to_work, sys.argv[1], line_to_print)
-        #print(f"line to print {line_to_print}, file to work {file_name_to_work}")
-        file_hdl = open(file_name_to_work, 'r', encoding="utf-8")
-        for i in range(int(line_to_print)):
-            print(file_hdl.readline())
-        file_hdl.close()
+        files = sys.argv[3:]
+        if len(files) > 1:
+            for fileName in files:
+                print(f"==> {fileName} <==")
+                filehdl = open(fileName, 'r', encoding="utf-8")
+                for i in range(int(line_to_print)):
+                    line = filehdl.readline()
+                    if line == "": ## end of file has been reached
+                        break
+                    print(line)
+                filehdl.close()
+        else:
+            file_name_to_work = sys.argv[3]
+            file_hdl = open(file_name_to_work, 'r', encoding="utf-8")
+            for i in range(int(line_to_print)):
+                line = file_hdl.readline()
+                if line == "": ## end of file has been reached
+                    break
+                print(line)
+
+            file_hdl.close()
 
     elif sys.argv[1] == "-c":
         chars_to_print = sys.argv[2]