def __init__(self, velib_id):
self.id = int(velib_id)
self.informations = self.acquire_informations()
if not self.informations:
raise UnknowStation('The Station ID does not exist.')
self.status = StationStatus(self.id)
self.finder = StationGeoFinder(self)
class Station(object):
"""Station object,
combinating informations and status"""
almost_full = STATION_ALMOST_FULL
almost_empty = STATION_ALMOST_EMPTY
around_radius = STATION_AROUND_RADIUS
def __init__(self, velib_id):
self.id = int(velib_id)
self.informations = self.acquire_informations()
if not self.informations:
raise UnknowStation('The Station ID does not exist.')
self.status = StationStatus(self.id)
self.finder = StationGeoFinder(self)
def acquire_informations(self):
"""Default method for acquiring informations
of a stations, by connecting to the database"""
try:
return StationInformation.get(self.id)
except AttributeError:
db_connection()
return StationInformation.get(self.id)
@property
def is_open(self):
"""Property telling if the station is open"""
return self.informations.opened
@property
def is_bonus(self):
"""Property telling if the station is bonus"""
return self.informations.bonus
def is_free(self, places=1):
"""Method property if the station
has free places"""
return self.is_open and self.status.free >= places
def is_available(self, places=1):
"""Method property if the station
has available places"""
return self.is_open and self.status.available >= places
@property
def stations_around(self):
"""Find the stations around"""
return self.finder.get_stations_around(self.around_radius)
@property
def state(self):
"""Return the status for the station"""
if not self.is_open:
return STATUS_CLOSE
free = self.status.free
available = self.status.available
total = self.status.total
if not available and not free:
if not total:
return STATUS_ERROR
return STATUS_NO_SERVICE
elif available and not free:
return STATUS_BIKE_ONLY
elif not available and free:
return STATUS_PARKING_ONLY
elif self.almost_full >= free:
return STATUS_ALMOST_FULL
elif self.almost_empty >= available:
return STATUS_ALMOST_EMPTY
elif self.is_bonus:
return STATUS_BONUS
return STATUS_OPEN
def __repr__(self):
if self.informations:
return '<Station "%s" (%s)>' % (self.id, self.informations.address)
return '<Station "%s">' % self.id