From a027e88fb7c6b12070ee86d740acb6bd42d29f0f Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Tue, 23 May 2017 05:25:35 +0300 Subject: bugfix --- vk/tracking/db/io.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'vk') diff --git a/vk/tracking/db/io.py b/vk/tracking/db/io.py index d280634..487a136 100644 --- a/vk/tracking/db/io.py +++ b/vk/tracking/db/io.py @@ -25,16 +25,11 @@ class FileReaderCSV: @contextmanager def _open_file(path=None, default=None, **kwargs): - fd = default if path is None: - pass + yield default else: - fd = open(path, **kwargs) - try: - yield fd - finally: - if fd is not default: - fd.close() + with open(path, **kwargs) as fd: + yield fd _DEFAULT_ENCODING = 'utf-8' -- cgit v1.2.3