Add Smoke, Engine Breakdown, Made sure non-ab planes do not get ab after repairs

3 files changed, 17 insertions, 2 deletions

diff --git a/lib/Aircraft.py b/lib/Aircraft.py
index 2672621..ecbecff 100644
--- a/lib/Aircraft.py
+++ b/lib/Aircraft.py
@@ -1,4 +1,5 @@
-from lib.PacketManager.packets import FSNETCMD_AIRPLANESTATE, FSNETCMD_AIRCMD
+from lib.PacketManager.packets import FSNETCMD_AIRPLANESTATE
+from lib.PacketManager.packets.FSNETCMD_AIRCMD import FSNETCMD_AIRCMD
 from logging import debug
 
 class Aircraft:
@@ -16,6 +17,7 @@ class Aircraft:
         self.id = -1
         self.last_packet = None
         self.damage_engine_warn_sent = False
+        self.just_repaired = False
 
     def reset(self):
         """Resets the aircraft"""
@@ -29,6 +31,7 @@ class Aircraft:
         self.id = -1
         self.last_packet = None
         self.damage_engine_warn_sent = False
+        self.just_repaired = False
 
     def set_position(self, position:list):
         """Sets the position of the aircraft from the Airplane state packet"""
diff --git a/lib/PacketManager/packets/FSNETCMD_AIRCMD.py b/lib/PacketManager/packets/FSNETCMD_AIRCMD.py
index 50a9228..6770079 100644
--- a/lib/PacketManager/packets/FSNETCMD_AIRCMD.py
+++ b/lib/PacketManager/packets/FSNETCMD_AIRCMD.py
@@ -53,7 +53,7 @@ class FSNETCMD_AIRCMD: #30
         return FSNETCMD_AIRCMD.encode(aircraft_id, message, with_size)
 
     @staticmethod
-    def set_command(aircraft_id:int, command:str, value:int, with_size:bool=False):
+    def set_command(aircraft_id:int, command:str, value:int|str, with_size:bool=False):
         """
         This will set the command of the aircraft, useful if you want to
         set the engine power, fuel, etc.
diff --git a/lib/PacketManager/packets/FSNETCMD_AIRPLANESTATE.py b/lib/PacketManager/packets/FSNETCMD_AIRPLANESTATE.py
index 2282367..3ebf8d6 100644
--- a/lib/PacketManager/packets/FSNETCMD_AIRPLANESTATE.py
+++ b/lib/PacketManager/packets/FSNETCMD_AIRPLANESTATE.py
@@ -299,3 +299,15 @@ class FSNETCMD_AIRPLANESTATE: #11
         if with_size:
             return pack("I",len(buffer))+buffer
         return buffer
+
+    def __str__(self):
+        return f"Player ID : {self.player_id}; Remote Time : {self.remote_time}; \
+                Position : {self.position}; Attitude : {self.atti}; Velocity : {self.velocity}; \
+                Attitude Velocity : {self.atti_velocity}; Smoke Oil : {self.smoke_oil}; Fuel : {self.fuel}; \
+                Payload : {self.payload}; Flight State : {self.flight_state}; VGW : {self.vgw}; \
+                Spoiler : {self.spoiler}; Landing Gear : {self.landing_gear}; Flap : {self.flap}; \
+                Brake : {self.brake}; Flags : {self.flags}; Gun Ammo : {self.gun_ammo}; \
+                Rocket Ammo : {self.rocket_ammo}; AAM : {self.aam}; AGM : {self.agm}; Bomb : {self.bomb}; \
+                Life : {self.life}; G Value : {self.g_value}; Throttle : {self.throttle}; Elev : {self.elev}; \
+                Ail : {self.ail}; Rud : {self.rud}; Trim : {self.trim}; Thrust Vector : {self.thrust_vector}; \
+                Bomb Bay Info : {self.bomb_bay_info}"