Ex30 arg parse

Numbers.txt

@@ -0,0 +1 @@
+1 2 4

Weights.txt

@@ -0,0 +1 @@
+1 1 2

squares.py

@@ -1,5 +1,6 @@
 """Computation of weighted average of squares."""
-
+from argparse import ArgumentParser
+from argparse import FileType
 
 def average_of_squares(list_of_numbers, list_of_weights=None):
     """ Return the weighted average of a list of values.
@@ -29,7 +30,7 @@ def average_of_squares(list_of_numbers, list_of_weights=None):
         for number, weight
         in zip(list_of_numbers, effective_weights)
     ]
-    return sum(squares)
+    return sum(squares) / len(list_of_numbers)
 
 
 def convert_numbers(list_of_strings):
@@ -38,7 +39,7 @@ def convert_numbers(list_of_strings):
     Example:
     --------
     >>> convert_numbers(["4", " 8 ", "15 16", " 23    42 "])
-    [4, 8, 15, 16]
+    [4.0, 8.0, 15.0, 16.0, 23.0, 42.0]
 
     """
     all_numbers = []
@@ -51,12 +52,21 @@ def convert_numbers(list_of_strings):
 
 
 if __name__ == "__main__":
-    numbers_strings = ["1","2","4"]
-    weight_strings = ["1","1","1"]        
+
+    parser = ArgumentParser(description="Returns weighted mean of squares")
+
+    parser.add_argument("Numbers", type=FileType('r'), help="File name for input numbers")
+    parser.add_argument("--Weights", "-w", type=FileType('r'), help='File name for input weights')
 
-    numbers = convert_numbers(numbers_strings)
-    weights = convert_numbers(weight_strings)
+    INP = parser.parse_args()
 
-    result = average_of_squares(numbers, weights)
+    Numbers_list = convert_numbers(INP.Numbers.readlines())
+
+    if INP.Weights is not None:
+        Weights_list = convert_numbers(INP.Weights.readlines())
+    else:
+        Weights_list = None
+
+    result = average_of_squares(Numbers_list, Weights_list)
 
     print(result)