From 1c0e2c13d4f0acbe42e43d886df4f1067506efed Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Sat, 28 Jan 2017 02:40:27 +0300
Subject: vk: move file i/o to a separate module

---
 vk/tracking/db/io.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 vk/tracking/db/io.py

(limited to 'vk/tracking/db/io.py')

diff --git a/vk/tracking/db/io.py b/vk/tracking/db/io.py
new file mode 100644
index 0000000..04bab3d
--- /dev/null
+++ b/vk/tracking/db/io.py
@@ -0,0 +1,43 @@
+# Copyright (c) 2017 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+from contextlib import contextmanager
+import csv
+import sys
+
+class OutputWriterCSV:
+    def __init__(self, fd=sys.stdout):
+        self._fd = fd
+        self._writer = csv.writer(fd, lineterminator='\n')
+
+    def write_row(self, row):
+        self._writer.writerow(row)
+        self._fd.flush()
+
+class InputReaderCSV:
+    def __init__(self, fd=sys.stdin):
+        self._reader = csv.reader(fd)
+
+    def __iter__(self):
+        return iter(self._reader)
+
+@contextmanager
+def _open_file(path=None, default=None, **kwargs):
+    fd = default
+    if path is None:
+        pass
+    else:
+        fd = open(path, **kwargs)
+    try:
+        yield fd
+    finally:
+        if fd is not default:
+            fd.close()
+
+def open_output_text_file(path=None):
+    return _open_file(path, default=sys.stdout, mode='w', encoding='utf-8')
+
+def open_input_text_file(path=None):
+    return _open_file(path, default=sys.stdin, mode='r', encoding='utf-8')
-- 
cgit v1.2.3