pep8-ify renderer.py after changes.

Author: theengineear

plotly/matplotlylib/renderer.py

@@ -15,12 +15,12 @@
 import plotly.graph_objs as go
 
 
-
 # Warning format
-def warning_on_one_line(message, category, filename, lineno, file=None, line=None):
-    return '%s:%s: %s:\n\n%s\n\n' % (filename, lineno, category.__name__, message)
+def warning_on_one_line(msg, category, filename, lineno, file=None, line=None):
+    return '%s:%s: %s:\n\n%s\n\n' % (filename, lineno, category.__name__, msg)
 warnings.formatwarning = warning_on_one_line
 
+
 class PlotlyRenderer(Renderer):
     """A renderer class inheriting from base for rendering mpl plots in plotly.
 
@@ -77,16 +77,18 @@ def open_figure(self, fig, props):
         self.msg += "Opening figure\n"
         self.mpl_fig = fig
         self.plotly_fig['layout'] = go.Layout(
-            width=int(props['figwidth']*props['dpi']),
-            height=int(props['figheight']*props['dpi']),
+            width=int(props['figwidth'] * props['dpi']),
+            height=int(props['figheight'] * props['dpi']),
             autosize=False,
             hovermode='closest')
         self.mpl_x_bounds, self.mpl_y_bounds = mpltools.get_axes_bounds(fig)
         margin = go.Margin(
-            l=int(self.mpl_x_bounds[0]*self.plotly_fig['layout']['width']),
-            r=int((1-self.mpl_x_bounds[1])*self.plotly_fig['layout']['width']),
-            t=int((1-self.mpl_y_bounds[1])*self.plotly_fig['layout']['height']),
-            b=int(self.mpl_y_bounds[0]*self.plotly_fig['layout']['height']),
+            l=int(self.mpl_x_bounds[0] * self.plotly_fig['layout']['width']),
+            r=int(
+                (1-self.mpl_x_bounds[1]) * self.plotly_fig['layout']['width']),
+            t=int((1-self.mpl_y_bounds[1]) * self.plotly_fig['layout'][
+                'height']),
+            b=int(self.mpl_y_bounds[0] * self.plotly_fig['layout']['height']),
             pad=0)
         self.plotly_fig['layout']['margin'] = margin
 
@@ -154,10 +156,11 @@ def open_axes(self, ax, props):
             zeroline=False,
             ticks='inside')
         # update defaults with things set in mpl
-        mpl_xaxis, mpl_yaxis = mpltools.prep_xy_axis(ax=ax,
-                                                     props=props,
-                                                     x_bounds=self.mpl_x_bounds,
-                                                     y_bounds=self.mpl_y_bounds)
+        mpl_xaxis, mpl_yaxis = mpltools.prep_xy_axis(
+            ax=ax,
+            props=props,
+            x_bounds=self.mpl_x_bounds,
+            y_bounds=self.mpl_y_bounds)
         xaxis.update(mpl_xaxis)
         yaxis.update(mpl_yaxis)
         bottom_spine = mpltools.get_spine_visible(ax, 'bottom')
@@ -169,7 +172,6 @@ def open_axes(self, ax, props):
         xaxis['showline'] = bottom_spine
         yaxis['showline'] = top_spine
 
-
         # put axes in our figure
         self.plotly_fig['layout']['xaxis{0}'.format(self.axis_ct)] = xaxis
         self.plotly_fig['layout']['yaxis{0}'.format(self.axis_ct)] = yaxis
@@ -216,15 +218,15 @@ def draw_bar(self, coll):
         widths = [bar_props['x1'] - bar_props['x0'] for bar_props in trace]
         heights = [bar_props['y1'] - bar_props['y0'] for bar_props in trace]
         vertical = abs(
-            sum(widths[0]-widths[iii] for iii in range(len(widths)))
+            sum(widths[0] - widths[iii] for iii in range(len(widths)))
         ) < tol
         horizontal = abs(
-            sum(heights[0]-heights[iii] for iii in range(len(heights)))
+            sum(heights[0] - heights[iii] for iii in range(len(heights)))
         ) < tol
         if vertical and horizontal:
             # Check for monotonic x. Can't both be true!
             x_zeros = [bar_props['x0'] for bar_props in trace]
-            if all((x_zeros[iii+1] > x_zeros[iii]
+            if all((x_zeros[iii + 1] > x_zeros[iii]
                     for iii in range(len(x_zeros[:-1])))):
                 orientation = 'v'
             else:
@@ -242,9 +244,9 @@ def draw_bar(self, coll):
             # check if we're stacked or not...
             for old, new in zip(old_heights, new_heights):
                 if abs(old - new) > tol:
-                    self.plotly_fig['layout']['barmode']='stack'
-                    self.plotly_fig['layout']['hovermode']='x'
-            x = [bar['x0']+(bar['x1']-bar['x0'])/2 for bar in trace]
+                    self.plotly_fig['layout']['barmode'] = 'stack'
+                    self.plotly_fig['layout']['hovermode'] = 'x'
+            x = [bar['x0'] + (bar['x1'] - bar['x0']) / 2 for bar in trace]
             y = [bar['y1'] for bar in trace]
             bar_gap = mpltools.get_bar_gap([bar['x0'] for bar in trace],
                                            [bar['x1'] for bar in trace])
@@ -257,21 +259,22 @@ def draw_bar(self, coll):
             # check if we're stacked or not...
             for old, new in zip(old_rights, new_rights):
                 if abs(old - new) > tol:
-                    self.plotly_fig['layout']['barmode']='stack'
-                    self.plotly_fig['layout']['hovermode']='y'
+                    self.plotly_fig['layout']['barmode'] = 'stack'
+                    self.plotly_fig['layout']['hovermode'] = 'y'
             x = [bar['x1'] for bar in trace]
-            y = [bar['y0']+(bar['y1']-bar['y0'])/2 for bar in trace]
+            y = [bar['y0'] + (bar['y1'] - bar['y0']) / 2 for bar in trace]
             bar_gap = mpltools.get_bar_gap([bar['y0'] for bar in trace],
                                            [bar['y1'] for bar in trace])
-        bar = go.Bar(orientation=orientation,
-                  x=x,
-                  y=y,
-                  xaxis='x{0}'.format(self.axis_ct),
-                  yaxis='y{0}'.format(self.axis_ct),
-                  opacity=trace[0]['alpha'],  # TODO: get all alphas if array?
-                  marker=go.Marker(
-                      color=trace[0]['facecolor'],  # TODO: get all
-                      line=go.Line(width=trace[0]['edgewidth'])))  # TODO ditto
+        bar = go.Bar(
+            orientation=orientation,
+            x=x,
+            y=y,
+            xaxis='x{0}'.format(self.axis_ct),
+            yaxis='y{0}'.format(self.axis_ct),
+            opacity=trace[0]['alpha'],  # TODO: get all alphas if array?
+            marker=go.Marker(
+                color=trace[0]['facecolor'],  # TODO: get all
+                line=go.Line(width=trace[0]['edgewidth'])))  # TODO ditto
         if len(bar['x']) > 1:
             self.msg += "    Heck yeah, I drew that bar chart\n"
             self.plotly_fig['data'] += bar,
@@ -282,7 +285,6 @@ def draw_bar(self, coll):
             warnings.warn('found box chart data with length <= 1, '
                           'assuming data redundancy, not plotting.')
 
-
     def draw_marked_line(self, **props):
         """Create a data dict for a line obj.
 
@@ -346,14 +348,15 @@ def draw_marked_line(self, **props):
                 )
             )
         if props['coordinates'] == 'data':
-            marked_line = go.Scatter(mode=mode,
-                                  name=props['label'],
-                                  x=[xy_pair[0] for xy_pair in props['data']],
-                                  y=[xy_pair[1] for xy_pair in props['data']],
-                                  xaxis='x{0}'.format(self.axis_ct),
-                                  yaxis='y{0}'.format(self.axis_ct),
-                                  line=line,
-                                  marker=marker)
+            marked_line = go.Scatter(
+                mode=mode,
+                name=props['label'],
+                x=[xy_pair[0] for xy_pair in props['data']],
+                y=[xy_pair[1] for xy_pair in props['data']],
+                xaxis='x{0}'.format(self.axis_ct),
+                yaxis='y{0}'.format(self.axis_ct),
+                line=line,
+                marker=marker)
             self.plotly_fig['data'] += marked_line,
             self.msg += "    Heck yeah, I drew that line\n"
         else:
@@ -459,7 +462,6 @@ def draw_path(self, **props):
             warnings.warn("I found a path object that I don't think is part "
                           "of a bar chart. Ignoring.")
 
-
     def draw_text(self, **props):
         """Create an annotation dict for a text obj.
 
@@ -491,16 +493,17 @@ def draw_text(self, **props):
         """
         self.msg += "    Attempting to draw an mpl text object\n"
         if not mpltools.check_corners(props['mplobj'], self.mpl_fig):
-            warnings.warn("Looks like the annotation(s) you are trying \n" 
-                          "to draw lies/lay outside the given figure size.\n\n"
-                          "Therefore, the resulting Plotly figure may not be \n"
-                          "large enough to view the full text. To adjust \n"
-                          "the size of the figure, use the 'width' and \n"
-                          "'height' keys in the Layout object. Alternatively,\n"
-                          "use the Margin object to adjust the figure's margins.")
+            warnings.warn(
+                "Looks like the annotation(s) you are trying \n"
+                "to draw lies/lay outside the given figure size.\n\n"
+                "Therefore, the resulting Plotly figure may not be \n"
+                "large enough to view the full text. To adjust \n"
+                "the size of the figure, use the 'width' and \n"
+                "'height' keys in the Layout object. Alternatively,\n"
+                "use the Margin object to adjust the figure's margins.")
         align = props['mplobj']._multialignment
         if not align:
-            align = props['style']['halign'] # mpl default
+            align = props['style']['halign']  # mpl default
         if 'annotations' not in self.plotly_fig['layout']:
             self.plotly_fig['layout']['annotations'] = go.Annotations()
         if props['text_type'] == 'xlabel':
@@ -519,8 +522,9 @@ def draw_text(self, **props):
                             "coordinates\n"
                 x_px, y_px = props['mplobj'].get_transform().transform(
                     props['position'])
-                x, y = mpltools.display_to_paper(x_px, y_px,
-                                              self.plotly_fig['layout'])
+                x, y = mpltools.display_to_paper(
+                    x_px, y_px, self.plotly_fig['layout']
+                )
                 xref = 'paper'
                 yref = 'paper'
                 xanchor = props['style']['halign']  # no difference here!
@@ -602,8 +606,9 @@ def draw_title(self, **props):
                                              self.plotly_fig['layout'])
             annotation = go.Annotation(
                 text=props['text'],
-                font=go.Font(color=props['style']['color'],
-                         size=props['style']['fontsize']
+                font=go.Font(
+                    color=props['style']['color'],
+                    size=props['style']['fontsize']
                 ),
                 xref='paper',
                 yref='paper',
@@ -618,8 +623,9 @@ def draw_title(self, **props):
             self.msg += "          Only one subplot found, adding as a " \
                         "plotly title\n"
             self.plotly_fig['layout']['title'] = props['text']
-            titlefont = go.Font(size=props['style']['fontsize'],
-                             color=props['style']['color']
+            titlefont = go.Font(
+                size=props['style']['fontsize'],
+                color=props['style']['color']
             )
             self.plotly_fig['layout']['titlefont'] = titlefont
 
@@ -649,8 +655,9 @@ def draw_xlabel(self, **props):
         self.msg += "        Adding xlabel\n"
         axis_key = 'xaxis{0}'.format(self.axis_ct)
         self.plotly_fig['layout'][axis_key]['title'] = props['text']
-        titlefont = go.Font(size=props['style']['fontsize'],
-                         color=props['style']['color'])
+        titlefont = go.Font(
+            size=props['style']['fontsize'],
+            color=props['style']['color'])
         self.plotly_fig['layout'][axis_key]['titlefont'] = titlefont
 
     def draw_ylabel(self, **props):
@@ -679,8 +686,9 @@ def draw_ylabel(self, **props):
         self.msg += "        Adding ylabel\n"
         axis_key = 'yaxis{0}'.format(self.axis_ct)
         self.plotly_fig['layout'][axis_key]['title'] = props['text']
-        titlefont = go.Font(size=props['style']['fontsize'],
-                         color=props['style']['color'])
+        titlefont = go.Font(
+            size=props['style']['fontsize'],
+            color=props['style']['color'])
         self.plotly_fig['layout'][axis_key]['titlefont'] = titlefont
 
     def resize(self):