reformat everything for fun

1 files changed, 77 insertions, 43 deletions

diff --git a/vk/tracking/sessions.py b/vk/tracking/sessions.py
index 4b6fd66..4622c2e 100644
--- a/vk/tracking/sessions.py
+++ b/vk/tracking/sessions.py
@@ -339,10 +339,12 @@ class OutputSinkPlot(OutputSinkOnlineSessions):
 
     @staticmethod
     def _extract_values(durations):
-        return (OutputSinkPlot._duration_to_seconds(duration) for duration in durations.values())
+        return (
+            OutputSinkPlot._duration_to_seconds(duration)
+            for duration in durations.values()
+        )
 
-    def process_database(
-            self, group_by, db_reader, time_from=None, time_to=None):
+    def process_database(self, group_by, db_reader, time_from=None, time_to=None):
 
         durations = group_by.group(db_reader, time_from, time_to)
 
@@ -350,8 +352,9 @@ class OutputSinkPlot(OutputSinkOnlineSessions):
         bar_chart.set_title(OutputSinkPlot.TITLE)
         bar_chart.enable_grid_for_values()
         bar_chart.only_integer_values()
-        bar_chart.set_property(bar_chart.get_values_labels(),
-                               fontsize='small', rotation=30)
+        bar_chart.set_property(
+            bar_chart.get_values_labels(), fontsize='small', rotation=30
+        )
         bar_chart.set_value_label_formatter(self._format_duration)
 
         labels = tuple(self._extract_labels(group_by, durations))
@@ -363,9 +366,8 @@ class OutputSinkPlot(OutputSinkOnlineSessions):
         else:
             bar_height = bar_chart.THICK_BAR_HEIGHT
 
-        bars = bar_chart.plot_bars(
-            labels, durations, bar_height=bar_height)
-        bar_chart.set_property(bars, alpha=.33)
+        bars = bar_chart.plot_bars(labels, durations, bar_height=bar_height)
+        bar_chart.set_property(bars, alpha=0.33)
 
         if self._fd is sys.stdout:
             bar_chart.show()
@@ -426,8 +428,7 @@ def _parse_date_range_limit(s):
         return dt.replace(tzinfo=timezone.utc)
     except ValueError:
         msg = 'invalid date range limit (must be in the \'{}\' format): {}'
-        raise argparse.ArgumentTypeError(
-            msg.format(_DATE_RANGE_LIMIT_FORMAT, s))
+        raise argparse.ArgumentTypeError(msg.format(_DATE_RANGE_LIMIT_FORMAT, s))
 
 
 def _parse_args(args=None):
@@ -435,44 +436,78 @@ def _parse_args(args=None):
         args = sys.argv[1:]
 
     parser = argparse.ArgumentParser(
-        description='View/visualize the amount of time people spend online.')
+        description='View/visualize the amount of time people spend online.'
+    )
 
     vk.version.add_to_arg_parser(parser)
 
-    parser.add_argument('db_path', metavar='input', nargs='?',
-                        help='database file path (standard input by default)')
-    parser.add_argument('out_path', metavar='output', nargs='?',
-                        help='output file path (standard output by default)')
-    parser.add_argument('-g', '--group-by',
-                        type=_parse_group_by,
-                        choices=GroupBy,
-                        default=GroupBy.USER,
-                        help='group online sessions by user/date/etc.')
-    parser.add_argument('-i', '--input-format', dest='db_fmt',
-                        type=_parse_database_format,
-                        default=DatabaseFormat.CSV,
-                        choices=DatabaseFormat,
-                        help='specify database format')
-    parser.add_argument('-o', '--output-format', dest='out_fmt',
-                        type=_parse_output_format,
-                        choices=OutputFormat,
-                        default=OutputFormat.CSV,
-                        help='specify output format')
-    parser.add_argument('-a', '--from', dest='time_from',
-                        type=_parse_date_range_limit, default=None,
-                        help='discard online activity prior to this moment')
-    parser.add_argument('-b', '--to', dest='time_to',
-                        type=_parse_date_range_limit, default=None,
-                        help='discard online activity after this moment')
+    parser.add_argument(
+        'db_path',
+        metavar='input',
+        nargs='?',
+        help='database file path (standard input by default)',
+    )
+    parser.add_argument(
+        'out_path',
+        metavar='output',
+        nargs='?',
+        help='output file path (standard output by default)',
+    )
+    parser.add_argument(
+        '-g',
+        '--group-by',
+        type=_parse_group_by,
+        choices=GroupBy,
+        default=GroupBy.USER,
+        help='group online sessions by user/date/etc.',
+    )
+    parser.add_argument(
+        '-i',
+        '--input-format',
+        dest='db_fmt',
+        type=_parse_database_format,
+        default=DatabaseFormat.CSV,
+        choices=DatabaseFormat,
+        help='specify database format',
+    )
+    parser.add_argument(
+        '-o',
+        '--output-format',
+        dest='out_fmt',
+        type=_parse_output_format,
+        choices=OutputFormat,
+        default=OutputFormat.CSV,
+        help='specify output format',
+    )
+    parser.add_argument(
+        '-a',
+        '--from',
+        dest='time_from',
+        type=_parse_date_range_limit,
+        default=None,
+        help='discard online activity prior to this moment',
+    )
+    parser.add_argument(
+        '-b',
+        '--to',
+        dest='time_to',
+        type=_parse_date_range_limit,
+        default=None,
+        help='discard online activity after this moment',
+    )
 
     return parser.parse_args(args)
 
 
 def process_online_sessions(
-        db_path=None, db_fmt=DatabaseFormat.CSV,
-        out_path=None, out_fmt=OutputFormat.CSV,
-        group_by=GroupBy.USER,
-        time_from=None, time_to=None):
+    db_path=None,
+    db_fmt=DatabaseFormat.CSV,
+    out_path=None,
+    out_fmt=OutputFormat.CSV,
+    group_by=GroupBy.USER,
+    time_from=None,
+    time_to=None,
+):
 
     if time_from is not None and time_to is not None:
         if time_from > time_to:
@@ -483,9 +518,8 @@ def process_online_sessions(
         with out_fmt.open_file(out_path) as out_fd:
             out_sink = out_fmt.create_sink(out_fd)
             out_sink.process_database(
-                group_by, db_reader,
-                time_from=time_from,
-                time_to=time_to)
+                group_by, db_reader, time_from=time_from, time_to=time_to
+            )
 
 
 def main(args=None):