implement most functions in python wrapper

Author: malytomas

python/bot/bot.py

@@ -1,3 +1,4 @@
+import random
 from uwapi import *
 
 class Bot:
@@ -8,10 +9,25 @@ def __init__(self):
         uw_events.on_update(self.on_update)
 
     def attack_nearest_enemies(self):
-        pass # todo
+        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):
-        pass # todo
+        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:

python/uwapi/__init__.py

@@ -5,12 +5,12 @@
 from .interop import *
 from .admin import uw_admin
 from .commands import uw_commands
-from .entity import Entity
+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","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"]
+__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/uwapi/admin.py

@@ -1,5 +1,7 @@
 from .interop import *
 
+INVALID: int = 0xFFFFFFFF
+
 class Admin:
     _instance = None
 
@@ -8,4 +10,64 @@ def __new__(cls):
             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()

python/uwapi/commands.py

@@ -1,5 +1,7 @@
 from .interop import *
 
+INVALID: int = 0xFFFFFFFF
+
 class Commands:
     _instance = None
 
@@ -8,4 +10,81 @@ def __new__(cls):
             cls._instance = super().__new__(cls)
         return cls._instance
 
+    def orders(self, unit_id: int) -> list[UwOrder]:
+        return uw_interop.uwOrders(unit_id).orders
+
+    def order(self, unit_id: int, order: UwOrder) -> None:
+        uw_interop.uwOrder(unit_id, order)
+
+    def stop(self) -> UwOrder:
+        o = self._default_order()
+        o.order = UwOrderTypeEnum.Stop
+        return o
+
+    def guard(self) -> UwOrder:
+        o = self._default_order()
+        o.order = UwOrderTypeEnum.Guard
+        return o
+
+    def run_to_position(self, position: int) -> UwOrder:
+        o = self._default_order()
+        o.position = position
+        o.order = UwOrderTypeEnum.Run
+        return o
+
+    def run_to_entity(self, entity_id: int) -> UwOrder:
+        o = self._default_order()
+        o.entity = entity_id
+        o.order = UwOrderTypeEnum.Run
+        return o
+
+    def fight_to_position(self, position: int) -> UwOrder:
+        o = self._default_order()
+        o.position = position
+        o.order = UwOrderTypeEnum.Fight
+        return o
+
+    def fight_to_entity(self, entity_id: int) -> UwOrder:
+        o = self._default_order()
+        o.entity = entity_id
+        o.order = UwOrderTypeEnum.Fight
+        return o
+
+    def place_construction(
+        self,
+        construction_proto: int,
+        position: int,
+        yaw: float = 0,
+        recipe_proto: int = 0,
+        priority: UwPriorityEnum = UwPriorityEnum.Normal
+    ) -> None:
+        uw_interop.uwCommandPlaceConstruction(construction_proto, position, yaw, recipe_proto, priority)
+
+    def set_recipe(self, unit_id: int, recipe_proto: int) -> None:
+        uw_interop.uwCommandSetRecipe(unit_id, recipe_proto)
+
+    def set_priority(self, unit_id: int, priority : UwPriorityEnum) -> None:
+        uw_interop.uwCommandSetPriority(unit_id, priority)
+
+    def load(self, unit_id: int, resource_proto: int) -> None:
+        uw_interop.uwCommandLoad(unit_id, resource_proto)
+
+    def unload(self, unit_id: int) -> None:
+        uw_interop.uwCommandUnload(unit_id)
+
+    def move(self, unit_id: int, position: int, yaw: float = 0) -> None:
+        uw_interop.uwCommandMove(unit_id, position, yaw)
+
+    def aim(self, unit_id: int, target_id: int) -> None:
+        uw_interop.uwCommandAim(unit_id, target_id)
+
+    def renounce_control(self, entity_id: int) -> None:
+        uw_interop.uwCommandRenounceControl(entity_id)
+
+    def self_destruct(self, entity_id: int) -> None:
+        uw_interop.uwCommandSelfDestruct(entity_id)
+
+    def _default_order(self) -> UwOrder:
+        return UwOrder(INVALID, INVALID, UwOrderTypeEnum.Nothing, UwOrderPriorityFlags.User)
+
 uw_commands = Commands()

python/uwapi/entity.py

@@ -1,10 +1,18 @@
 from dataclasses import dataclass
 from typing import Optional
 from .interop import *
+from .prototypes import uw_prototypes, Prototype
+
+INVALID: int = 0xFFFFFFFF
+
+def _make_empty_UwUnitUpgrades() -> UwUnitUpgrades:
+    return UwUnitUpgrades(0, 0, 0, 0, 0, 0, 0)
 
 @dataclass
 class Entity:
     id: int
+    fresh: bool = True
+    destroyed: bool = False
 
     Proto: Optional[UwProtoComponent] = None
     Owner: Optional[UwOwnerComponent] = None
@@ -28,3 +36,33 @@ class Entity:
     ForceDetails: Optional[UwForceDetailsComponent] = None
     ForeignPolicy: Optional[UwForeignPolicyComponent] = None
     DiplomacyProposal: Optional[UwDiplomacyProposalComponent] = None
+
+    def pos(self) -> int:
+        return self.Position.position if self.Position is not None else INVALID
+
+    def policy(self) -> UwForeignPolicyEnum:
+        from .world import uw_world
+        return uw_world.policy(self.Owner.force) if self.Owner is not None else UwForeignPolicyEnum.Nothing
+
+    def own(self) -> bool:
+        from .world import uw_world
+        return self.Owner is not None and self.Owner.force == uw_world.my_force_id()
+
+    def ally(self) -> bool:
+        return self.policy() == UwForeignPolicyEnum.Ally
+
+    def enemy(self) -> bool:
+        return self.policy() == UwForeignPolicyEnum.Enemy
+
+    def type(self) -> UwPrototypeTypeEnum:
+        return uw_prototypes.type(self.Proto.proto) if self.Proto is not None else UwPrototypeTypeEnum.Nothing
+
+    def proto(self) -> Prototype:
+        if self.Proto is not None:
+            return uw_prototypes.get(self.Proto.proto)
+        else:
+            raise Exception("entity does not have a Proto")
+
+    def unit_upgrades(self) -> UwUnitUpgrades:
+        from .world import uw_world
+        return uw_world.unit_upgrades(self.id) if self.type() == UwPrototypeTypeEnum.Unit else _make_empty_UwUnitUpgrades()

python/uwapi/events.py

@@ -1,24 +1,23 @@
+from dataclasses import field
 from typing import Callable, List
 from .interop import *
 
 class Events:
     _instance = None
+    _connection_state_listeners: List[Callable[[UwConnectionStateEnum], None]] = []
+    _game_state_listeners: List[Callable[[UwGameStateEnum], None]] = []
+    _update_listeners: List[Callable[[int, bool], None]] = []
+    _shooting_listeners: List[Callable[[UwShootingArray], None]] = []
+    _explosions_listeners: List[Callable[[UwExplosionsArray], None]] = []
+    _force_eliminated_listeners: List[Callable[[int], None]] = []
+    _chat_listeners: List[Callable[[str, int, UwChatTargetFlags], None]] = []
+    _map_state_listeners: List[Callable[[UwMapStateEnum], None]] = []
 
     def __new__(cls):
         if cls._instance is None:
             cls._instance = super().__new__(cls)
         return cls._instance
 
-    def __init__(self):
-        self._connection_state_listeners: List[Callable[[UwConnectionStateEnum], None]] = []
-        self._game_state_listeners: List[Callable[[UwGameStateEnum], None]] = []
-        self._update_listeners: List[Callable[[int, bool], None]] = []
-        self._shooting_listeners: List[Callable[[UwShootingArray], None]] = []
-        self._explosions_listeners: List[Callable[[UwExplosionsArray], None]] = []
-        self._force_eliminated_listeners: List[Callable[[int], None]] = []
-        self._chat_listeners: List[Callable[[str, int, UwChatTargetFlags], None]] = []
-        self._map_state_listeners: List[Callable[[UwMapStateEnum], None]] = []
-
     def initialize(self) -> None:
         uw_interop.uwSetExceptionCallback(self._exception_callback)
         # uw_interop.uwSetLogCallback(self._log_callback)

python/uwapi/game.py

@@ -8,23 +8,14 @@ def __new__(cls):
             cls._instance = super().__new__(cls)
         return cls._instance
 
-    def log_info(self, message: str) -> None:
-        uw_interop.uwLog(UwSeverityEnum.Info, message)
+    def log_info(self, msg: str):
+        uw_interop.uwLog(UwSeverityEnum.Info, msg)
 
-    def try_reconnect(self) -> bool:
-        return uw_interop.uwTryReconnect()
-
-    def set_connect_start_gui(self, enabled: bool, extra_cmd_params: str = "") -> None:
-        return uw_interop.uwSetConnectStartGui(enabled, extra_cmd_params)
-
-    def connect_environment(self) -> bool:
-        return uw_interop.uwConnectEnvironment()
+    def log_warning(self, msg: str):
+        uw_interop.uwLog(UwSeverityEnum.Warning, msg)
 
-    def connect_new_server(self, visibility: int = 0, name: str = "", extra_cmd_params: str = "") -> None:
-        uw_interop.uwConnectNewServer(visibility, name, extra_cmd_params)
-
-    def game_state(self) -> GameState:
-        return uw_interop.uwGameState()
+    def log_error(self, msg: str):
+        uw_interop.uwLog(UwSeverityEnum.Error, msg)
 
     def set_player_name(self, name: str) -> None:
         uw_interop.uwSetPlayerName(name)
@@ -35,4 +26,55 @@ def player_join_force(self, force_id: int) -> None:
     def set_force_color(self, r: float, g: float, b: float) -> None:
         uw_interop.uwSetForceColor(r, g, b)
 
+    def set_force_race(self, race_proto: int) -> None:
+        uw_interop.uwSetForceRace(race_proto)
+
+    def force_join_team(self, team: int) -> None:
+        uw_interop.uwForceJoinTeam(team)
+
+    def set_connect_start_gui(self, start_gui: bool, extra_params: str = "--observer 1") -> None:
+        uw_interop.uwSetConnectStartGui(start_gui, extra_params)
+
+    def connect_find_lan(self, timeout_microseconds: int = 1000000) -> bool:
+        return uw_interop.uwConnectFindLan(timeout_microseconds)
+
+    def connect_direct(self, address: str, port: int) -> None:
+        uw_interop.uwConnectDirect(address, port)
+
+    def connect_lobby_id(self, lobby_id: int) -> None:
+        uw_interop.uwConnectLobbyId(lobby_id)
+
+    def connect_environment(self) -> bool:
+        return uw_interop.uwConnectEnvironment()
+
+    def connect_new_server(self, visibility: int = 0, name: str = "", extra_params: str = "") -> None:
+        uw_interop.uwConnectNewServer(visibility, name, extra_params)
+
+    def try_reconnect(self) -> bool:
+        return uw_interop.uwTryReconnect()
+
+    def disconnect(self) -> None:
+        uw_interop.uwDisconnect()
+
+    def connection_state(self) -> UwConnectionStateEnum:
+        return uw_interop.uwConnectionState()
+
+    def game_state(self) -> UwGameStateEnum:
+        return uw_interop.uwGameState()
+
+    def map_state(self) -> UwMapStateEnum:
+        return uw_interop.uwMapState()
+
+    def performance_statistics(self) -> UwPerformanceStatistics:
+        return uw_interop.uwPerformanceStatistics()
+
+    def performance_profiling(self, enable: bool) -> None:
+        uw_interop.uwPerformanceProfiling(enable)
+
+    def profiling_event_begin(self) -> int:
+        return uw_interop.uwProfilingEventBegin()
+
+    def profiling_event_end(self, name: str, event_start_time: int) -> None:
+        uw_interop.uwProfilingEventEnd(name, event_start_time)
+
 uw_game = Game()