es02 4+5 reading from file refactored

kainoj · Apr 10, 2018 · c8f2451 · c8f2451
1 parent 790e49e
commit c8f2451
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/es02/ex4.py b/es02/ex4.py
@@ -142,6 +142,12 @@ def playBfs(self):
 
         return self.traceback(state)
 
+def readBoard(finput):
+    board = []
+    with open(finput) as f:
+        board = [list(line.strip('\n')) for line in f]
+    return board
+
 
 if __name__ == '__main__':
 
@@ -151,11 +157,7 @@ def playBfs(self):
     if len(sys.argv) == 2:
         finput = sys.argv[1]
 
-    board = []
-    with open(finput) as f:
-        board = [list(line.strip('\n')) for line in f]
-
-    coma = Commando(board, uncert=True)
+    coma = Commando(readBoard(finput), uncert=True)
 
     ans = coma.playBfs()
     print(ans)

diff --git a/es02/ex5.py b/es02/ex5.py
@@ -83,9 +83,7 @@ def astar(self):
     if len(sys.argv) == 2:
         finput = sys.argv[1]
 
-    board = []
-    with open(finput) as f:
-        board = [list(line.strip('\n')) for line in f]
+    board = ex4.readBoard(finput)
 
     coma = Commando(board)
     ans = coma.astar()