From 7112abbe5dd8aebf44c097043f413a0542c05e5c Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sat, 28 Jan 2017 03:11:23 +0300 Subject: add abstract classes for db readers/writers --- vk/tracking/db/backend/csv.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'vk/tracking/db/backend/csv.py') diff --git a/vk/tracking/db/backend/csv.py b/vk/tracking/db/backend/csv.py index 5b2cc7d..6176018 100644 --- a/vk/tracking/db/backend/csv.py +++ b/vk/tracking/db/backend/csv.py @@ -3,13 +3,12 @@ # For details, see https://github.com/egor-tensin/vk-scripts. # Distributed under the MIT License. -from collections.abc import Iterable - +from .. import meta from ..io import InputReaderCSV, OutputWriterCSV from ..record import Record from ..timestamp import Timestamp -class Writer: +class Writer(meta.Writer): def __init__(self, fd): self._writer = OutputWriterCSV(fd) @@ -31,7 +30,7 @@ class Writer: def _record_to_row(record): return [str(record.get_timestamp())] + [str(record[field]) for field in record] -class Reader(Iterable): +class Reader(meta.Reader): def __init__(self, fd): self._reader = InputReaderCSV(fd) -- cgit v1.2.3