changeset 14:a72cf3c7c976 draft

main: use args to get input filename instead of hardwiring it
author Jordi Gutiérrez Hermoso <jordigh@octave.org>
date Wed, 11 Mar 2015 16:46:25 -0400
parents ad39a394a011
children d9aa5e25859c
files optim.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/optim.py
+++ b/optim.py
@@ -165,7 +165,8 @@
     return maxcapital, maxplan
 
 def main():
-    cases = parseinput("input.txt")
+    args = process_options()
+    cases = parseinput(args.file)
     for (number, case) in enumerate(cases):
         maxcapital = brute_force(case)
         print "Case %d: %d" % (number + 1, maxcapital)