[919] | 1 | from . import const
|
---|
| 2 | from . import rpccommon
|
---|
[745] | 3 |
|
---|
[955] | 4 | from .common import MAVA_BASE_URL, fixUnpickled
|
---|
[745] | 5 |
|
---|
| 6 | import jsonrpclib
|
---|
| 7 | import hashlib
|
---|
| 8 | import datetime
|
---|
[854] | 9 | import calendar
|
---|
| 10 | import sys
|
---|
[1039] | 11 | import ssl
|
---|
| 12 | import certifi
|
---|
[745] | 13 |
|
---|
| 14 | #---------------------------------------------------------------------------------------
|
---|
| 15 |
|
---|
| 16 | class RPCObject(object):
|
---|
| 17 | """Base class for objects read from RPC calls.
|
---|
| 18 |
|
---|
| 19 | It is possible to construct it from a dictionary."""
|
---|
| 20 | def __init__(self, value, instructions = {}):
|
---|
| 21 | """Construct the object.
|
---|
| 22 |
|
---|
| 23 | value is the dictionary returned by the call.
|
---|
| 24 |
|
---|
| 25 | info is a mapping from names to 'instructions' on what to do with the
|
---|
| 26 | corresponding values. If the instruction is None, it will be ignored.
|
---|
| 27 | If the instruction is a function, the value will be passed to it and
|
---|
| 28 | the return value will be stored in the object.
|
---|
| 29 |
|
---|
| 30 | For all other names, the value will be stored as the same-named
|
---|
| 31 | attribute."""
|
---|
[919] | 32 | for (key, value) in value.items():
|
---|
[745] | 33 | if key in instructions:
|
---|
| 34 | instruction = instructions[key]
|
---|
| 35 | if instruction is None:
|
---|
| 36 | continue
|
---|
| 37 |
|
---|
[854] | 38 | try:
|
---|
| 39 | value = instruction(value)
|
---|
| 40 | except:
|
---|
[919] | 41 | print("Failed to convert value '%s' of attribute '%s':" % \
|
---|
| 42 | (value, key), file=sys.stderr)
|
---|
[854] | 43 | import traceback
|
---|
| 44 | traceback.print_exc()
|
---|
[745] | 45 | setattr(self, key, value)
|
---|
| 46 |
|
---|
| 47 | #---------------------------------------------------------------------------------------
|
---|
| 48 |
|
---|
| 49 | class Reply(RPCObject):
|
---|
| 50 | """The generic reply structure."""
|
---|
| 51 |
|
---|
| 52 | #---------------------------------------------------------------------------------------
|
---|
| 53 |
|
---|
[858] | 54 | class ScheduledFlight(RPCObject):
|
---|
| 55 | """A scheduled flight in the time table."""
|
---|
| 56 | # The instructions for the construction
|
---|
| 57 | # Type: normal flight
|
---|
| 58 | TYPE_NORMAL = 0
|
---|
| 59 |
|
---|
| 60 | # Type: VIP flight
|
---|
| 61 | TYPE_VIP = 1
|
---|
| 62 |
|
---|
| 63 | _instructions = {
|
---|
| 64 | "id" : int,
|
---|
| 65 | "pairID": int,
|
---|
| 66 | "typeCode": lambda value: BookedFlight._decodeAircraftType(value),
|
---|
| 67 | "departureTime": lambda value: ScheduledFlight._decodeTime(value),
|
---|
| 68 | "arrivalTime": lambda value: ScheduledFlight._decodeTime(value),
|
---|
| 69 | "duration": lambda value: ScheduledFlight._decodeDuration(value),
|
---|
[859] | 70 | "spec": int,
|
---|
| 71 | "validFrom": lambda value: ScheduledFlight._decodeDate(value),
|
---|
| 72 | "validTo": lambda value: ScheduledFlight._decodeDate(value),
|
---|
| 73 | "date": lambda value: ScheduledFlight._decodeDate(value)
|
---|
[858] | 74 | }
|
---|
| 75 |
|
---|
| 76 | @staticmethod
|
---|
| 77 | def _decodeTime(value):
|
---|
| 78 | """Decode the given value as a time value."""
|
---|
| 79 | return datetime.datetime.strptime(value, "%H:%M:%S").time()
|
---|
| 80 |
|
---|
| 81 | @staticmethod
|
---|
[859] | 82 | def _decodeDate(value):
|
---|
| 83 | """Decode the given value as a date value."""
|
---|
| 84 | if not value or value=="0000-00-00":
|
---|
| 85 | return const.defaultDate
|
---|
| 86 | else:
|
---|
| 87 | return datetime.datetime.strptime(value, "%Y-%m-%d").date()
|
---|
| 88 |
|
---|
| 89 | @staticmethod
|
---|
[858] | 90 | def _decodeDuration(value):
|
---|
| 91 | """Decode the given value as a duration.
|
---|
| 92 |
|
---|
| 93 | A number of seconds will be returned."""
|
---|
| 94 | t = datetime.datetime.strptime(value, "%H:%M:%S")
|
---|
| 95 | return (t.hour*60 + t.minute) * 60 + t.second
|
---|
| 96 |
|
---|
| 97 | def __init__(self, value):
|
---|
| 98 | """Construct the scheduled flight object from the given JSON value."""
|
---|
| 99 | super(ScheduledFlight, self).__init__(value,
|
---|
| 100 | ScheduledFlight._instructions)
|
---|
| 101 | self.aircraftType = self.typeCode
|
---|
| 102 | del self.typeCode
|
---|
| 103 |
|
---|
[859] | 104 | self.type = ScheduledFlight.TYPE_VIP if self.spec==1 \
|
---|
| 105 | else ScheduledFlight.TYPE_NORMAL
|
---|
| 106 | del self.spec
|
---|
| 107 |
|
---|
| 108 | def compareBy(self, other, name):
|
---|
| 109 | """Compare this flight with the other one according to the given
|
---|
| 110 | attribute name."""
|
---|
| 111 | if name=="callsign":
|
---|
| 112 | try:
|
---|
| 113 | cs1 = int(self.callsign[2:])
|
---|
| 114 | cs2 = int(other.callsign[2:])
|
---|
[923] | 115 | return 0 if cs1==cs2 else -1 if cs1<cs2 else 1
|
---|
[859] | 116 | except:
|
---|
[923] | 117 | return 0 if self.callsign==other.callsign \
|
---|
| 118 | else -1 if self.callsign<other.callsign else 1
|
---|
[859] | 119 | else:
|
---|
[923] | 120 | v1 = getattr(self, name)
|
---|
| 121 | v2 = getattr(other, name)
|
---|
| 122 | return 0 if v1==v2 else -1 if v1<v2 else 1
|
---|
[859] | 123 |
|
---|
[858] | 124 | def __repr__(self):
|
---|
| 125 | return "ScheduledFlight<%d, %d, %s, %s (%s) - %s (%s) -> %d, %d>" % \
|
---|
| 126 | (self.id, self.pairID, BookedFlight.TYPE2TYPECODE[self.aircraftType],
|
---|
| 127 | self.departureICAO, str(self.departureTime),
|
---|
| 128 | self.arrivalICAO, str(self.arrivalTime),
|
---|
[859] | 129 | self.duration, self.type)
|
---|
[858] | 130 |
|
---|
| 131 | #---------------------------------------------------------------------------------------
|
---|
| 132 |
|
---|
| 133 | class ScheduledFlightPair(object):
|
---|
| 134 | """A pair of scheduled flights.
|
---|
| 135 |
|
---|
| 136 | Occasionally, one of the flights may be missing."""
|
---|
| 137 | @staticmethod
|
---|
[859] | 138 | def scheduledFlights2Pairs(scheduledFlights, date):
|
---|
[858] | 139 | """Convert the given list of scheduled flights into a list of flight
|
---|
| 140 | pairs."""
|
---|
[859] | 141 | weekday = str(date.weekday()+1)
|
---|
| 142 |
|
---|
[858] | 143 | flights = {}
|
---|
[859] | 144 | weekdayFlights = {}
|
---|
[858] | 145 | for flight in scheduledFlights:
|
---|
| 146 | flights[flight.id] = flight
|
---|
[859] | 147 | if (flight.type==ScheduledFlight.TYPE_NORMAL and
|
---|
| 148 | flight.arrivalICAO!="LHBP" and weekday in flight.days and
|
---|
| 149 | flight.validFrom<=date and flight.validTo>=date) or \
|
---|
| 150 | flight.type==ScheduledFlight.TYPE_VIP:
|
---|
| 151 | weekdayFlights[flight.id] = flight
|
---|
[858] | 152 |
|
---|
| 153 | flightPairs = []
|
---|
| 154 |
|
---|
[859] | 155 | while weekdayFlights:
|
---|
| 156 | (id, flight) = weekdayFlights.popitem()
|
---|
| 157 | if flight.type==ScheduledFlight.TYPE_NORMAL:
|
---|
| 158 | pairID = flight.pairID
|
---|
| 159 | if pairID in flights:
|
---|
| 160 | pairFlight = flights[pairID]
|
---|
| 161 | if flight.departureICAO=="LHBP" or \
|
---|
| 162 | (pairFlight.departureICAO!="LHBP" and
|
---|
| 163 | flight.callsign<pairFlight.callsign):
|
---|
| 164 | flightPairs.append(ScheduledFlightPair(flight, pairFlight))
|
---|
| 165 | else:
|
---|
| 166 | flightPairs.append(ScheduledFlightPair(pairFlight, flight))
|
---|
| 167 | del flights[pairID]
|
---|
| 168 | if pairID in weekdayFlights:
|
---|
| 169 | del weekdayFlights[pairID]
|
---|
| 170 | elif flight.type==ScheduledFlight.TYPE_VIP:
|
---|
[858] | 171 | flightPairs.append(ScheduledFlightPair(flight))
|
---|
| 172 |
|
---|
[923] | 173 | flightPairs.sort(key = lambda pair: pair.flight0.date)
|
---|
[859] | 174 |
|
---|
[858] | 175 | return flightPairs
|
---|
| 176 |
|
---|
| 177 | def __init__(self, flight0, flight1 = None):
|
---|
| 178 | """Construct the pair with the given flights."""
|
---|
| 179 | self.flight0 = flight0
|
---|
| 180 | self.flight1 = flight1
|
---|
| 181 |
|
---|
[859] | 182 | def compareBy(self, other, name):
|
---|
| 183 | """Compare this flight pair with the other one according to the given
|
---|
| 184 | attribute name, considering the first flights."""
|
---|
| 185 | return self.flight0.compareBy(other.flight0, name)
|
---|
| 186 |
|
---|
| 187 | def __repr__(self):
|
---|
| 188 | return "ScheduledFlightPair<%s, %s, %s>" % \
|
---|
| 189 | (self.flight0.callsign, self.flight0.departureICAO,
|
---|
| 190 | self.flight0.arrivalICAO)
|
---|
| 191 |
|
---|
[858] | 192 | #---------------------------------------------------------------------------------------
|
---|
| 193 |
|
---|
[745] | 194 | class BookedFlight(RPCObject):
|
---|
| 195 | """A booked flight."""
|
---|
[1032] | 196 | TYPECODE2TYPE = { "B736" : const.AIRCRAFT_B736,
|
---|
[1044] | 197 | "736" : const.AIRCRAFT_B736,
|
---|
[1032] | 198 | "B737" : const.AIRCRAFT_B737,
|
---|
[1044] | 199 | "73G" : const.AIRCRAFT_B737,
|
---|
[1032] | 200 | "B738" : const.AIRCRAFT_B738,
|
---|
[1044] | 201 | "738" : const.AIRCRAFT_B738,
|
---|
[1032] | 202 | "B73H" : const.AIRCRAFT_B738C,
|
---|
[1044] | 203 | "73H" : const.AIRCRAFT_B738C,
|
---|
[1032] | 204 | "B732" : const.AIRCRAFT_B732,
|
---|
[1044] | 205 | "732" : const.AIRCRAFT_B732,
|
---|
[1032] | 206 | "B733" : const.AIRCRAFT_B733,
|
---|
[1044] | 207 | "733" : const.AIRCRAFT_B733,
|
---|
[1032] | 208 | "B734" : const.AIRCRAFT_B734,
|
---|
[1044] | 209 | "734" : const.AIRCRAFT_B734,
|
---|
[1032] | 210 | "B735" : const.AIRCRAFT_B735,
|
---|
[1044] | 211 | "735" : const.AIRCRAFT_B735,
|
---|
[1032] | 212 | "DH8D" : const.AIRCRAFT_DH8D,
|
---|
[1044] | 213 | "DH4" : const.AIRCRAFT_DH8D,
|
---|
[1032] | 214 | "B762" : const.AIRCRAFT_B762,
|
---|
[1044] | 215 | "762" : const.AIRCRAFT_B762,
|
---|
[1032] | 216 | "B763" : const.AIRCRAFT_B763,
|
---|
[1044] | 217 | "763" : const.AIRCRAFT_B763,
|
---|
[1032] | 218 | "CRJ2" : const.AIRCRAFT_CRJ2,
|
---|
[1044] | 219 | "CR2" : const.AIRCRAFT_CRJ2,
|
---|
| 220 | "F70" : const.AIRCRAFT_F70,
|
---|
| 221 | "LI2" : const.AIRCRAFT_DC3,
|
---|
[1032] | 222 | "T134" : const.AIRCRAFT_T134,
|
---|
[1044] | 223 | "TU3" : const.AIRCRAFT_T134,
|
---|
[1032] | 224 | "T154" : const.AIRCRAFT_T154,
|
---|
[1044] | 225 | "TU5" : const.AIRCRAFT_T154,
|
---|
[1032] | 226 | "YK40" : const.AIRCRAFT_YK40,
|
---|
[1044] | 227 | "YK4" : const.AIRCRAFT_YK40,
|
---|
[1032] | 228 | "B462" : const.AIRCRAFT_B462,
|
---|
[1044] | 229 | "146" : const.AIRCRAFT_B462,
|
---|
[1032] | 230 | "IL62" : const.AIRCRAFT_IL62 }
|
---|
[745] | 231 |
|
---|
[1032] | 232 | TYPE2TYPECODE = { const.AIRCRAFT_B736 : "B736",
|
---|
| 233 | const.AIRCRAFT_B737 : "B737",
|
---|
| 234 | const.AIRCRAFT_B738 : "B738",
|
---|
| 235 | const.AIRCRAFT_B738C : "B73H",
|
---|
| 236 | const.AIRCRAFT_B732 : "B732",
|
---|
| 237 | const.AIRCRAFT_B733 : "B733",
|
---|
| 238 | const.AIRCRAFT_B734 : "B734",
|
---|
| 239 | const.AIRCRAFT_B735 : "B735",
|
---|
| 240 | const.AIRCRAFT_DH8D : "DH8D",
|
---|
| 241 | const.AIRCRAFT_B762 : "B762",
|
---|
| 242 | const.AIRCRAFT_B763 : "B763",
|
---|
| 243 | const.AIRCRAFT_CRJ2 : "CRJ2",
|
---|
[858] | 244 | const.AIRCRAFT_F70 : "F70",
|
---|
| 245 | const.AIRCRAFT_DC3 : "LI2",
|
---|
[1032] | 246 | const.AIRCRAFT_T134 : "T134",
|
---|
| 247 | const.AIRCRAFT_T154 : "T155",
|
---|
| 248 | const.AIRCRAFT_YK40 : "YK40",
|
---|
| 249 | const.AIRCRAFT_B462 : "B462",
|
---|
| 250 | const.AIRCRAFT_IL62 : "IL62" }
|
---|
[858] | 251 |
|
---|
[1044] | 252 | checkFlightTypes = [ const.AIRCRAFT_B736, const.AIRCRAFT_B737,
|
---|
| 253 | const.AIRCRAFT_B738, const.AIRCRAFT_DH8D ]
|
---|
| 254 |
|
---|
[745] | 255 | @staticmethod
|
---|
| 256 | def _decodeAircraftType(typeCode):
|
---|
| 257 | """Decode the aircraft type from the given typeCode."""
|
---|
| 258 | if typeCode in BookedFlight.TYPECODE2TYPE:
|
---|
| 259 | return BookedFlight.TYPECODE2TYPE[typeCode]
|
---|
| 260 | else:
|
---|
| 261 | raise Exception("Invalid aircraft type code: '" + typeCode + "'")
|
---|
| 262 |
|
---|
[809] | 263 | @staticmethod
|
---|
| 264 | def _decodeStatus(status):
|
---|
| 265 | """Decode the status from the status string."""
|
---|
| 266 | if status=="booked":
|
---|
| 267 | return BookedFlight.STATUS_BOOKED
|
---|
| 268 | elif status=="reported":
|
---|
| 269 | return BookedFlight.STATUS_REPORTED
|
---|
| 270 | elif status=="accepted":
|
---|
| 271 | return BookedFlight.STATUS_ACCEPTED
|
---|
| 272 | elif status=="rejected":
|
---|
| 273 | return BookedFlight.STATUS_REJECTED
|
---|
| 274 | else:
|
---|
| 275 | raise Exception("Invalid flight status code: '" + status + "'")
|
---|
| 276 |
|
---|
[1034] | 277 | @staticmethod
|
---|
| 278 | def _convertFlightType(ft):
|
---|
| 279 | """Convert the in-database flight-type to one of our constants."""
|
---|
| 280 | ft = int(ft)
|
---|
| 281 | if ft==0:
|
---|
| 282 | return const.FLIGHTTYPE_SCHEDULED
|
---|
| 283 | elif ft==1:
|
---|
| 284 | return const.FLIGHTTYPE_VIP
|
---|
| 285 | elif ft==2:
|
---|
| 286 | return const.FLIGHTTYPE_CHARTER
|
---|
| 287 | else:
|
---|
| 288 | return const.FLIGHTTYPE_SCHEDULED
|
---|
| 289 |
|
---|
[745] | 290 | @staticmethod
|
---|
| 291 | def getDateTime(date, time):
|
---|
| 292 | """Get a datetime object from the given textual date and time."""
|
---|
| 293 | return datetime.datetime.strptime(date + " " + time,
|
---|
| 294 | "%Y-%m-%d %H:%M:%S")
|
---|
| 295 |
|
---|
[809] | 296 | STATUS_BOOKED = 1
|
---|
| 297 |
|
---|
| 298 | STATUS_REPORTED = 2
|
---|
| 299 |
|
---|
| 300 | STATUS_ACCEPTED = 3
|
---|
| 301 |
|
---|
| 302 | STATUS_REJECTED = 4
|
---|
| 303 |
|
---|
[1044] | 304 | @staticmethod
|
---|
| 305 | def forCheckFlight(aircraftType):
|
---|
| 306 | """Create a booked flight for a check flight with the given aircraft
|
---|
| 307 | type."""
|
---|
| 308 | flight = BookedFlight()
|
---|
| 309 |
|
---|
| 310 | flight.departureICAO = "LHBP"
|
---|
| 311 | flight.arrivalICAO = "LHBP"
|
---|
| 312 |
|
---|
| 313 | flight.aircraftType = aircraftType
|
---|
| 314 | flight.aircraftTypeName = BookedFlight.TYPE2TYPECODE[aircraftType]
|
---|
| 315 |
|
---|
| 316 | # FIXME: perhaps find one for the type
|
---|
| 317 | flight.tailNumber = "HA-CHK"
|
---|
| 318 | flight.callsign = "HA-CHK"
|
---|
| 319 |
|
---|
| 320 | flight.numPassengers = 0
|
---|
| 321 | flight.numChildren = 0
|
---|
| 322 | flight.numInfants = 0
|
---|
| 323 | flight.numCabinCrew = 0
|
---|
| 324 | flight.bagWeight = 0
|
---|
| 325 | flight.cargoWeight = 0
|
---|
| 326 | flight.mailWeight = 0
|
---|
| 327 | flight.route = "DCT"
|
---|
| 328 |
|
---|
| 329 | t = datetime.datetime.now() + datetime.timedelta(minutes = 20)
|
---|
| 330 | flight.departureTime = datetime.datetime(t.year, t.month, t.day,
|
---|
| 331 | t.hour, t.minute)
|
---|
| 332 | t = flight.departureTime + datetime.timedelta(minutes = 30)
|
---|
| 333 | flight.arrivalTime = datetime.datetime(t.year, t.month, t.day,
|
---|
| 334 | t.hour, t.minute)
|
---|
| 335 |
|
---|
| 336 | return flight
|
---|
| 337 |
|
---|
[745] | 338 | # The instructions for the construction
|
---|
| 339 | _instructions = {
|
---|
| 340 | "numPassengers" : int,
|
---|
[1033] | 341 | "numChildren" : int,
|
---|
| 342 | "numInfants" : int,
|
---|
| 343 | "numCabinCrew" : int,
|
---|
| 344 | "dowNumCabinCrew" : int,
|
---|
| 345 | "numCockpitCrew" : int,
|
---|
[745] | 346 | "bagWeight" : int,
|
---|
| 347 | "cargoWeight" : int,
|
---|
| 348 | "mailWeight" : int,
|
---|
[1034] | 349 | "flightType" : lambda value: BookedFlight._convertFlightType(value),
|
---|
[1033] | 350 | "dow": int,
|
---|
| 351 | "maxPassengers": int,
|
---|
[809] | 352 | "aircraftType" : lambda value: BookedFlight._decodeAircraftType(value),
|
---|
| 353 | "status" : lambda value: BookedFlight._decodeStatus(value)
|
---|
[745] | 354 | }
|
---|
| 355 |
|
---|
[1044] | 356 | def __init__(self, value = None, id = None):
|
---|
[745] | 357 | """Construct the booked flight object from the given RPC result
|
---|
| 358 | value."""
|
---|
[820] | 359 | self.status = BookedFlight.STATUS_BOOKED
|
---|
[1044] | 360 | if value is None:
|
---|
| 361 | self.id = id
|
---|
| 362 | else:
|
---|
| 363 | super(BookedFlight, self).__init__(value, BookedFlight._instructions)
|
---|
| 364 | self.departureTime = \
|
---|
| 365 | BookedFlight.getDateTime(self.date, self.departureTime)
|
---|
| 366 | self.arrivalTime = \
|
---|
| 367 | BookedFlight.getDateTime(self.date, self.arrivalTime)
|
---|
| 368 | if self.arrivalTime<self.departureTime:
|
---|
| 369 | self.arrivalTime += datetime.timedelta(days = 1)
|
---|
| 370 |
|
---|
[1046] | 371 | @property
|
---|
| 372 | def payload(self):
|
---|
| 373 | """Get the default payload of the flight."""
|
---|
| 374 | payload= (self.numCabinCrew - self.dowNumCabinCrew) * \
|
---|
| 375 | const.WEIGHT_CABIN_CREW
|
---|
| 376 | payload += self.numPassengers * \
|
---|
| 377 | (const.WEIGHT_PASSENGER_CHARTER
|
---|
| 378 | if self.flightType==const.FLIGHTTYPE_CHARTER
|
---|
| 379 | else const.WEIGHT_PASSENGER)
|
---|
| 380 | payload += self.numChildren * const.WEIGHT_CHILD
|
---|
| 381 | payload += self.numInfants * const.WEIGHT_INFANT
|
---|
| 382 | payload += self.bagWeight
|
---|
| 383 | payload += self.cargoWeight
|
---|
| 384 | payload += self.mailWeight
|
---|
| 385 | return payload
|
---|
| 386 |
|
---|
[1044] | 387 | def readFromFile(self, f, fleet):
|
---|
| 388 | """Read the data of the flight from a file via the given file
|
---|
| 389 | object."""
|
---|
| 390 | date = None
|
---|
| 391 | departureTime = None
|
---|
| 392 | arrivalTime = None
|
---|
| 393 |
|
---|
| 394 | line = f.readline()
|
---|
| 395 | lineNumber = 0
|
---|
| 396 | self.numChildren = 0
|
---|
| 397 | self.numInfants = 0
|
---|
| 398 | self.numCockpitCrew = 2
|
---|
| 399 | self.flightType = const.FLIGHTTYPE_SCHEDULED
|
---|
| 400 | while line:
|
---|
| 401 | lineNumber += 1
|
---|
| 402 | line = line.strip()
|
---|
| 403 |
|
---|
| 404 | hashIndex = line.find("#")
|
---|
| 405 | if hashIndex>=0: line = line[:hashIndex]
|
---|
| 406 | if line:
|
---|
| 407 | equalIndex = line.find("=")
|
---|
| 408 | lineOK = equalIndex>0
|
---|
| 409 |
|
---|
| 410 | if lineOK:
|
---|
| 411 | key = line[:equalIndex].strip()
|
---|
| 412 | value = line[equalIndex+1:].strip().replace("\:", ":")
|
---|
| 413 |
|
---|
| 414 | lineOK = key and value
|
---|
| 415 |
|
---|
| 416 | if lineOK:
|
---|
| 417 | if key=="callsign": self.callsign = value
|
---|
| 418 | elif key=="date": date = value
|
---|
| 419 | elif key=="dep_airport": self.departureICAO = value
|
---|
| 420 | elif key=="dest_airport": self.arrivalICAO = value
|
---|
| 421 | elif key=="planecode": self.aircraftType = \
|
---|
| 422 | BookedFlight._decodeAircraftType(value)
|
---|
| 423 | elif key=="planetype": self.aircraftTypeName = value
|
---|
| 424 | elif key=="tail_nr": self.tailNumber = value
|
---|
| 425 | elif key=="passenger": self.numPassengers = int(value)
|
---|
| 426 | elif key=="child": self.numChildren = int(value)
|
---|
| 427 | elif key=="infant": self.numInfants = int(value)
|
---|
| 428 | elif key=="crew": self.numCabinCrew = int(value) - 2
|
---|
| 429 | elif key=="cabin_crew": self.numCabinCrew = int(value)
|
---|
| 430 | elif key=="cockpit_crew": self.numCockpitCrew = int(value)
|
---|
| 431 | elif key=="bag": self.bagWeight = int(value)
|
---|
| 432 | elif key=="cargo": self.cargoWeight = int(value)
|
---|
| 433 | elif key=="mail": self.mailWeight = int(value)
|
---|
| 434 | elif key=="flight_route": self.route = value
|
---|
| 435 | elif key=="departure_time": departureTime = value
|
---|
| 436 | elif key=="arrival_time": arrivalTime = value
|
---|
| 437 | elif key=="foglalas_id":
|
---|
| 438 | self.id = None if value=="0" else value
|
---|
| 439 | elif key=="flight_type": self.flightType = int(value)
|
---|
| 440 | else: lineOK = False
|
---|
| 441 |
|
---|
| 442 | if not lineOK:
|
---|
| 443 | print("web.BookedFlight.readFromFile: line %d is invalid" % \
|
---|
| 444 | (lineNumber,))
|
---|
| 445 |
|
---|
| 446 | line = f.readline()
|
---|
| 447 |
|
---|
[1101] | 448 | self.date = date
|
---|
[1044] | 449 | if date is not None:
|
---|
| 450 | if departureTime is not None:
|
---|
| 451 | self.departureTime = BookedFlight.getDateTime(date,
|
---|
| 452 | departureTime)
|
---|
| 453 | if arrivalTime is not None:
|
---|
| 454 | self.arrivalTime = BookedFlight.getDateTime(date,
|
---|
| 455 | arrivalTime)
|
---|
| 456 |
|
---|
| 457 | d = dir(self)
|
---|
| 458 | for attribute in ["callsign", "departureICAO", "arrivalICAO",
|
---|
| 459 | "aircraftType", "tailNumber",
|
---|
| 460 | "numPassengers", "numCockpitCrew", "numCabinCrew",
|
---|
| 461 | "bagWeight", "cargoWeight", "mailWeight",
|
---|
| 462 | "route", "departureTime", "arrivalTime"]:
|
---|
| 463 | if attribute not in d:
|
---|
| 464 | raise Exception("Attribute %s could not be read" % (attribute,))
|
---|
| 465 |
|
---|
| 466 | if "aircraftTypeName" not in d:
|
---|
| 467 | self.aircraftTypeName = \
|
---|
| 468 | BookedFlight.TYPE2TYPECODE[self.aircraftType]
|
---|
| 469 |
|
---|
| 470 | plane = fleet[self.tailNumber]
|
---|
| 471 | if plane is None:
|
---|
| 472 | self.dow = 0
|
---|
| 473 | self.maxPassengers = 0
|
---|
| 474 | self.dowNumCabinCrew = 0
|
---|
| 475 | else:
|
---|
| 476 | self.dow = plane.dow
|
---|
| 477 | self.maxPassengers = plane.maxPassengers
|
---|
| 478 | self.dowNumCabinCrew = plane.dowNumCabinCrew
|
---|
| 479 |
|
---|
| 480 | def setupFromPIREPData(self, pirepData):
|
---|
| 481 | """Setup the booked flight from the given PIREP data."""
|
---|
| 482 | bookedFlightData = pirepData["bookedFlight"]
|
---|
| 483 |
|
---|
| 484 | self.callsign = bookedFlightData["callsign"]
|
---|
| 485 |
|
---|
| 486 | date = bookedFlightData["date"]
|
---|
| 487 |
|
---|
| 488 | departureTime = bookedFlightData["departureTime"]
|
---|
| 489 | self.departureTime = BookedFlight.getDateTime(date, departureTime)
|
---|
| 490 |
|
---|
| 491 | arrivalTime = bookedFlightData["arrivalTime"]
|
---|
| 492 | self.arrivalTime = BookedFlight.getDateTime(date, arrivalTime)
|
---|
[745] | 493 | if self.arrivalTime<self.departureTime:
|
---|
| 494 | self.arrivalTime += datetime.timedelta(days = 1)
|
---|
| 495 |
|
---|
[1044] | 496 | self.departureICAO = bookedFlightData["departureICAO"]
|
---|
| 497 | self.arrivalICAO = bookedFlightData["arrivalICAO"]
|
---|
| 498 |
|
---|
| 499 | self.aircraftType = \
|
---|
| 500 | BookedFlight._decodeAircraftType(bookedFlightData["aircraftType"])
|
---|
| 501 | self.tailNumber = bookedFlightData["tailNumber"]
|
---|
| 502 | self.numPassengers = int(bookedFlightData["numPassengers"])
|
---|
| 503 | self.numChildren = int(bookedFlightData["numChildren"])
|
---|
| 504 | self.numInfants = int(bookedFlightData["numInfants"])
|
---|
| 505 | self.maxPassengers = int(bookedFlightData["maxPassengers"])
|
---|
| 506 | self.numCockpitCrew = int(bookedFlightData["numCockpitCrew"])
|
---|
| 507 | self.numCabinCrew = int(bookedFlightData["numCabinCrew"])
|
---|
| 508 | self.bagWeight = int(bookedFlightData["bagWeight"])
|
---|
| 509 | self.cargoWeight = int(bookedFlightData["cargoWeight"])
|
---|
| 510 | self.mailWeight = int(bookedFlightData["mailWeight"])
|
---|
| 511 | self.route = bookedFlightData["route"]
|
---|
| 512 | self.flightType = BookedFlight._convertFlightType(bookedFlightData["flightType"])
|
---|
| 513 |
|
---|
[889] | 514 | def writeIntoFile(self, f):
|
---|
| 515 | """Write the flight into a file."""
|
---|
[919] | 516 | print("callsign=%s" % (self.callsign,), file=f)
|
---|
[889] | 517 | date = self.departureTime.date()
|
---|
[919] | 518 | print("date=%04d-%02d-%0d" % (date.year, date.month, date.day), file=f)
|
---|
| 519 | print("dep_airport=%s" % (self.departureICAO,), file=f)
|
---|
| 520 | print("dest_airport=%s" % (self.arrivalICAO,), file=f)
|
---|
| 521 | print("planecode=%s" % \
|
---|
| 522 | (BookedFlight.TYPE2TYPECODE[self.aircraftType],), file=f)
|
---|
| 523 | print("planetype=%s" % (self.aircraftTypeName,), file=f)
|
---|
| 524 | print("tail_nr=%s" % (self.tailNumber,), file=f)
|
---|
| 525 | print("passenger=%d" % (self.numPassengers,), file=f)
|
---|
[1044] | 526 | print("child=%d" % (self.numChildren,), file=f)
|
---|
| 527 | print("infant=%d" % (self.numInfants,), file=f)
|
---|
| 528 | print("cockpit_crew=%d" % (self.numCockpitCrew,), file=f)
|
---|
| 529 | print("cabin_crew=%d" % (self.numCabinCrew,), file=f)
|
---|
[919] | 530 | print("bag=%d" % (self.bagWeight,), file=f)
|
---|
| 531 | print("cargo=%d" % (self.cargoWeight,), file=f)
|
---|
| 532 | print("mail=%d" % (self.mailWeight,), file=f)
|
---|
| 533 | print("flight_route=%s" % (self.route,), file=f)
|
---|
[889] | 534 | departureTime = self.departureTime
|
---|
[919] | 535 | print("departure_time=%02d\\:%02d\\:%02d" % \
|
---|
| 536 | (departureTime.hour, departureTime.minute, departureTime.second), file=f)
|
---|
[889] | 537 | arrivalTime = self.arrivalTime
|
---|
[919] | 538 | print("arrival_time=%02d\\:%02d\\:%02d" % \
|
---|
| 539 | (arrivalTime.hour, arrivalTime.minute, arrivalTime.second), file=f)
|
---|
| 540 | print("foglalas_id=%s" % ("0" if self.id is None else self.id,), file=f)
|
---|
[1044] | 541 | print("flight_type=%d" % (self.flightType,))
|
---|
[889] | 542 |
|
---|
[955] | 543 | def __setstate__(self, state):
|
---|
| 544 | """Set the state from the given unpickled dictionary."""
|
---|
| 545 | self.__dict__.update(fixUnpickled(state))
|
---|
| 546 |
|
---|
[1044] | 547 | def __repr__(self):
|
---|
| 548 | """Get a representation of the flight."""
|
---|
| 549 | s = "<Flight: %s-%s, %s, %s-%s," % (self.departureICAO,
|
---|
| 550 | self.arrivalICAO,
|
---|
| 551 | self.route,
|
---|
| 552 | self.departureTime, self.arrivalTime)
|
---|
| 553 | s += " %d %s," % (self.aircraftType, self.tailNumber)
|
---|
[1069] | 554 | s += " pax=%d+%d+%d, crew=%d+%d, bag=%d, cargo=%d, mail=%d" % \
|
---|
| 555 | (self.numPassengers, self.numChildren, self.numInfants,
|
---|
| 556 | self.numCockpitCrew, self.numCabinCrew,
|
---|
[1044] | 557 | self.bagWeight, self.cargoWeight, self.mailWeight)
|
---|
| 558 | s += ">"
|
---|
| 559 | return s
|
---|
| 560 |
|
---|
[745] | 561 | #---------------------------------------------------------------------------------------
|
---|
| 562 |
|
---|
[854] | 563 | class AcceptedFlight(RPCObject):
|
---|
| 564 | """A flight that has been already accepted."""
|
---|
| 565 | # The instructions for the construction
|
---|
| 566 | @staticmethod
|
---|
| 567 | def parseTimestamp(s):
|
---|
| 568 | """Parse the given RPC timestamp."""
|
---|
| 569 | dt = datetime.datetime.strptime(s, "%Y-%m-%d %H:%M:%S")
|
---|
| 570 | return calendar.timegm(dt.utctimetuple())
|
---|
| 571 |
|
---|
| 572 | _instructions = {
|
---|
| 573 | "bookedFlight" : lambda value: BookedFlight(value),
|
---|
| 574 | "numPassengers" : int,
|
---|
[1033] | 575 | "numChildren" : int,
|
---|
| 576 | "numInfants" : int,
|
---|
[854] | 577 | "fuelUsed" : int,
|
---|
| 578 | "rating" : lambda value: float(value) if value else 0.0
|
---|
| 579 | }
|
---|
| 580 |
|
---|
| 581 | def __init__(self, value):
|
---|
| 582 | """Construct the booked flight object from the given RPC result
|
---|
| 583 | value."""
|
---|
| 584 | super(AcceptedFlight, self).__init__(value, AcceptedFlight._instructions)
|
---|
| 585 | self.flightTimeStart = \
|
---|
| 586 | AcceptedFlight.parseTimestamp(self.flightDate + " " +
|
---|
| 587 | self.flightTimeStart)
|
---|
| 588 | self.flightTimeEnd = \
|
---|
| 589 | AcceptedFlight.parseTimestamp(self.flightDate + " " +
|
---|
| 590 | self.flightTimeEnd)
|
---|
| 591 | if self.flightTimeEnd<self.flightTimeStart:
|
---|
| 592 | self.flightTimeEnd += 24*60*60
|
---|
| 593 |
|
---|
[1033] | 594 | self.totalNumPassengers = self.numPassengers + self.numChildren + self.numInfants
|
---|
| 595 |
|
---|
[854] | 596 | #---------------------------------------------------------------------------------------
|
---|
| 597 |
|
---|
[745] | 598 | class Plane(rpccommon.Plane, RPCObject):
|
---|
| 599 | """An airplane in the fleet."""
|
---|
| 600 | _instructions = {
|
---|
| 601 | "status" : lambda value: rpccommon.Plane.str2status(value),
|
---|
[859] | 602 | "gateNumber" : lambda value: value if value else None,
|
---|
[1044] | 603 | "typeCode": lambda value: BookedFlight._decodeAircraftType(value),
|
---|
| 604 | "dow": int,
|
---|
| 605 | "dowNumCabinCrew": int,
|
---|
| 606 | "maxPassengers": int,
|
---|
[745] | 607 | }
|
---|
| 608 |
|
---|
| 609 | def __init__(self, value):
|
---|
| 610 | """Construct the plane."""
|
---|
| 611 | RPCObject.__init__(self, value, instructions = Plane._instructions)
|
---|
[859] | 612 | self.aircraftType = self.typeCode
|
---|
| 613 | del self.typeCode
|
---|
[745] | 614 |
|
---|
| 615 | #---------------------------------------------------------------------------------------
|
---|
| 616 |
|
---|
| 617 | class Fleet(rpccommon.Fleet):
|
---|
| 618 | """The fleet."""
|
---|
| 619 | def __init__(self, value):
|
---|
| 620 | """Construct the fleet."""
|
---|
| 621 | super(Fleet, self).__init__()
|
---|
| 622 | for planeValue in value:
|
---|
| 623 | self._addPlane(Plane(planeValue))
|
---|
| 624 |
|
---|
| 625 | #---------------------------------------------------------------------------------------
|
---|
| 626 |
|
---|
[756] | 627 | class Registration(object):
|
---|
| 628 | """Data for registration."""
|
---|
[759] | 629 | def __init__(self, surName, firstName, nameOrder,
|
---|
| 630 | yearOfBirth, emailAddress, emailAddressPublic,
|
---|
[756] | 631 | vatsimID, ivaoID, phoneNumber, nationality, password):
|
---|
| 632 | """Construct the registration data."""
|
---|
[759] | 633 | self.surName = surName
|
---|
| 634 | self.firstName = firstName
|
---|
| 635 | self.nameOrder = nameOrder
|
---|
[756] | 636 | self.yearOfBirth = yearOfBirth
|
---|
| 637 | self.emailAddress = emailAddress
|
---|
| 638 | self.emailAddressPublic = 1 if emailAddressPublic is True else \
|
---|
| 639 | 0 if emailAddressPublic is False else emailAddressPublic
|
---|
| 640 | self.vatsimID = "" if vatsimID is None else vatsimID
|
---|
| 641 | self.ivaoID = "" if ivaoID is None else ivaoID
|
---|
| 642 | self.phoneNumber = phoneNumber
|
---|
| 643 | self.nationality = nationality
|
---|
| 644 | self.password = password
|
---|
| 645 |
|
---|
| 646 | #---------------------------------------------------------------------------------------
|
---|
| 647 |
|
---|
[745] | 648 | class RPCException(Exception):
|
---|
| 649 | """An exception thrown by RPC operations."""
|
---|
| 650 | def __init__(self, result, message = None):
|
---|
| 651 | """Construct the exception."""
|
---|
| 652 | self._result = result
|
---|
| 653 | if message is None:
|
---|
| 654 | message = "RPC call failed with result code: %d" % (result,)
|
---|
| 655 | super(RPCException, self).__init__(message)
|
---|
| 656 |
|
---|
| 657 | @property
|
---|
| 658 | def result(self):
|
---|
| 659 | """Get the result code."""
|
---|
| 660 | return self._result
|
---|
| 661 |
|
---|
| 662 | #---------------------------------------------------------------------------------------
|
---|
| 663 |
|
---|
| 664 | class Client(object):
|
---|
| 665 | """The RPC client interface."""
|
---|
[820] | 666 | # The client protocol version
|
---|
| 667 | VERSION = 2
|
---|
| 668 |
|
---|
[745] | 669 | # Result code: OK
|
---|
| 670 | RESULT_OK = 0
|
---|
| 671 |
|
---|
| 672 | # Result code: the login has failed
|
---|
| 673 | RESULT_LOGIN_FAILED = 1
|
---|
| 674 |
|
---|
| 675 | # Result code: the given session ID is unknown (it might have expired).
|
---|
| 676 | RESULT_SESSION_INVALID = 2
|
---|
| 677 |
|
---|
| 678 | # Result code: some database error
|
---|
| 679 | RESULT_DATABASE_ERROR = 3
|
---|
| 680 |
|
---|
[756] | 681 | # Result code: invalid data
|
---|
| 682 | RESULT_INVALID_DATA = 4
|
---|
| 683 |
|
---|
[745] | 684 | # Result code: the flight does not exist
|
---|
| 685 | RESULT_FLIGHT_NOT_EXISTS = 101
|
---|
| 686 |
|
---|
| 687 | # Result code: the flight has already been reported.
|
---|
| 688 | RESULT_FLIGHT_ALREADY_REPORTED = 102
|
---|
| 689 |
|
---|
[756] | 690 | # Result code: a user with the given e-mail address already exists
|
---|
| 691 | RESULT_EMAIL_ALREADY_REGISTERED = 103
|
---|
| 692 |
|
---|
[745] | 693 | def __init__(self, getCredentialsFn):
|
---|
| 694 | """Construct the client."""
|
---|
| 695 | self._getCredentialsFn = getCredentialsFn
|
---|
| 696 |
|
---|
[1039] | 697 | sslContext = ssl.SSLContext()
|
---|
| 698 | sslContext.load_verify_locations(cafile = certifi.where())
|
---|
| 699 | transport = jsonrpclib.jsonrpc.SafeTransport(jsonrpclib.config.DEFAULT,
|
---|
| 700 | sslContext)
|
---|
| 701 |
|
---|
| 702 | self._server = jsonrpclib.Server(MAVA_BASE_URL + "/jsonrpc.php",
|
---|
| 703 | transport = transport)
|
---|
[745] | 704 |
|
---|
| 705 | self._userName = None
|
---|
| 706 | self._passwordHash = None
|
---|
| 707 | self._sessionID = None
|
---|
| 708 | self._loginCount = 0
|
---|
| 709 |
|
---|
| 710 | @property
|
---|
| 711 | def valid(self):
|
---|
| 712 | """Determine if the client is valid, i.e. there is a session ID
|
---|
| 713 | stored."""
|
---|
| 714 | return self._sessionID is not None
|
---|
| 715 |
|
---|
| 716 | def setCredentials(self, userName, password):
|
---|
| 717 | """Set the credentials for future logins."""
|
---|
| 718 |
|
---|
| 719 | self._userName = userName
|
---|
| 720 |
|
---|
| 721 | md5 = hashlib.md5()
|
---|
[928] | 722 | md5.update(password.encode("utf-8"))
|
---|
[745] | 723 | self._passwordHash = md5.hexdigest()
|
---|
| 724 |
|
---|
| 725 | self._sessionID = None
|
---|
| 726 |
|
---|
[756] | 727 | def register(self, registrationData):
|
---|
| 728 | """Register with the given data.
|
---|
| 729 |
|
---|
| 730 | Returns a tuple of:
|
---|
| 731 | - the error code,
|
---|
| 732 | - the PID if there is no error."""
|
---|
| 733 | reply = Reply(self._server.register(registrationData))
|
---|
| 734 |
|
---|
| 735 | return (reply.result,
|
---|
| 736 | reply.value["pid"] if reply.result==Client.RESULT_OK else None)
|
---|
| 737 |
|
---|
[745] | 738 | def login(self):
|
---|
| 739 | """Login using the given previously set credentials.
|
---|
| 740 |
|
---|
| 741 | The session ID is stored in the object and used for later calls.
|
---|
| 742 |
|
---|
[756] | 743 | Returns the name of the pilot on success, or None on error."""
|
---|
[745] | 744 | self._sessionID = None
|
---|
| 745 |
|
---|
[820] | 746 | reply = Reply(self._server.login(self._userName, self._passwordHash,
|
---|
| 747 | Client.VERSION))
|
---|
[745] | 748 | if reply.result == Client.RESULT_OK:
|
---|
| 749 | self._loginCount += 1
|
---|
| 750 | self._sessionID = reply.value["sessionID"]
|
---|
[858] | 751 |
|
---|
| 752 | types = [BookedFlight.TYPECODE2TYPE[typeCode]
|
---|
| 753 | for typeCode in reply.value["typeCodes"]]
|
---|
| 754 |
|
---|
[1074] | 755 | return (reply.value["name"], reply.value["rank"], types,
|
---|
| 756 | self._sessionID)
|
---|
[745] | 757 | else:
|
---|
| 758 | return None
|
---|
| 759 |
|
---|
| 760 | def getFlights(self):
|
---|
| 761 | """Get the flights available for performing."""
|
---|
[809] | 762 | bookedFlights = []
|
---|
| 763 | reportedFlights = []
|
---|
| 764 | rejectedFlights = []
|
---|
[745] | 765 |
|
---|
| 766 | value = self._performCall(lambda sessionID:
|
---|
| 767 | self._server.getFlights(sessionID))
|
---|
| 768 | for flightData in value:
|
---|
[809] | 769 | flight = BookedFlight(flightData)
|
---|
| 770 | if flight.status == BookedFlight.STATUS_BOOKED:
|
---|
| 771 | bookedFlights.append(flight)
|
---|
| 772 | elif flight.status == BookedFlight.STATUS_REPORTED:
|
---|
| 773 | reportedFlights.append(flight)
|
---|
| 774 | elif flight.status == BookedFlight.STATUS_REJECTED:
|
---|
| 775 | rejectedFlights.append(flight)
|
---|
[745] | 776 |
|
---|
[809] | 777 | for flights in [bookedFlights, reportedFlights, rejectedFlights]:
|
---|
[923] | 778 | flights.sort(key = lambda flight: flight.departureTime)
|
---|
[745] | 779 |
|
---|
[809] | 780 | return (bookedFlights, reportedFlights, rejectedFlights)
|
---|
[745] | 781 |
|
---|
[854] | 782 | def getAcceptedFlights(self):
|
---|
| 783 | """Get the flights that are already accepted."""
|
---|
| 784 | value = self._performCall(lambda sessionID:
|
---|
| 785 | self._server.getAcceptedFlights(sessionID))
|
---|
| 786 | flights = []
|
---|
| 787 | for flight in value:
|
---|
| 788 | flights.append(AcceptedFlight(flight))
|
---|
| 789 | return flights
|
---|
| 790 |
|
---|
[761] | 791 | def getEntryExamStatus(self):
|
---|
| 792 | """Get the status of the exams needed for joining MAVA."""
|
---|
| 793 | value = self._performCall(lambda sessionID:
|
---|
| 794 | self._server.getEntryExamStatus(sessionID))
|
---|
[764] | 795 | return (value["entryExamPassed"], value["entryExamLink"],
|
---|
[769] | 796 | value["checkFlightStatus"], value["madeFO"])
|
---|
[761] | 797 |
|
---|
[745] | 798 | def getFleet(self):
|
---|
| 799 | """Query and return the fleet."""
|
---|
| 800 | value = self._performCall(lambda sessionID:
|
---|
| 801 | self._server.getFleet(sessionID))
|
---|
| 802 |
|
---|
| 803 | return Fleet(value)
|
---|
| 804 |
|
---|
| 805 | def updatePlane(self, tailNumber, status, gateNumber):
|
---|
| 806 | """Update the state and position of the plane with the given tail
|
---|
| 807 | number."""
|
---|
| 808 | status = rpccommon.Plane.status2str(status)
|
---|
| 809 | self._performCall(lambda sessionID:
|
---|
| 810 | self._server.updatePlane(sessionID, tailNumber,
|
---|
| 811 | status, gateNumber))
|
---|
| 812 |
|
---|
[853] | 813 | def addPIREP(self, flightID, pirep, update = False):
|
---|
[745] | 814 | """Add the PIREP for the given flight."""
|
---|
| 815 | (result, _value) = \
|
---|
| 816 | self._performCall(lambda sessionID:
|
---|
[853] | 817 | self._server.addPIREP(sessionID, flightID, pirep,
|
---|
| 818 | update),
|
---|
[745] | 819 | acceptResults = [Client.RESULT_FLIGHT_ALREADY_REPORTED,
|
---|
| 820 | Client.RESULT_FLIGHT_NOT_EXISTS])
|
---|
| 821 | return result
|
---|
| 822 |
|
---|
| 823 | def updateOnlineACARS(self, acars):
|
---|
| 824 | """Update the online ACARS from the given data."""
|
---|
| 825 | self._performCall(lambda sessionID:
|
---|
| 826 | self._server.updateOnlineACARS(sessionID, acars))
|
---|
| 827 |
|
---|
[769] | 828 | def setCheckFlightPassed(self, type):
|
---|
| 829 | """Mark the check flight of the user passed with the given type."""
|
---|
| 830 | self._performCall(lambda sessionID:
|
---|
| 831 | self._server.setCheckFlightPassed(sessionID, type))
|
---|
| 832 |
|
---|
[829] | 833 | def getPIREP(self, flightID):
|
---|
| 834 | """Get the PIREP data for the flight with the given ID."""
|
---|
| 835 | value = self._performCall(lambda sessionID:
|
---|
| 836 | self._server.getPIREP(sessionID, flightID))
|
---|
| 837 | return value
|
---|
| 838 |
|
---|
[821] | 839 | def reflyFlights(self, flightIDs):
|
---|
| 840 | """Mark the flights with the given IDs for reflying."""
|
---|
| 841 | self._performCall(lambda sessionID:
|
---|
| 842 | self._server.reflyFlights(sessionID, flightIDs))
|
---|
| 843 |
|
---|
[824] | 844 | def deleteFlights(self, flightIDs):
|
---|
| 845 | """Delete the flights with the given IDs."""
|
---|
| 846 | self._performCall(lambda sessionID:
|
---|
| 847 | self._server.deleteFlights(sessionID, flightIDs))
|
---|
| 848 |
|
---|
[858] | 849 | def getTimetable(self, date, types = None):
|
---|
| 850 | """Get the time table for the given date restricted to the given list
|
---|
| 851 | of type codes, if any."""
|
---|
| 852 | typeCodes = None if types is None else \
|
---|
| 853 | [BookedFlight.TYPE2TYPECODE[type] for type in types]
|
---|
| 854 |
|
---|
| 855 | values = self._performCall(lambda sessionID:
|
---|
| 856 | self._server.getTimetable(sessionID,
|
---|
| 857 | date.strftime("%Y-%m-%d"),
|
---|
[859] | 858 | date.weekday()+1,
|
---|
[858] | 859 | typeCodes))
|
---|
| 860 | return ScheduledFlightPair.scheduledFlights2Pairs([ScheduledFlight(value)
|
---|
[859] | 861 | for value in values],
|
---|
| 862 | date)
|
---|
| 863 |
|
---|
| 864 | def bookFlights(self, flightIDs, date, tailNumber):
|
---|
| 865 | """Book the flights with the given IDs on the given date to be flown
|
---|
| 866 | with the plane of the given tail number."""
|
---|
| 867 | values = self._performCall(lambda sessionID:
|
---|
| 868 | self._server.bookFlights(sessionID,
|
---|
| 869 | flightIDs,
|
---|
| 870 | date.strftime("%Y-%m-%d"),
|
---|
| 871 | tailNumber))
|
---|
| 872 | return [BookedFlight(value) for value in values]
|
---|
[858] | 873 |
|
---|
[1074] | 874 | def getSimBriefResult(self, timestamp):
|
---|
| 875 | """Get the SimBrief results for the given timestamp."""
|
---|
| 876 | return self._performCall(lambda sessionID:
|
---|
| 877 | self._server.getSimBriefResult(sessionID,
|
---|
| 878 | timestamp))
|
---|
| 879 |
|
---|
[745] | 880 | def _performCall(self, callFn, acceptResults = []):
|
---|
| 881 | """Perform a call using the given call function.
|
---|
| 882 |
|
---|
| 883 | acceptResults should be a list of result codes that should be accepted
|
---|
| 884 | besides RESULT_OK. If this list is not empty, the returned value is a
|
---|
| 885 | tuple of the result code and the corresponding value. Otherwise only
|
---|
| 886 | RESULT_OK is accepted, and the value is returned.
|
---|
| 887 |
|
---|
| 888 | All other error codes are converted to exceptions."""
|
---|
| 889 | numAttempts = 0
|
---|
| 890 | while True:
|
---|
| 891 | reply = Reply(callFn(self._ensureSession()))
|
---|
| 892 | numAttempts += 1
|
---|
| 893 | result = reply.result
|
---|
| 894 | if result==Client.RESULT_SESSION_INVALID:
|
---|
| 895 | self._sessionID = None
|
---|
| 896 | if numAttempts==3:
|
---|
| 897 | raise RPCException(result)
|
---|
| 898 | elif result!=Client.RESULT_OK and result not in acceptResults:
|
---|
| 899 | raise RPCException(result)
|
---|
| 900 | elif acceptResults:
|
---|
| 901 | return (result, reply.value)
|
---|
| 902 | else:
|
---|
| 903 | return reply.value
|
---|
| 904 |
|
---|
| 905 | def _ensureSession(self):
|
---|
| 906 | """Ensure that there is a valid session ID."""
|
---|
| 907 | while self._sessionID is None:
|
---|
| 908 | if self._userName is not None and self._passwordHash is not None:
|
---|
| 909 | if not self.login():
|
---|
| 910 | self._userName = self._passwordHash = None
|
---|
| 911 |
|
---|
| 912 | if self._userName is None or self._passwordHash is None:
|
---|
| 913 | (self._userName, password) = self._getCredentialsFn()
|
---|
| 914 | if self._userName is None:
|
---|
| 915 | raise RPCException(Client.RESULT_LOGIN_FAILED)
|
---|
| 916 |
|
---|
| 917 | md5 = hashlib.md5()
|
---|
| 918 | md5.update(password)
|
---|
| 919 | self._passwordHash = md5.hexdigest()
|
---|
| 920 |
|
---|
| 921 | return self._sessionID
|
---|
| 922 |
|
---|
| 923 | #---------------------------------------------------------------------------------------
|
---|