class StreetNetworkSerializer(OutsideServiceCommon):
modes = StringListField(display_none=True)
timeout = MethodField(schema_type=float, display_none=False)
backend_class = MethodField(schema_type=str, display_none=False)
def get_timeout(self, obj):
return obj.get('timeout', None)
def get_backend_class(self, obj):
return obj.get('backend_class', None)
class EquipmentProvidersSerializer(NullableDictSerializer):
key = Field(schema_type=str, display_none=False)
codes_types = Field(schema_type=str, many=True, display_none=True)
timeout = MethodField(schema_type=int, display_none=False)
fail_max = MethodField(schema_type=int, display_none=False)
def get_timeout(self, obj):
return obj.get('timeout', None)
def get_fail_max(self, obj):
return obj.get('fail_max', None)
class RidesharingServicesSerializer(OutsideServiceCommon):
rating_scale_min = MethodField(schema_type=int, display_none=False)
rating_scale_max = MethodField(schema_type=int, display_none=False)
crowfly_radius = Field(schema_type=int, display_none=False)
network = Field(schema_type=str, display_none=False)
def get_rating_scale_min(self, obj):
return obj.get('rating_scale_min', None)
def get_rating_scale_max(self, obj):
return obj.get('rating_scale_max', None)
class OutsideServiceCommon(serpy.DictSerializer):
class_ = Field(schema_type=str, label='class', attr='class')
_id = Field(label='id', attr='id', schema_type=str)
circuit_breaker = MethodField(schema_type=CircuitBreakerSerializer,
display_none=False)
zmq_socket_ttl = MethodField(schema_type=float, display_none=False)
def get_circuit_breaker(self, obj):
o = obj.get('circuit_breaker', None)
return CircuitBreakerSerializer(o,
display_none=False).data if o else None
def get_zmq_socket_ttl(self, obj):
return obj.get('zmq_socket_ttl', None)
class CommonStatusSerializer(NullableDictSerializer):
status = Field(schema_type=str, display_none=False)
dataset_created_at = Field(schema_type=str, display_none=False)
is_realtime_loaded = Field(schema_type=bool, display_none=False)
data_version = Field(schema_type=int, display_none=False)
nb_threads = Field(schema_type=int, display_none=False)
is_open_service = Field(schema_type=bool, display_none=False)
is_connected_to_rabbitmq = Field(schema_type=bool, display_none=False)
autocomplete = AutocompleteSerializer(display_none=False)
end_production_date = Field(schema_type=str, display_none=False)
realtime_proxies = StringListField(display_none=True)
last_load_at = Field(schema_type=str, display_none=False)
publication_date = Field(schema_type=str, display_none=False)
street_networks = StreetNetworkSerializer(many=True, display_none=False)
ridesharing_services = RidesharingServicesSerializer(many=True, display_none=False)
start_production_date = Field(schema_type=str, display_none=False)
last_load_status = Field(schema_type=bool, display_none=False)
is_open_data = Field(schema_type=bool, display_none=False)
last_rt_data_loaded = Field(schema_type=str, display_none=False)
kraken_version = MethodField(schema_type=str, display_none=False)
region_id = Field(schema_type=str, display_none=False, description='Identifier of the coverage')
error = CoverageErrorSerializer(display_none=False)
def get_kraken_version(self, obj):
if "navitia_version" in obj:
return obj["navitia_version"]
if "kraken_version" in obj:
return obj["kraken_version"]
return None
class ExternalServiceProvidersServicesSerializer(NullableDictSerializer):
external_service_providers_keys = MethodField(schema_type=str,
many=True,
display_none=True)
def get_external_service_providers_keys(self, obj):
return obj.get('external_service_providers_keys', [])
class StatusSerializer(serpy.DictSerializer):
data_version = Field(schema_type=int)
dataset_created_at = Field(schema_type=str)
end_production_date = Field(schema_type=str)
is_connected_to_rabbitmq = Field(schema_type=bool)
is_open_data = Field(schema_type=bool)
is_open_service = Field(schema_type=bool)
is_realtime_loaded = Field(schema_type=bool)
kraken_version = Field(schema_type=str, attr=str("navitia_version"))
last_load_at = Field(schema_type=str)
last_load_status = Field(schema_type=bool)
last_rt_data_loaded = Field(schema_type=str)
nb_threads = Field(schema_type=int)
parameters = ParametersSerializer()
publication_date = Field(schema_type=str)
realtime_contributors = MethodField(schema_type=str,
many=True,
display_none=True)
realtime_proxies = StringListField(display_none=True)
start_production_date = Field(schema_type=str)
status = Field(schema_type=str)
traveler_profiles = TravelerProfilesSerializer(many=True)
def get_realtime_contributors(self, obj):
# so far, serpy cannot serialize an optional attr
# this has to be done manually
return obj.get('rt_contributors', [])
class RedisStatusSerializer(serpy.DictSerializer):
circuit_breaker = MethodField(schema_type=CircuitBreakerSerializer,
display_none=False)
def get_circuit_breaker(self, obj):
o = obj.get('circuit_breaker', None)
return CircuitBreakerSerializer(o,
display_none=False).data if o else None
class StatusSerializer(CommonStatusSerializer):
parameters = ParametersSerializer()
realtime_contributors = MethodField(schema_type=str, many=True, display_none=True)
traveler_profiles = TravelerProfilesSerializer(many=True)
def get_realtime_contributors(self, obj):
# so far, serpy cannot serialize an optional attr
# this has to be done manually
return obj.get('rt_contributors', [])
class AutocompleteSerializer(serpy.DictSerializer):
class_ = Field(schema_type=str, label='class', attr='class')
timeout = MethodField(schema_type=float, display_none=False)
timeout_bragi_es = MethodField(schema_type=float, display_none=False)
fast_timeout = MethodField(schema_type=float, display_none=False)
fast_timeout_bragi_es = MethodField(schema_type=float, display_none=False)
def get_timeout(self, obj):
return obj.get('timeout', None)
def get_timeout_bragi_es(self, obj):
return obj.get('timeout_bragi_es', None)
def get_fast_timeout(self, obj):
return obj.get('fast_timeout', None)
def get_fast_timeout_bragi_es(self, obj):
return obj.get('fast_timeout_bragi_es', None)
class EquipmentProvidersServicesSerializer(NullableDictSerializer):
equipment_providers_keys = MethodField(schema_type=str,
many=True,
display_none=True)
equipment_providers = EquipmentProvidersSerializer(many=True,
display_none=False)
def get_equipment_providers_keys(self, obj):
return obj.get('equipment_providers_keys', [])
def get_equipment_providers(self, obj):
return obj.get('equipment_providers', [])
class StreetNetworkSerializer(OutsideServiceCommon):
modes = StringListField(display_none=True)
timeout = MethodField(schema_type=float, display_none=False)
backend_class = MethodField(schema_type=str, display_none=False)
matrix_type = MethodField(schema_type=str, display_none=False)
max_matrix_points = MethodField(schema_type=int, display_none=False)
realtime_traffic = MethodField(schema_type=str, display_none=False)
language = MethodField(schema_type=str, display_none=False)
def get_timeout(self, obj):
return obj.get('timeout', None)
def get_backend_class(self, obj):
return obj.get('backend_class', None)
def get_matrix_type(self, obj):
return obj.get('matrix_type', None)
def get_max_matrix_points(self, obj):
return obj.get('max_matrix_points', None)
def get_realtime_traffic(self, obj):
return obj.get('realtime_traffic', None)
def get_language(self, obj):
return obj.get('language', None)
