Changes in src/mlx/rpc.py [889:5f3b3d4a9957:908:fcf3c44650f1]

File:

• src/mlx/rpc.py (modified) (9 diffs)

src/mlx/rpc.py

@@ -208 +208 @@
                       "TU5"  : const.AIRCRAFT_T154,
                       "YK4"  : const.AIRCRAFT_YK40,
-                      "146"  : const.AIRCRAFT_B462 }
+                      "146"  : const.AIRCRAFT_B462,
+                      "IL6"  : const.AIRCRAFT_IL62 }
 
     # FIXME: copied from web.BookedFlight
@@ -228 +229 @@
                       const.AIRCRAFT_T154  : "TU5",
                       const.AIRCRAFT_YK40  : "YK4",
-                      const.AIRCRAFT_B462  : "146" }
+                      const.AIRCRAFT_B462  : "146",
+                      const.AIRCRAFT_IL62  : "IL6" }
 
     # FIXME: copied from web.BookedFlight
@@ -234 +236 @@
     def _decodeAircraftType(typeCode):
         """Decode the aircraft type from the given typeCode."""
-        if typeCode in BookedFlight.TYPECODE2TYPE:
-            return BookedFlight.TYPECODE2TYPE[typeCode]
+        if typeCode in const.icao2Type:
+            return const.icao2Type[typeCode]
         else:
             raise Exception("Invalid aircraft type code: '" + typeCode + "'")
@@ -275 +277 @@
     _instructions = {
         "numPassengers" : int,
-        "numCrew" : int,
+        "numCockpitCrew" : int,
+        "numCabinCrew" : int,
         "bagWeight" : int,
         "cargoWeight" : int,
@@ -307 +310 @@
         print >> f, "tail_nr=%s" % (self.tailNumber,)
         print >> f, "passenger=%d" % (self.numPassengers,)
-        print >> f, "crew=%d" % (self.numCrew,)
+        print >> f, "crew=%d" % (self.numCockpitrew + self.numCabinCrew,)
         print >> f, "bag=%d" % (self.bagWeight,)
         print >> f, "cargo=%d" % (self.cargoWeight,)
@@ -376 +379 @@
         for planeValue in value:
             self._addPlane(Plane(planeValue))
+
+#---------------------------------------------------------------------------------------
+
+class AircraftTypeInfo(RPCObject):
+    """Information about an aircraft type."""
+    # The instructions for the construction
+    _instructions = {
+        "dow" : int,
+        "dowCockpit" : int,
+        "dowCabin" : int,
+        }
+
+    def __init__(self, value):
+        """Construct the booked flight object from the given RPC result
+        value."""
+        super(AircraftTypeInfo, self).__init__(value,
+                                               AircraftTypeInfo._instructions)
+        self.aircraftType = const.icao2Type[value["typeCode"]]
 
 #---------------------------------------------------------------------------------------
@@ -498 +519 @@
             self._sessionID = reply.value["sessionID"]
 
-            types = [BookedFlight.TYPECODE2TYPE[typeCode]
+            types = [const.icao2Type[typeCode]
                      for typeCode in reply.value["typeCodes"]]
 
@@ -504 +525 @@
         else:
             return None
+
+    def getTypes(self):
+        """Get the data of the aircraft types supported by us."""
+        value = self._performCall(lambda sessionID:
+                                  self._server.getTypes(sessionID))
+
+        return [AircraftTypeInfo(t) for t in value]
 
     def getFlights(self):
@@ -599 +627 @@
         of type codes, if any."""
         typeCodes = None if types is None else \
-            [BookedFlight.TYPE2TYPECODE[type] for type in types]
+          [const.icaoCodes[type] for type in types]
 
         values = self._performCall(lambda sessionID: