Merge https://github.com/enkore/i3pystatus into credentials

i3pystatus/core/util.py

@@ -365,29 +365,66 @@ def internet():
         return False
 
 
-def make_graph(values, upper_limit=100.0):
+def make_graph(values, lower_limit=0.0, upper_limit=100.0, style="blocks"):
     """
     Draws a graph made of unicode characters.
 
     :param values: An array of values to graph.
-    :param upper_limit: Maximum value for the y axis.
+    :param lower_limit: Minimum value for the y axis (or None for dynamic).
+    :param upper_limit: Maximum value for the y axis (or None for dynamic).
+    :param style: Drawing style ('blocks', 'braille-fill', 'braille-peak', or 'braille-snake').
     :returns: Bar as a string
     """
+
     values = [float(n) for n in values]
+    mn, mx = min(values), max(values)
+    mn = mn if lower_limit is None else min(mn, float(lower_limit))
+    mx = mx if upper_limit is None else max(mx, float(upper_limit))
+    extent = mx - mn
 
-    # Add the upper limit to the end of the array so the graph doesn't distort
+    if style == 'blocks':
-    # as high values drop off the end.
-    values.append(float(upper_limit))
-
         bar = u'_▁▂▃▄▅▆▇█'
         bar_count = len(bar) - 1
-    mn, mx = min(values), max(values)
-    extent = mx - mn
         if extent == 0:
             graph = '_' * len(values)
         else:
-        graph = ''.join(bar[int((n - mn) / extent * bar_count)]
+            graph = ''.join(bar[int((n - mn) / extent * bar_count)] for n in values)
-                        for n in values[:len(values) - 1])  # Don't show the upper limit value.
+    elif style in ['braille-fill', 'braille-peak', 'braille-snake']:
+        # idea from https://github.com/asciimoo/drawille
+        # unicode values from http://en.wikipedia.org/wiki/Braille
+
+        vpad = values if len(values) % 2 == 0 else values + [mn]
+        vscale = [round(4 * (vp - mn) / extent) for vp in vpad]
+        l = len(vscale) // 2
+
+        # do the 2-character collapse separately for clarity
+        if 'fill' in style:
+            vbits = [[0, 0x40, 0x44, 0x46, 0x47][vs] for vs in vscale]
+        elif 'peak' in style:
+            vbits = [[0, 0x40, 0x04, 0x02, 0x01][vs] for vs in vscale]
+        else:
+            assert('snake' in style)
+            # there are a few choices for what to put last in vb2.
+            # arguable vscale[-1] from the _previous_ call is best.
+            vb2 = [vscale[0]] + vscale + [0]
+            vbits = []
+            for i in range(1, l + 1):
+                c = 0
+                for j in range(min(vb2[i - 1], vb2[i], vb2[i + 1]), vb2[i] + 1):
+                    c |= [0, 0x40, 0x04, 0x02, 0x01][j]
+                vbits.append(c)
+
+        # 2-character collapse
+        graph = ''
+        for i in range(0, l, 2):
+            b1 = vbits[i]
+            b2 = vbits[i + 1]
+            if b2 & 0x40:
+                b2 = b2 - 0x30
+            b2 = b2 << 3
+            graph += chr(0x2800 + b1 + b2)
+    else:
+        raise NotImplementedError("Graph drawing style '%s' unimplemented." % style)
     return graph
 
 

i3pystatus/cpu_usage_graph.py

@@ -24,10 +24,13 @@ class CpuUsageGraph(CpuUsage, ColorRangeModule):
         ("cpu", "cpu to monitor, choices are 'usage_cpu' for all or 'usage_cpu*'. R"
                 "eplace '*' by core number starting at 0."),
         ("start_color", "Hex or English name for start of color range, eg '#00FF00' or 'green'"),
-        ("end_color", "Hex or English name for end of color range, eg '#FF0000' or 'red'")
+        ("end_color", "Hex or English name for end of color range, eg '#FF0000' or 'red'"),
+        ("graph_width", "Width of the cpu usage graph"),
+        ("graph_style", "Graph style ('blocks', 'braille-fill', 'braille-peak', or 'braille-snake')"),
     )
 
     graph_width = 15
+    graph_style = 'blocks'
     format = '{cpu_graph}'
     cpu = 'usage_cpu'
 
@@ -43,7 +46,7 @@ class CpuUsageGraph(CpuUsage, ColorRangeModule):
         self.cpu_readings.insert(0, core_reading)
         self.cpu_readings = self.cpu_readings[:self.graph_width]
 
-        graph = make_graph(self.cpu_readings, 100.0)
+        graph = make_graph(self.cpu_readings, 0.0, 100.0, self.graph_style)
         format_options.update({'cpu_graph': graph})
 
         color = self.get_gradient(core_reading, self.colors)

i3pystatus/network.py

@@ -254,6 +254,7 @@ class Network(IntervalModule, ColorRangeModule):
         ("start_color", "Hex or English name for start of color range, eg '#00FF00' or 'green'"),
         ("end_color", "Hex or English name for end of color range, eg '#FF0000' or 'red'"),
         ("graph_width", "Width of the network traffic graph"),
+        ("graph_style", "Graph style ('blocks', 'braille-fill', 'braille-peak', or 'braille-snake')"),
         ("upper_limit",
          "Expected max kb/s. This value controls how the network traffic graph is drawn and in what color"),
         ("graph_type", "Whether to draw the network traffic graph for input or output. "
@@ -276,6 +277,7 @@ class Network(IntervalModule, ColorRangeModule):
     dynamic_color = True
     graph_type = 'input'
     graph_width = 15
+    graph_style = 'blocks'
     upper_limit = 150.0
 
     # Network traffic settings
@@ -329,7 +331,7 @@ class Network(IntervalModule, ColorRangeModule):
         # Cycle array by inserting at the start and chopping off the last element
         self.kbs_arr.insert(0, kbs)
         self.kbs_arr = self.kbs_arr[:self.graph_width]
-        return make_graph(self.kbs_arr, self.upper_limit)
+        return make_graph(self.kbs_arr, 0.0, self.upper_limit, self.graph_style)
 
     def run(self):
         format_values = dict(kbs="", network_graph="", bytes_sent="", bytes_recv="", packets_sent="", packets_recv="",