rustplus 5.6.17__tar.gz → 5.6.18__tar.gz

{rustplus-5.6.17/rustplus.egg-info → rustplus-5.6.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustplus
-Version: 5.6.17
+Version: 5.6.18
 Summary: A python wrapper for the Rust Plus API
 Home-page: https://github.com/olijeffers0n/rustplus
 Author: olijeffers0n
@@ -12,7 +12,7 @@ License-File: LICENSE
 Requires-Dist: websockets
 Requires-Dist: Pillow
 Requires-Dist: asyncio
-Requires-Dist: rustPlusPushReceiver==0.4.1
+Requires-Dist: rustPlusPushReceiver==0.6.0
 Requires-Dist: http-ece
 Requires-Dist: requests
 Requires-Dist: numpy

{rustplus-5.6.17 → rustplus-5.6.18}/rustplus/__init__.py

@@ -22,5 +22,5 @@ from .utils import *
 
 __name__ = "rustplus"
 __author__ = "olijeffers0n"
-__version__ = "5.6.17"
+__version__ = "5.6.18"
 __support__ = "Discord: https://discord.gg/nQqJe8qvP8"

{rustplus-5.6.17 → rustplus-5.6.18}/rustplus/api/rust_api.py

@@ -37,6 +37,7 @@ from ..utils import (
     translate_id_to_stack,
     deprecated,
     generate_grid,
+    avatar_processing,
 )
 
 
@@ -168,6 +169,7 @@ class RustSocket(BaseRustSocket):
         add_icons: bool = False,
         add_events: bool = False,
         add_vending_machines: bool = False,
+        add_team_positions: bool = False,
         override_images: dict = None,
         add_grid: bool = False,
     ) -> Image.Image:
@@ -181,7 +183,7 @@ class RustSocket(BaseRustSocket):
             + (
                 1
                 if [add_icons, add_events, add_vending_machines].count(True) >= 1
-                else 0
+                else 0 + 1 if add_team_positions else 0
             )
         )
 
@@ -271,6 +273,28 @@ class RustSocket(BaseRustSocket):
                         (int(marker.x) - 50, map_size - int(marker.y) - 50),
                         vending_machine,
                     )
+        if add_team_positions:
+            team = await self.get_team_info()
+            for member in team.members:
+                if member.is_alive:
+                    avatar = (
+                        Image.open(
+                            requests.get(
+                                f"https://companion-rust.facepunch.com/api/avatar/{member.steam_id}",
+                                stream=True,
+                            ).raw
+                        )
+                        .resize((100, 100), Image.LANCZOS)
+                        .convert("RGBA")
+                    )
+
+                    player_avatar = avatar_processing(avatar, 5, member.is_online)
+
+                    game_map.paste(
+                        player_avatar,
+                        (format_coord(int(member.x), int(member.y), map_size)),
+                        player_avatar,
+                    )
 
         return game_map.resize((2000, 2000), Image.LANCZOS)
 

{rustplus-5.6.17 → rustplus-5.6.18}/rustplus/utils/rust_utils.py

@@ -11,6 +11,9 @@ ICONS_PATH = "rustplus.api.icons"
 FONT_PATH = "rustplus.utils.fonts"
 GRID_DIAMETER = 146.28571428571428
 
+PLAYER_MARKER_ONLINE_COLOR = (201, 242, 155, 255)
+PLAYER_MARKER_OFFLINE_COLOR = (128, 128, 128, 255)
+
 
 def format_time(protobuf: AppMessage) -> RustTime:
     def convert_time(time) -> str:
@@ -261,3 +264,34 @@ def generate_grid(
             d.text(text_pos, text, fill=color, font=font)
 
     return img
+
+
+def avatar_processing(
+    image: Image.Image, border_size: int, player_online: bool = False
+) -> Image.Image:
+    size_with_border = (
+        image.size[0] + 2 * border_size,
+        image.size[1] + 2 * border_size,
+    )
+
+    border_image = Image.new("RGBA", size_with_border, (0, 0, 0, 0))
+
+    mask = Image.new("L", size_with_border, 0)
+    draw = ImageDraw.Draw(mask)
+
+    draw.ellipse([0, 0, size_with_border[0], size_with_border[1]], fill=255)
+
+    border_layer = Image.new(
+        "RGBA",
+        size_with_border,
+        PLAYER_MARKER_ONLINE_COLOR if player_online else PLAYER_MARKER_OFFLINE_COLOR,
+    )
+    border_image.paste(border_layer, mask=mask)
+
+    image_mask = Image.new("L", image.size, 0)
+    draw = ImageDraw.Draw(image_mask)
+    draw.ellipse([0, 0, image.size[0], image.size[1]], fill=255)
+
+    border_image.paste(image, (border_size, border_size), image_mask)
+
+    return border_image

{rustplus-5.6.17 → rustplus-5.6.18/rustplus.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rustplus
-Version: 5.6.17
+Version: 5.6.18
 Summary: A python wrapper for the Rust Plus API
 Home-page: https://github.com/olijeffers0n/rustplus
 Author: olijeffers0n
@@ -12,7 +12,7 @@ License-File: LICENSE
 Requires-Dist: websockets
 Requires-Dist: Pillow
 Requires-Dist: asyncio
-Requires-Dist: rustPlusPushReceiver==0.4.1
+Requires-Dist: rustPlusPushReceiver==0.6.0
 Requires-Dist: http-ece
 Requires-Dist: requests
 Requires-Dist: numpy

{rustplus-5.6.17 → rustplus-5.6.18}/rustplus.egg-info/requires.txt

@@ -1,7 +1,7 @@
 websockets
 Pillow
 asyncio
-rustPlusPushReceiver==0.4.1
+rustPlusPushReceiver==0.6.0
 http-ece
 requests
 numpy