diff options
author | Marius Peter <blendoit@gmail.com> | 2019-06-12 11:58:00 -0700 |
---|---|---|
committer | Marius Peter <blendoit@gmail.com> | 2019-06-12 11:58:00 -0700 |
commit | 976006fbe5f0a141af1041510a640a8e71a79150 (patch) | |
tree | bee77a0ee3f50c79c216221c0d49883c70e00654 /creator.py | |
parent | de1ec9a5fafae98fa1168223a78453e819840c0f (diff) |
added context guard in save_coord
Diffstat (limited to 'creator.py')
-rw-r--r-- | creator.py | 15 |
1 files changed, 9 insertions, 6 deletions
@@ -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):
|