diff options
author | Egor Tensin <Egor.Tensin@gmail.com> | 2017-01-28 03:19:25 +0300 |
---|---|---|
committer | Egor Tensin <Egor.Tensin@gmail.com> | 2017-01-28 03:19:25 +0300 |
commit | e49227975fd10e9f130f46116f905351c17167c4 (patch) | |
tree | c2b5f75a5e6a7ec5dbf4eaf080f46a96e3e29ac0 /vk | |
parent | add abstract classes for db readers/writers (diff) | |
download | vk-scripts-e49227975fd10e9f130f46116f905351c17167c4.tar.gz vk-scripts-e49227975fd10e9f130f46116f905351c17167c4.zip |
refactoring
Diffstat (limited to 'vk')
-rw-r--r-- | vk/tracking/db/backend/csv.py | 6 | ||||
-rw-r--r-- | vk/tracking/db/io.py | 4 |
2 files changed, 5 insertions, 5 deletions
diff --git a/vk/tracking/db/backend/csv.py b/vk/tracking/db/backend/csv.py index 6176018..4943ff2 100644 --- a/vk/tracking/db/backend/csv.py +++ b/vk/tracking/db/backend/csv.py @@ -4,13 +4,13 @@ # Distributed under the MIT License. from .. import meta -from ..io import InputReaderCSV, OutputWriterCSV +from ..io import FileReaderCSV, FileWriterCSV from ..record import Record from ..timestamp import Timestamp class Writer(meta.Writer): def __init__(self, fd): - self._writer = OutputWriterCSV(fd) + self._writer = FileWriterCSV(fd) def on_initial_status(self, user): self._write_record(user) @@ -32,7 +32,7 @@ class Writer(meta.Writer): class Reader(meta.Reader): def __init__(self, fd): - self._reader = InputReaderCSV(fd) + self._reader = FileReaderCSV(fd) def __iter__(self): return map(Reader._record_from_row, self._reader) diff --git a/vk/tracking/db/io.py b/vk/tracking/db/io.py index 04bab3d..805334d 100644 --- a/vk/tracking/db/io.py +++ b/vk/tracking/db/io.py @@ -7,7 +7,7 @@ from contextlib import contextmanager import csv import sys -class OutputWriterCSV: +class FileWriterCSV: def __init__(self, fd=sys.stdout): self._fd = fd self._writer = csv.writer(fd, lineterminator='\n') @@ -16,7 +16,7 @@ class OutputWriterCSV: self._writer.writerow(row) self._fd.flush() -class InputReaderCSV: +class FileReaderCSV: def __init__(self, fd=sys.stdin): self._reader = csv.reader(fd) |