completely rewrite python wrapper using generated interop

Author: malytomas

.github/workflows/python.yml

@@ -0,0 +1,38 @@
+name: Python
+
+on: [push, pull_request]
+
+jobs:
+  mypy:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: setup python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.13"
+
+      - name: install dependencies
+        run: |
+          pip install mypy cffi types-cffi
+
+      - name: run mypy
+        run: mypy python
+
+  pyright:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: setup python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.13"
+
+      - name: install dependencies
+        run: |
+          pip install pyright cffi types-cffi
+
+      - name: run pyright
+        run: pyright python

python/bot/.gitignore renamed to python/.gitignore

@@ -0,0 +1,3 @@
+from .bot import Bot
+
+__all__ = ["Bot"]

python/bot/__init__.py

@@ -0,0 +1,63 @@
+import random
+from uwapi import *
+
+class Bot:
+    is_configured: bool = False
+    work_step: int = 0 # save some cpu cycles by splitting work over multiple steps
+
+    def __init__(self):
+        uw_events.on_update(self.on_update)
+
+    def attack_nearest_enemies(self):
+        own_units = [x for x in uw_world.entities().values() if x.own() and x.Unit is not None and x.proto().data.get("dps", 0) > 0]
+        if not own_units:
+            return
+        enemy_units = [x for x in uw_world.entities().values() if x.enemy() and x.Unit is not None]
+        if not enemy_units:
+            return
+        for own in own_units:
+            if len(uw_commands.orders(own.id)) == 0:
+                enemy = min(enemy_units, key=lambda x: uw_map.distance_estimate(own.pos(), x.pos()))
+                uw_commands.order(own.id, uw_commands.fight_to_entity(enemy.id))
+
+    def assign_random_recipes(self):
+        for own in uw_world.entities().values():
+            if not own.own() or own.Unit is None or own.Recipe is not None:
+                continue
+            recipes = own.proto().data.get("recipes", [])
+            if recipes:
+                recipe = random.choice(recipes)
+                uw_commands.set_recipe(own.id, recipe)
+
+    def configure(self):
+        if self.is_configured:
+            return
+        self.is_configured = True
+
+        uw_game.set_player_name("bot-cs")
+        uw_game.player_join_force(0) # create new force
+        uw_game.set_force_color(1, 0, 0)
+        # todo choose race
+
+    def on_update(self, tick: int, stepping: bool):
+        if uw_game.game_state() == GameState.Session:
+            self.configure()
+            return
+
+        if not stepping:
+            return
+
+        self.work_step += 1
+        match self.work_step % 10: # save some cpu cycles by splitting work over multiple steps
+            case 1:
+                self.attack_nearest_enemies()
+            case 5:
+                self.assign_random_recipes()
+
+    def run(self):
+        uw_game.log_info("bot-py start")
+        if not uw_game.try_reconnect():
+            uw_game.set_connect_start_gui(True)
+            if not uw_game.connect_environment():
+                uw_game.connect_new_server()
+        uw_game.log_info("bot-py done")

python/bot/bot.py

@@ -0,0 +1,6 @@
+from uwapi import UwapiLibrary
+from bot import Bot
+
+if __name__ == "__main__":
+    with UwapiLibrary():
+        Bot().run()

python/bot/main.py

@@ -0,0 +1,16 @@
+
+# THIS FILE IS GENERATED BY SCRIPT
+# DO NOT MODIFY
+
+from .interop import *
+from .admin import uw_admin
+from .commands import uw_commands
+from .entity import Entity, INVALID
+from .events import uw_events
+from .game import uw_game
+from .library import UwapiLibrary
+from .map import uw_map
+from .prototypes import uw_prototypes
+from .world import uw_world
+
+__all__ = ["uw_admin","uw_commands","Entity","INVALID","uw_events","uw_game","UwapiLibrary","uw_map","uw_prototypes","uw_world","Severity","LogCallback","ConnectionState","MyPlayer","AssistConfig","PerformanceStatistics","OrderType","OrderPriority","Order","Orders","Ids","Priority","Ping","PathState","ForeignPolicy","ChatTarget","ProtoComponent","OwnerComponent","ControllerComponent","PositionComponent","UnitState","UnitComponent","LifeComponent","ManaComponent","MoveComponent","AimComponent","RecipeComponent","RecipeStatisticsComponent","LogisticsTimestampComponent","PriorityComponent","AmountComponent","AttachmentComponent","PingComponent","PlayerState","PlayerConnectionClass","PlayerComponent","PlayerAiConfigComponent","ForceState","ForceComponent","ForceDetailsComponent","ForeignPolicyComponent","DiplomacyProposalComponent","GameState","ShootingUnit","ShootingData","ShootingArray","ExplosionData","ExplosionsArray","TaskType","MapState","MapInfo","MapStartingPosition","MapStartingPositionsArray","Tile","Cluster","ClustersDistancesQuery","ClustersDistancesResult","PrototypeType","MyForceStatistics","UnitUpgrades","Overview","OverviewExtract","UnitPathfindingQuery","UnitPathfindingResult"]

python/bot/requirements.txt

@@ -0,0 +1,73 @@
+from .interop import *
+
+INVALID: int = 0xFFFFFFFF
+
+class Admin:
+    _instance = None
+
+    def __new__(cls):
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+        return cls._instance
+
+    def get_lobby_id(self) -> int:
+        return uw_interop.uwGetLobbyId()
+
+    def get_user_id(self) -> int:
+        return uw_interop.uwGetUserId()
+
+    def get_server_port(self) -> int:
+        return uw_interop.uwGetServerPort()
+
+    def set_map_selection(self, path: str) -> None:
+        uw_interop.uwAdminSetMapSelection(path)
+
+    def start_game(self) -> None:
+        uw_interop.uwAdminStartGame()
+
+    def terminate_game(self) -> None:
+        uw_interop.uwAdminTerminateGame()
+
+    def set_game_speed(self, speed: float) -> None:
+        uw_interop.uwAdminSetGameSpeed(speed)
+
+    def set_weather_speed(self, speed: float, offset: float) -> None:
+        uw_interop.uwAdminSetWeatherSpeed(speed, offset)
+
+    def add_ai(self) -> None:
+        uw_interop.uwAdminAddAi()
+
+    def kick_player(self, player_id: int) -> None:
+        uw_interop.uwAdminKickPlayer(player_id)
+
+    def player_set_admin(self, player_id: int, admin: bool) -> None:
+        uw_interop.uwAdminPlayerSetAdmin(player_id, admin)
+
+    def player_set_name(self, player_id: int, name: str) -> None:
+        uw_interop.uwAdminPlayerSetName(player_id, name)
+
+    def player_join_force(self, player_id: int, force_id: int) -> None:
+        uw_interop.uwAdminPlayerJoinForce(player_id, force_id)
+
+    def force_join_team(self, force_id: int, team: int) -> None:
+        uw_interop.uwAdminForceJoinTeam(force_id, team)
+
+    def force_set_color(self, force_id: int, r: float, g: float, b: float) -> None:
+        uw_interop.uwAdminForceSetColor(force_id, r, g, b)
+
+    def force_set_race(self, force_id: int, race_proto: int) -> None:
+        uw_interop.uwAdminForceSetRace(force_id, race_proto)
+
+    def send_suggested_camera_focus(self, position: int) -> None:
+        uw_interop.uwAdminSendSuggestedCameraFocus(position)
+
+    def set_automatic_suggested_camera_focus(self, enabled: bool) -> None:
+        uw_interop.uwAdminSetAutomaticSuggestedCameraFocus(enabled)
+
+    def send_chat(self, msg: str, flags, id: int = INVALID) -> None:
+        uw_interop.uwAdminSendChat(msg, flags, id)
+
+    def send_ping(self, position: int, ping, id: int) -> None:
+        uw_interop.uwAdminSendPing(position, ping, id)
+
+uw_admin = Admin()