Rework with black

ivatar/utils.py

@@ -1,59 +1,65 @@
-'''
+# -*- coding: utf-8 -*-
+"""
 Simple module providing reusable random_string function
-'''
+"""
 import random
 import string
 from PIL import Image, ImageDraw
 
 
 def random_string(length=10):
-    '''
+    """
     Return some random string with default length 10
-    '''
-    return ''.join(random.SystemRandom().choice(
-        string.ascii_lowercase + string.digits) for _ in range(length))
+    """
+    return "".join(
+        random.SystemRandom().choice(string.ascii_lowercase + string.digits)
+        for _ in range(length)
+    )
 
 
 def openid_variations(openid):
-    '''
+    """
     Return the various OpenID variations, ALWAYS in the same order:
     - http w/ trailing slash
     - http w/o trailing slash
     - https w/ trailing slash
     - https w/o trailing slash
-    '''
+    """
 
     # Make the 'base' version: http w/ trailing slash
-    if openid.startswith('https://'):
-        openid = openid.replace('https://', 'http://')
-    if openid[-1] != '/':
-        openid = openid + '/'
+    if openid.startswith("https://"):
+        openid = openid.replace("https://", "http://")
+    if openid[-1] != "/":
+        openid = openid + "/"
 
     # http w/o trailing slash
     var1 = openid[0:-1]
-    var2 = openid.replace('http://', 'https://')
+    var2 = openid.replace("http://", "https://")
     var3 = var2[0:-1]
     return (openid, var1, var2, var3)
 
-def mm_ng(idhash, size=80, add_red=0, add_green=0, add_blue=0):  #pylint: disable=too-many-locals
-    '''
+
+def mm_ng(
+    idhash, size=80, add_red=0, add_green=0, add_blue=0
+):  # pylint: disable=too-many-locals
+    """
     Return an MM (mystery man) image, based on a given hash
     add some red, green or blue, if specified
-    '''
+    """
 
     # Make sure the lightest bg color we paint is e0, else
     # we do not see the MM any more
-    if idhash[0] == 'f':
-        idhash = 'e0'
+    if idhash[0] == "f":
+        idhash = "e0"
 
     # How large is the circle?
-    circlesize = size*0.6
+    circlesize = size * 0.6
 
     # Coordinates for the circle
-    start_x = int(size*0.2)
-    end_x = start_x+circlesize
-    start_y = int(size*0.05)
-    end_y = start_y+circlesize
+    start_x = int(size * 0.2)
+    end_x = start_x + circlesize
+    start_y = int(size * 0.05)
+    end_y = start_y + circlesize
 
     # All are the same, based on the input hash
     # this should always result in a "gray-ish" background
@@ -62,44 +68,47 @@ def mm_ng(idhash, size=80, add_red=0, add_green=0, add_blue=0):  #pylint: disabl
     blue = idhash[0:2]
 
     # Add some red (i/a) and make sure it's not over 255
-    red = hex(int(red, 16)+add_red).replace('0x', '')
+    red = hex(int(red, 16) + add_red).replace("0x", "")
     if int(red, 16) > 255:
-        red = 'ff'
+        red = "ff"
     if len(red) == 1:
-        red = '0%s' % red
+        red = "0%s" % red
 
     # Add some green (i/a) and make sure it's not over 255
-    green = hex(int(green, 16)+add_green).replace('0x', '')
+    green = hex(int(green, 16) + add_green).replace("0x", "")
     if int(green, 16) > 255:
-        green = 'ff'
+        green = "ff"
     if len(green) == 1:
-        green = '0%s' % green
+        green = "0%s" % green
 
     # Add some blue (i/a) and make sure it's not over 255
-    blue = hex(int(blue, 16)+add_blue).replace('0x', '')
+    blue = hex(int(blue, 16) + add_blue).replace("0x", "")
     if int(blue, 16) > 255:
-        blue = 'ff'
+        blue = "ff"
     if len(blue) == 1:
-        blue = '0%s' % blue
+        blue = "0%s" % blue
 
     # Assemable the bg color "string" in webnotation. Eg. '#d3d3d3'
-    bg_color = '#' + red + green + blue
+    bg_color = "#" + red + green + blue
 
     # Image
-    image = Image.new('RGB', (size, size))
+    image = Image.new("RGB", (size, size))
     draw = ImageDraw.Draw(image)
 
     # Draw background
     draw.rectangle(((0, 0), (size, size)), fill=bg_color)
 
     # Draw MMs head
-    draw.ellipse((start_x, start_y, end_x, end_y), fill='white')
+    draw.ellipse((start_x, start_y, end_x, end_y), fill="white")
 
     # Draw MMs 'body'
-    draw.polygon((
-        (start_x+circlesize/2, size/2.5),
-        (size*0.15, size),
-        (size-size*0.15, size)),
-                 fill='white')
+    draw.polygon(
+        (
+            (start_x + circlesize / 2, size / 2.5),
+            (size * 0.15, size),
+            (size - size * 0.15, size),
+        ),
+        fill="white",
+    )
 
     return image