diff options
author | Egor Tensin <Egor.Tensin@gmail.com> | 2019-12-23 07:20:36 +0300 |
---|---|---|
committer | Egor Tensin <Egor.Tensin@gmail.com> | 2019-12-23 07:20:36 +0300 |
commit | 7b8cc8a9f455eda41b9c7d70f4561a84fcda941e (patch) | |
tree | b9e262e9a1dbb663c3b9f704a9fe4daf54be0ce9 /vk/tracking/db/io.py | |
parent | Travis: online_sessions.sh: refactoring (diff) | |
download | vk-scripts-7b8cc8a9f455eda41b9c7d70f4561a84fcda941e.tar.gz vk-scripts-7b8cc8a9f455eda41b9c7d70f4561a84fcda941e.zip |
pylint/pep8 fixes
Diffstat (limited to 'vk/tracking/db/io.py')
-rw-r--r-- | vk/tracking/db/io.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/vk/tracking/db/io.py b/vk/tracking/db/io.py index 37d9c53..a89865f 100644 --- a/vk/tracking/db/io.py +++ b/vk/tracking/db/io.py @@ -7,6 +7,7 @@ from contextlib import contextmanager import csv import sys + class FileWriterCSV: def __init__(self, fd=sys.stdout): self._fd = fd @@ -14,7 +15,7 @@ class FileWriterCSV: @staticmethod def _convert_row_old_python(row): - if isinstance(row, list) or isinstance(row, tuple): + if isinstance(row, (list, tuple)): return row return list(row) @@ -24,6 +25,7 @@ class FileWriterCSV: self._writer.writerow(row) self._fd.flush() + class FileReaderCSV: def __init__(self, fd=sys.stdin): self._reader = csv.reader(fd) @@ -31,6 +33,7 @@ class FileReaderCSV: def __iter__(self): return iter(self._reader) + @contextmanager def _open_file(path=None, default=None, **kwargs): if path is None: @@ -39,12 +42,15 @@ def _open_file(path=None, default=None, **kwargs): with open(path, **kwargs) as fd: yield fd + _DEFAULT_ENCODING = 'utf-8' + def open_output_text_file(path=None, mode='w'): return _open_file(path, default=sys.stdout, mode=mode, encoding=_DEFAULT_ENCODING) + def open_input_text_file(path=None): return _open_file(path, default=sys.stdin, mode='r', encoding=_DEFAULT_ENCODING) |