Changeset 854:3e7dca86c1ed for src/mlx/rpc.py

Timestamp:

05/21/17 15:21:55 (7 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

default

hg-Phase:

(<MercurialRepository 1 'hg:/home/ivaradi/mlx/hg' '/'>, 'public')

Message:

The accepted flights can be queried and viewed (re #307)

File:

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

src/mlx/rpc.py

@@ -7 +7 @@
 import hashlib
 import datetime
+import calendar
+import sys
 
 #---------------------------------------------------------------------------------------
@@ -32 +34 @@
                     continue
 
-                value = instruction(value)
+                try:
+                    value = instruction(value)
+                except:
+                    print >> sys.stderr, "Failed to convert value '%s' of attribute '%s':" % \
+                        (value, key)
+                    import traceback
+                    traceback.print_exc()
             setattr(self, key, value)
 
@@ -131 +139 @@
 #---------------------------------------------------------------------------------------
 
+class AcceptedFlight(RPCObject):
+    """A flight that has been already accepted."""
+    # The instructions for the construction
+    @staticmethod
+    def parseTimestamp(s):
+        """Parse the given RPC timestamp."""
+        dt = datetime.datetime.strptime(s, "%Y-%m-%d %H:%M:%S")
+        return calendar.timegm(dt.utctimetuple())
+
+    _instructions = {
+        "bookedFlight" : lambda value: BookedFlight(value),
+        "numPassengers" : int,
+        "fuelUsed" : int,
+        "rating" : lambda value: float(value) if value else 0.0
+        }
+
+    def __init__(self, value):
+        """Construct the booked flight object from the given RPC result
+        value."""
+        super(AcceptedFlight, self).__init__(value, AcceptedFlight._instructions)
+        self.flightTimeStart = \
+          AcceptedFlight.parseTimestamp(self.flightDate + " " +
+                                        self.flightTimeStart)
+        self.flightTimeEnd = \
+          AcceptedFlight.parseTimestamp(self.flightDate + " " +
+                                        self.flightTimeEnd)
+        if self.flightTimeEnd<self.flightTimeStart:
+            self.flightTimeEnd += 24*60*60
+
+#---------------------------------------------------------------------------------------
+
 class Plane(rpccommon.Plane, RPCObject):
     """An airplane in the fleet."""
@@ -299 +338 @@
         return (bookedFlights, reportedFlights, rejectedFlights)
 
+    def getAcceptedFlights(self):
+        """Get the flights that are already accepted."""
+        value = self._performCall(lambda sessionID:
+                                  self._server.getAcceptedFlights(sessionID))
+        flights = []
+        for flight in value:
+            flights.append(AcceptedFlight(flight))
+        return flights
+
     def getEntryExamStatus(self):
         """Get the status of the exams needed for joining MAVA."""