Initial commit, refactored some things from original code and removed WP functionality

2023-03-06 17:04:59 +01:00
commit a4a76966fe
25 changed files with 517 additions and 0 deletions
--- a/bot/helpers.py
+++ b/bot/helpers.py
@@ -0,0 +1,64 @@
+import logging
+import constants
+import requests
+from PIL import Image, ImageDraw, ImageFont
+
+# Enable logging
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
+)
+
+logger = logging.getLogger(__name__)
+
+def create_image(product_id, price):
+    width = 1280
+    height = 800
+    baseheight = int(height * 0.85)
+    # límite de ancho para la parte izquierda (producto)
+    wlimit = int(((width / 3) * 2) - 80)
+    # límite de ancho para los logos (homelabers y amazon)
+    wlogo = int(width * 0.2)
+    # fuente y tamaño
+    font = ImageFont.truetype("/app/data/fonts/Roboto-Bold.ttf", 90)
+    # inicializamos canvas
+    image = Image.new('RGBA', (width, height), (255, 255, 255))
+    # logo homelabers, redimensionamos y ponemos en la parte derecha arriba
+    logo_image = Image.open("/app/data/images/logo.png").convert("RGBA")
+    lpercent = wlogo / float(logo_image.size[0])
+    hlogo = int((float(logo_image.size[1]) * float(lpercent)))
+    logo_image = logo_image.resize((wlogo, hlogo), Image.Resampling.LANCZOS)
+    image.paste(logo_image, (int((width / 6) * 5 - logo_image.size[0] / 2), int(height * 0.1)), logo_image)
+    # logo amazon, redimensionamos y ponemos en la parte derecha abajo
+    amazon_logo = Image.open("/app/data/images/Amazon_logo.png").convert("RGBA")
+    lpercent = wlogo / float(amazon_logo.size[0])
+    hlogo = int((float(amazon_logo.size[1]) * float(lpercent)))
+    amazon_logo = amazon_logo.resize((wlogo, hlogo), Image.Resampling.LANCZOS)
+    image.paste(amazon_logo, (int((width / 6) * 5 - amazon_logo.size[0] / 2), int(height - height * 0.2)), amazon_logo)
+    # dibujamos rectángulo verde externo, con un margen externo y ancho determinado
+    draw = ImageDraw.Draw(image)
+    wtext, htext = draw.textsize(price, font=font)
+    draw.text(((width / 6) * 5 - wtext / 2, height / 2 - htext / 2), price, (0, 0, 0), font=font)
+    draw.rectangle([15, 15, width - 15, height - 15], width = 15, outline="#20e163")
+    # ponemos la imagen del producto en la parte izquierda y se redimensiona dependiendo de lo ancho
+    product_image = Image.open(f"/app/data/images/products/{product_id}.jpg")
+    hpercent = (baseheight / float(product_image.size[1]))
+    wsize = int((float(product_image.size[0]) * float(hpercent)))
+    if wsize < wlimit:
+        product_image = product_image.resize((wsize, baseheight), Image.Resampling.LANCZOS)
+    else:
+        wpercent = wlimit / float(product_image.size[0])
+        hsize = int((float(product_image.size[1]) * float(wpercent)))
+        product_image = product_image.resize((wlimit, hsize), Image.Resampling.LANCZOS)
+    image.paste(product_image, (int((width/3)-(product_image.size[0]/2)), int((height/2) - (product_image.size[1]/2))))
+    # guardamos la imagen con otro nombre
+    image.save(f"/app/data/images/products/{product_id}_composed.png", quality=95)
+
+def new_refer_url(pcode, extraparams=None):
+    return constants.baseURL+pcode+"?tag="+constants.affiliate_tag+extraparams
+
+def unshort_url(url):
+    session = requests.Session()
+    #resp = session.head("https://"+url, allow_redirects=True)
+    resp = session.get("https://"+url, allow_redirects=True)
+    logging.info(f"Unshorted URL: {resp.url}")
+    return resp.url