aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/bin/utils
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2019-12-23 07:20:36 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2019-12-23 07:20:36 +0300
commit7b8cc8a9f455eda41b9c7d70f4561a84fcda941e (patch)
treeb9e262e9a1dbb663c3b9f704a9fe4daf54be0ce9 /bin/utils
parentTravis: online_sessions.sh: refactoring (diff)
downloadvk-scripts-7b8cc8a9f455eda41b9c7d70f4561a84fcda941e.tar.gz
vk-scripts-7b8cc8a9f455eda41b9c7d70f4561a84fcda941e.zip
pylint/pep8 fixes
Diffstat (limited to 'bin/utils')
-rw-r--r--bin/utils/bar_chart.py2
-rw-r--r--bin/utils/io.py7
2 files changed, 8 insertions, 1 deletions
diff --git a/bin/utils/bar_chart.py b/bin/utils/bar_chart.py
index 522dfed..f051efc 100644
--- a/bin/utils/bar_chart.py
+++ b/bin/utils/bar_chart.py
@@ -7,6 +7,7 @@ import matplotlib.pyplot as plt
from matplotlib import ticker
import numpy as np
+
class BarChartBuilder:
_BAR_HEIGHT = .5
@@ -130,6 +131,7 @@ class BarChartBuilder:
def save(self, path):
self._fig.savefig(path, bbox_inches='tight')
+
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
diff --git a/bin/utils/io.py b/bin/utils/io.py
index 04baa6a..bb8eef9 100644
--- a/bin/utils/io.py
+++ b/bin/utils/io.py
@@ -8,6 +8,7 @@ import csv
import json
import sys
+
class FileWriterJSON:
def __init__(self, fd=sys.stdout):
self._fd = fd
@@ -16,13 +17,14 @@ class FileWriterJSON:
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')
@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)
@@ -31,6 +33,7 @@ class FileWriterCSV:
row = self._convert_row_old_python(row)
self._writer.writerow(row)
+
@contextmanager
def _open_file(path=None, default=None, **kwargs):
if path is None:
@@ -39,8 +42,10 @@ def _open_file(path=None, default=None, **kwargs):
with open(path, **kwargs) as fd:
yield fd
+
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')