Uprava stahovani elevace, slouceni gps utils do jednoho filu, retryable, pridani social image k pilotovi a letu

src/background_jobs/elevation.py

@@ -1,11 +1,13 @@
 from aiohttp import ClientResponseError
 from database import models
 from database.transaction import get_session
+from decorators.retryable import retryable
 from external.elevation import elevation_api
 from external.gpx_parser import GPXParser
 from paths import FLIGHT_GPX_TRACK_PATH
 
 
+@retryable
 async def add_terrain_elevation_to_flight(flight_id: int, gpx_filename: str):
     gpx_parser = GPXParser(f"{FLIGHT_GPX_TRACK_PATH}/{gpx_filename}")
     coordinates = await gpx_parser.get_coordinates_all()
@@ -25,6 +27,7 @@ async def add_terrain_elevation_to_flight(flight_id: int, gpx_filename: str):
         print(e)
 
 
+@retryable
 async def add_terrain_elevation_to_photo(photo):
     try:
         elevation = await elevation_api.get_elevation_for_points([
@@ -34,7 +37,7 @@ async def add_terrain_elevation_to_photo(photo):
             print("Cannot get elevation")
             return
 
-        terrain_elevation = elevation[0]['elevation']
+        terrain_elevation = elevation[0].elevation
         async with get_session() as db:
             await models.Photo.update(db_session=db, obj=photo, data={"terrain_elevation": terrain_elevation})
     except Exception as e:

src/background_jobs/flight_title_photo.py

@@ -0,0 +1,58 @@
+from PIL import Image, ImageDraw, ImageOps
+
+
+async def add_circular_avatar(base_image_path, avatar_image_path, output_image_path):
+    # Load the base image
+    base_image = Image.open(base_image_path)
+    base_width, base_height = base_image.size
+
+    # Calculate the new dimensions for cropping to a 1.9:1 aspect ratio
+    new_width = base_width
+    new_height = int(new_width / 1.9)
+    if new_height > base_height:
+        new_height = base_height
+        new_width = int(new_height * 1.9)
+
+    # Center crop the base image to the new dimensions
+    left = (base_width - new_width) // 2
+    top = (base_height - new_height) // 2
+    right = (base_width + new_width) // 2
+    bottom = (base_height + new_height) // 2
+    base_image = base_image.crop((left, top, right, bottom))
+    base_width, base_height = base_image.size
+
+    # Load the avatar image
+    avatar_image = Image.open(avatar_image_path)
+    avatar_size = int(base_height * 0.3)  # Set the size of the avatar
+    avatar_image = avatar_image.resize((avatar_size, avatar_size), Image.LANCZOS)
+
+    # Create a mask to make the avatar circular
+    mask = Image.new('L', (avatar_size, avatar_size), 0)
+    draw = ImageDraw.Draw(mask)
+    draw.ellipse((0, 0, avatar_size, avatar_size), fill=255)
+
+    # Apply the mask to the avatar image
+    avatar_image = ImageOps.fit(avatar_image, mask.size, centering=(0.5, 0.5))
+    avatar_image.putalpha(mask)
+
+    # Create a border around the avatar
+    border_size = int(avatar_size * 0.03)
+    border_image = Image.new('RGBA', (avatar_size + 2 * border_size, avatar_size + 2 * border_size), (255, 255, 255, 0))
+    border_mask = Image.new('L', (avatar_size + 2 * border_size, avatar_size + 2 * border_size), 0)
+    border_draw = ImageDraw.Draw(border_mask)
+    border_draw.ellipse((0, 0, avatar_size + 2 * border_size, avatar_size + 2 * border_size), fill=255)
+    border_draw.ellipse((border_size, border_size, avatar_size + border_size, avatar_size + border_size), fill=0)
+
+    avatar_margin = int(avatar_size * 0.1)
+
+    border_image.putalpha(border_mask)
+    border_image.paste(avatar_image, (border_size, border_size), avatar_image)
+
+    # Calculate the position to paste the avatar (bottom right corner)
+    position = (base_width - avatar_size - 2 * border_size - avatar_margin, base_height - avatar_size - 2 * border_size - avatar_margin)
+
+    # Paste the avatar with border onto the base image
+    base_image.paste(border_image, position, border_image)
+
+    # Save the result
+    base_image.save(output_image_path)