general tidy ups

Author: ghickman

generate.py

@@ -1,11 +1,6 @@
 from svg_wheel import generate_svg_wheel
-from utils import (
-    annotate_wheels,
-    get_top_packages,
-    remove_irrelevant_packages,
-    save_to_file,
-)
-
+from utils import (annotate_wheels, get_top_packages,
+                   remove_irrelevant_packages, save_to_file)
 
 TO_CHART = 360
 

svg_wheel.py

@@ -2,7 +2,6 @@
 import os
 import xml.etree.ElementTree as et
 
-
 HEADERS = '''<?xml version=\"1.0\" standalone=\"no\"?>
 <?xml-stylesheet href="wheel.css" type="text/css"?>
 <!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"
@@ -22,11 +21,11 @@
 PADDING = 10
 RADIUS = 180
 CENTER = PADDING + RADIUS
-TAU = 2*math.pi
+TAU = 2 * math.pi
 
 
 def annular_sector_path(start, stop):
-    inner_radius = RADIUS//2
+    inner_radius = RADIUS // 2
     outer_radius = RADIUS
     cos_stop = math.cos(stop)
     cos_start = math.cos(start)
@@ -36,14 +35,14 @@ def annular_sector_path(start, stop):
     points = {
         'inner_radius': inner_radius,
         'outer_radius': outer_radius,
-        'start_outer_x': CENTER + outer_radius*cos_start,
-        'start_outer_y': CENTER + outer_radius*sin_start,
-        'end_outer_x': CENTER + outer_radius*cos_stop,
-        'end_outer_y': CENTER + outer_radius*sin_stop,
-        'start_inner_x': CENTER + inner_radius*cos_stop,
-        'start_inner_y': CENTER + inner_radius*sin_stop,
-        'end_inner_x': CENTER + inner_radius*cos_start,
-        'end_inner_y': CENTER + inner_radius*sin_start,
+        'start_outer_x': CENTER + outer_radius * cos_start,
+        'start_outer_y': CENTER + outer_radius * sin_start,
+        'end_outer_x': CENTER + outer_radius * cos_stop,
+        'end_outer_y': CENTER + outer_radius * sin_stop,
+        'start_inner_x': CENTER + inner_radius * cos_stop,
+        'start_inner_y': CENTER + inner_radius * sin_stop,
+        'end_inner_x': CENTER + inner_radius * cos_start,
+        'end_inner_y': CENTER + inner_radius * sin_start,
     }
     return PATH_TEMPLATE.format(**points)
 
@@ -60,14 +59,14 @@ def angles(index, total):
     start = index * TAU / total
     stop = (index + 1) * TAU / total
 
-    return (start - TAU/4, stop - TAU/4)
+    return (start - TAU / 4, stop - TAU / 4)
 
 
 def add_fraction(wheel, packages, total):
     text_attributes = {
         'text-anchor': 'middle',
         'dominant-baseline': 'central',
-        'font-size': str(2*OFFSET),
+        'font-size': str(2 * OFFSET),
         'font-family': '"Helvetica Neue",Helvetica,Arial,sans-serif',
         'fill': '#333333',
     }
@@ -85,8 +84,8 @@ def add_fraction(wheel, packages, total):
     # Dividing line
     et.SubElement(
         wheel, 'line',
-        x1=str(CENTER - FRACTION_LINE//2), y1=str(CENTER),
-        x2=str(CENTER + FRACTION_LINE//2), y2=str(CENTER),
+        x1=str(CENTER - FRACTION_LINE // 2), y1=str(CENTER),
+        x2=str(CENTER + FRACTION_LINE // 2), y2=str(CENTER),
         attrib={'stroke': '#333333', 'stroke-width': '2'},
     )
 
@@ -102,7 +101,7 @@ def add_fraction(wheel, packages, total):
 def generate_svg_wheel(packages, total):
     wheel = et.Element(
         'svg',
-        viewBox='0 0 {0} {0}'.format(2*CENTER),
+        viewBox='0 0 {0} {0}'.format(2 * CENTER),
         version='1.1',
         xmlns='http://www.w3.org/2000/svg',
     )