From 31422dd2fcc657757beb37a386f9f2da5c5a8568 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sat, 28 Jan 2017 03:22:36 +0300 Subject: bin.utils.output -> bin.utils.io --- bin/utils/output.py | 43 ------------------------------------------- 1 file changed, 43 deletions(-) delete mode 100644 bin/utils/output.py (limited to 'bin/utils/output.py') diff --git a/bin/utils/output.py b/bin/utils/output.py deleted file mode 100644 index 70c1a3a..0000000 --- a/bin/utils/output.py +++ /dev/null @@ -1,43 +0,0 @@ -# Copyright (c) 2017 Egor Tensin -# 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 json -import sys - -class FileWriterJSON: - def __init__(self, fd=sys.stdout): - self._fd = fd - - def write(self, something): - self._fd.write(json.dumps(something, indent=3, ensure_ascii=False)) - self._fd.write('\n') - -class FileWriterCSV: - def __init__(self, fd=sys.stdout): - self._writer = csv.writer(fd, lineterminator='\n') - - def write_row(self, row): - self._writer.writerow(row) - -@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 sys.stdout: - fd.close() - -def open_output_text_file(path=None): - return _open_file(path, default=sys.stdout, mode='w', encoding='utf-8') - -def open_output_binary_file(path=None): - return _open_file(path, default=sys.stdout, mode='wb') -- cgit v1.2.3