From 976006fbe5f0a141af1041510a640a8e71a79150 Mon Sep 17 00:00:00 2001 From: Marius Peter Date: Wed, 12 Jun 2019 11:58:00 -0700 Subject: added context guard in save_coord --- creator.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'creator.py') 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): -- cgit v1.2.3