Merge pull request #27 from caravelahc/add-different-colors-for-username-and-avatar

Resolves #20: Generate random color for username and avatar background

Author: tarcisioe

picobot/handlers.py

@@ -160,7 +160,7 @@ def add_text(bot: Bot, msg: Message, user_id: int, pack_name: str, emoji: str):
 
     text = msg.reply_to_message.text
     # save as png
-    img_path = sticker_from_text(user_id, username, text, avatar_path, msg_time)
+    img_path = sticker_from_text(user_id, username, text, avatar_path, msg_time, other_user_id)
     try:
         with open(img_path, 'rb') as png_sticker:
             bot.add_sticker_to_set(

picobot/painter.py

@@ -32,6 +32,7 @@
 TITLE_COLOR = "#338cf3"
 TEXT_COLOR = "#dddddd"
 TIME_COLOR = "#6A7B8C"
+FOREGROUND_COLORS = ["#1D9BF9","#FFCC00","#F91880","#7856FF","#FF7A00","#00BA7C"]
 EMOJI_JOINER = chr(0xFE0F)
 
 FONTS = {
@@ -151,15 +152,15 @@ def draw_time(txt_draw: ImageDraw.ImageDraw, points: Box, text="04:20"):
 
 
 def draw_avatar(
-    img: Image, draw: ImageDraw.ImageDraw, username: str, points_balloon: Box, avatar_path: str,
+    img: Image, draw: ImageDraw.ImageDraw, username: str, points_balloon: Box, avatar_path: str, background_color: str
 ):
     y0 = points_balloon.bottom_right.y - AVATAR_SIZE
     y1 = points_balloon.bottom_right.y
     points = Box(MARGIN, y0, MARGIN + AVATAR_SIZE, y1)
     box_position = tuple(a - 2 for a in points.top_left.to_tuple())
     size = AVATAR_SIZE + 4
     if avatar_path == '':
-        draw.ellipse(points.to_list(), fill=TITLE_COLOR)
+        draw.ellipse(points.to_list(), fill=background_color)
         avatar_center = points.center().to_tuple()
         draw.text(
             avatar_center, username[0], anchor='mm', font=FONTS['avatar'], fill='#FFFFFF',
@@ -182,7 +183,7 @@ def draw_avatar(
     img.paste(tmp, mask=avatar_mask)
 
 
-def sticker_from_text(user_id: int, username: str, text: str, avatar_path: str, msg_time: str):
+def sticker_from_text(user_id: int, username: str, text: str, avatar_path: str, msg_time: str, other_user_id: int):
     '''
     Creates an image from a text message, emulating Telegram's message layout/design.
     '''
@@ -225,11 +226,12 @@ def sticker_from_text(user_id: int, username: str, text: str, avatar_path: str,
 
     img = Image.new("RGBA", size, transparent)
     dr = ImageDraw.Draw(img)
-    draw_avatar(img, dr, username, points_balloon=points_balloon, avatar_path=avatar_path)
+    user_color = get_user_color(other_user_id)
+    draw_avatar(img, dr, username, points_balloon=points_balloon, avatar_path=avatar_path, background_color=user_color)
 
     draw_balloon(dr, points=points_balloon, fill=BOX_COLOR)
 
-    draw_username(dr, position=points_balloon.top_left, username=username)
+    draw_username(dr, position=points_balloon.top_left, username=username, fill=user_color)
     draw_message(dr, points=points_balloon, text=final_text, user_size=title_size)
     draw_time(dr, text=msg_time, points=points_balloon)
 
@@ -294,3 +296,7 @@ def generate_avatar_mask(img_size: tuple, points: Box):
     maskdraw.ellipse(points.to_list(), fill='#FFFFFF')
     del maskdraw
     return img
+
+
+def get_user_color(user_id: int) -> str:
+    return FOREGROUND_COLORS[user_id % len(FOREGROUND_COLORS)]