|
12 | 12 |
|
13 | 13 | '''
|
14 | 14 | from __future__ import print_function, division, unicode_literals, absolute_import
|
15 |
| -from builtins import str, zip, range |
| 15 | +from builtins import str, zip, range, open |
16 | 16 | from future.utils import raise_from
|
17 | 17 |
|
18 | 18 | import os
|
@@ -236,15 +236,17 @@ def _gen_output_file_name(self):
|
236 | 236 | return os.path.abspath(base + ".nii")
|
237 | 237 |
|
238 | 238 | def _run_interface(self, runtime):
|
239 |
| - hdr = nb.AnalyzeHeader.from_fileobj( |
240 |
| - open(self.inputs.header_file, 'rb')) |
| 239 | + with open(self.inputs.header_file, 'rb') as hdr_file: |
| 240 | + hdr = nb.AnalyzeHeader.from_fileobj(hdr_file) |
241 | 241 |
|
242 | 242 | if isdefined(self.inputs.affine):
|
243 | 243 | affine = self.inputs.affine
|
244 | 244 | else:
|
245 | 245 | affine = None
|
246 | 246 |
|
247 |
| - data = hdr.data_from_fileobj(open(self.inputs.data_file, 'rb')) |
| 247 | + with open(self.inputs.header_file, 'rb') as data_file: |
| 248 | + data = hdr.data_from_fileobj(data_file) |
| 249 | + |
248 | 250 | img = nb.Nifti1Image(data, affine, hdr)
|
249 | 251 | nb.save(img, self._gen_output_file_name())
|
250 | 252 |
|
@@ -364,11 +366,9 @@ def _gen_output_file_name(self):
|
364 | 366 |
|
365 | 367 | def _run_interface(self, runtime):
|
366 | 368 | import gzip
|
367 |
| - in_file = gzip.open(self.inputs.in_file, 'rb') |
368 |
| - out_file = open(self._gen_output_file_name(), 'wb') |
369 |
| - out_file.write(in_file.read()) |
370 |
| - out_file.close() |
371 |
| - in_file.close() |
| 369 | + with gzip.open(self.inputs.in_file, 'rb') as in_file: |
| 370 | + with open(self._gen_output_file_name(), 'wb') as out_file: |
| 371 | + out_file.write(in_file.read()) |
372 | 372 | return runtime
|
373 | 373 |
|
374 | 374 | def _list_outputs(self):
|
@@ -501,8 +501,9 @@ def merge_csvs(in_list):
|
501 | 501 | try:
|
502 | 502 | in_array = np.loadtxt(in_file, delimiter=',', skiprows=1)
|
503 | 503 | except ValueError as ex:
|
504 |
| - first = open(in_file, 'r') |
505 |
| - header_line = first.readline() |
| 504 | + with open(in_file, 'r') as first: |
| 505 | + header_line = first.readline() |
| 506 | + |
506 | 507 | header_list = header_line.split(',')
|
507 | 508 | n_cols = len(header_list)
|
508 | 509 | try:
|
@@ -681,8 +682,8 @@ def _run_interface(self, runtime):
|
681 | 682 | ext = '.csv'
|
682 | 683 |
|
683 | 684 | out_file = op.abspath(name + ext)
|
684 |
| - file_handle = open(out_file, 'w') |
685 |
| - file_handle.write(csv_headings) |
| 685 | + with open(out_file, 'w') as file_handle: |
| 686 | + file_handle.write(csv_headings) |
686 | 687 |
|
687 | 688 | shape = np.shape(output_array)
|
688 | 689 | typelist = maketypelist(
|
@@ -711,8 +712,9 @@ def _run_interface(self, runtime):
|
711 | 712 | output[extraheading] = extrafieldlist
|
712 | 713 | iflogger.info(output)
|
713 | 714 | iflogger.info(fmt)
|
714 |
| - np.savetxt(file_handle, output, fmt, delimiter=',') |
715 |
| - file_handle.close() |
| 715 | + with open(out_file, 'a') as file_handle: |
| 716 | + np.savetxt(file_handle, output, fmt, delimiter=',') |
| 717 | + |
716 | 718 | return runtime
|
717 | 719 |
|
718 | 720 | def _list_outputs(self):
|
@@ -774,6 +776,8 @@ def _run_interface(self, runtime):
|
774 | 776 | new_line = line.replace('\n', '')
|
775 | 777 | new_line = new_line + ',' + self.inputs.extra_field + '\n'
|
776 | 778 | out_file.write(new_line)
|
| 779 | + in_file.close() |
| 780 | + out_file.close() |
777 | 781 | return runtime
|
778 | 782 |
|
779 | 783 | def _list_outputs(self):
|
|
0 commit comments