from argparse import ArgumentParser
from matplotlib import pyplot as plt
from graph import Greengraph


def process():
    parser = ArgumentParser(
        description="Produce graph quantifying the amount of green land between two locations")
    parser.add_argument("--start", required=True, nargs="+",
                        help="The starting location ")
    parser.add_argument("--end",  required=True, nargs="+",
                        help="The ending location")
    parser.add_argument("--steps",
                        help="The number of steps between the starting and ending locations, defaults to 10")
    parser.add_argument("--out",
                        help="The output filename, defaults to graph.png")
    arguments = parser.parse_args()

    #mygraph = Greengraph(arguments.start, arguments.end)
    if arguments.steps:
        data = mygraph.green_between(arguments.steps)
    else:
        data = mygraph.green_between(10)
    plt.plot(data)
    # TODO add a title and axis labels to this graph
    if arguments.out:
        plt.savefig(arguments.out)
    else:
        plt.savefig("graph.png")
    print arguments.start
    print arguments.end

if __name__ == "__main__":
    process()

from argparse import ArgumentParser
from matplotlib import pyplot as plt
from graph import Greengraph


def process():
    parser = ArgumentParser(
        description="Produce graph quantifying the amount of green land between two locations")
    parser.add_argument("--start", required=True,
                        help="The starting location ")
    parser.add_argument("--end",  required=True,
                        help="The ending location")
    parser.add_argument("--steps",
                        help="The number of steps between the starting and ending locations, defaults to 10")
    parser.add_argument("--out",
                        help="The output filename, defaults to graph.png")
    arguments = parser.parse_args()

    mygraph = Greengraph(arguments.start, arguments.end)
    if arguments.steps:
        data = mygraph.green_between(arguments.steps)
    else:
        data = mygraph.green_between(10)
    plt.plot(data)
    # TODO add a title and axis labels to this graph
    if arguments.out:
        plt.savefig(arguments.out)
    else:
        plt.savefig("graph.png")

if __name__ == "__main__":
    process()

--- 
+++ 
@@ -6,9 +6,9 @@
 def process():
     parser = ArgumentParser(
         description="Produce graph quantifying the amount of green land between two locations")
-    parser.add_argument("--start", required=True,
+    parser.add_argument("--start", required=True, nargs="+",
                         help="The starting location ")
-    parser.add_argument("--end",  required=True,
+    parser.add_argument("--end",  required=True, nargs="+",
                         help="The ending location")
     parser.add_argument("--steps",
                         help="The number of steps between the starting and ending locations, defaults to 10")
@@ -16,7 +16,7 @@
                         help="The output filename, defaults to graph.png")
     arguments = parser.parse_args()
 
-    mygraph = Greengraph(arguments.start, arguments.end)
+    #mygraph = Greengraph(arguments.start, arguments.end)
     if arguments.steps:
         data = mygraph.green_between(arguments.steps)
     else:
@@ -27,6 +27,8 @@
         plt.savefig(arguments.out)
     else:
         plt.savefig("graph.png")
+    print arguments.start
+    print arguments.end
 
 if __name__ == "__main__":
     process()
Allow start and end arguments to take inputs of multiple words such as 'New York'
