Added UseItemRevive

pogo/session.py

@@ -17,6 +17,7 @@
 from POGOProtos.Networking.Requests.Messages import RecycleInventoryItemMessage_pb2
 from POGOProtos.Networking.Requests.Messages import NicknamePokemonMessage_pb2
 from POGOProtos.Networking.Requests.Messages import UseItemPotionMessage_pb2
+from POGOProtos.Networking.Requests.Messages import UseItemReviveMessage_pb2
 
 # Load local
 import api
@@ -415,6 +416,27 @@ def useItemPotion(self, item_id, pokemon):
         # Return everything
         return self._state.itemPotion
 
+    # Use a Revive
+    def useItemRevive(self, item_id,pokemon):
+
+        # Create request
+        payload = [Request_pb2.Request(
+            request_type = RequestType_pb2.USEITEMREVIVEMESSAGE,
+            request_message = UseItemReviveMessage_pb2.UseItemReviveMessage(
+                item_id = item_id
+                pokemon_id = pokemon.id
+            ).SerializeToString()
+        )]
+
+        # Send
+        res = self.wrapAndRequest(payload, defaults=False)
+
+        # Parse
+        self._state.itemRevive.ParseFromString(res.returns[0])
+
+        # Return everything
+        return self._state.itemRevive
+
     # Evolve Pokemon
     def evolvePokemon(self, pokemon):
 

pogo/state.py

@@ -15,7 +15,7 @@
 from Networking.Responses import UseItemCaptureResponse_pb2
 from Networking.Responses import NicknamePokemonResponse_pb2
 from Networking.Responses import UseItemPotionResponse_pb2
-
+from Networking.Responses import UseItemReviveResponse_pb2
 
 class State(object):
     """Class to wrap the current state of responses"""
@@ -32,6 +32,7 @@ def __init__(self):
         self.catch = CatchPokemonResponse_pb2.CatchPokemonResponse()
         self.itemCapture = UseItemCaptureResponse_pb2.UseItemCaptureResponse()
         self.itemPotion = UseItemPotionResponse_pb2.UseItemPotionResponse()
+        self.itemRevive = UseItemReviveResponse_pb2.UseItemReviveResponse()
         self.evolve = EvolvePokemonResponse_pb2.EvolvePokemonResponse()
         self.release = ReleasePokemonResponse_pb2.ReleasePokemonResponse()
         self.recycle = RecycleInventoryItemResponse_pb2.RecycleInventoryItemResponse()