From 976006fbe5f0a141af1041510a640a8e71a79150 Mon Sep 17 00:00:00 2001
From: Marius Peter <blendoit@gmail.com>
Date: Wed, 12 Jun 2019 11:58:00 -0700
Subject: added context guard in save_coord

---
 creator.py | 15 +++++++++------
 main.py    |  8 ++++----
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/creator.py b/creator.py
index c6cae4e..35784a8 100644
--- a/creator.py
+++ b/creator.py
@@ -85,16 +85,19 @@ class Coordinates:
         # print('\n')
         return None
 
-    def save_coord(self, save_dir_path):
+    def save_coord(self, save_dir_path, number):
         """
         Save all the object's coordinates (must be full path).
         """
 
-        file_name = str(self)
-        full_path = os.path.join(save_dir_path, file_name + '.txt')
-        file = open(full_path, 'w')
-        sys.stdout = file
-        self.print_coord(4)
+        file_name = '{}_{}.txt'.format(self, number)
+        full_path = os.path.join(save_dir_path, file_name)
+        # sys.stdout = open(full_path, 'w')
+        # self.print_coord(2)
+        with open(full_path, 'w') as sys.stdout:
+            self.print_coord(2)
+        # It is cleaner to use this context guard to ensure file is closed
+
         return None
 
     def pack_coord(self):
diff --git a/main.py b/main.py
index a2dbd94..5d2e90a 100644
--- a/main.py
+++ b/main.py
@@ -21,7 +21,7 @@ import random
 import time
 start_time = time.time()
 
-CHORD_LENGTH = 1000
+CHORD_LENGTH = 100
 SEMI_SPAN = 200
 
 POP_SIZE = 1
@@ -58,9 +58,9 @@ def main():
         creator.plot(af, af.spar, af.stringer)
 
         # Save component coordinates
-        af.save_coord(SAVE_PATH)
-        af.spar.save_coord(SAVE_PATH)
-        af.stringer.save_coord(SAVE_PATH)
+        af.save_coord(SAVE_PATH, _)
+        af.spar.save_coord(SAVE_PATH, _)
+        af.stringer.save_coord(SAVE_PATH, _)
 
     # Print final execution time
     print("--- %s seconds ---" % (time.time() - start_time))
-- 
cgit v1.2.3