Initial commit

game_server/packet/handlers/SetWarshipAvatarReq.py

@@ -0,0 +1,26 @@
+import betterproto
+from game_server.net.session import Session
+from game_server.game.enum.data_type import DataType
+from database import mongo
+from lib.proto import (
+    SetWarshipAvatarReq,
+    SetWarshipAvatarRsp,
+    GetMainDataRsp,
+    WarshipAvatarData
+)
+
+async def handle(session: Session, msg: SetWarshipAvatarReq) -> betterproto.Message:
+    await session.send(session.create_packet(
+            GetMainDataRsp(
+                retcode=0,
+                warship_avatar=WarshipAvatarData(
+                    warship_first_avatar_id=msg.first_avatar_id,
+                    warship_second_avatar_id=0
+                ),
+                type_list=[35]
+            )
+        )
+    )
+    session.player.warship_avatar.warship_first_avatar_id = msg.first_avatar_id
+    mongo.save(session,DataType.PLAYER)
+    return SetWarshipAvatarRsp(retcode=0)