Skip to content

Commit cf9493d

Browse files
committed
merge telemetry props into Feature props
pass common definitions to reuse telemetry props, make argument required for these initialization functions
1 parent 3ee2e3d commit cf9493d

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

schema/common.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -74,14 +74,14 @@ def vehicle_state_machine(vehicle_state=None, vehicle_events=None):
7474
return (state_machine_defs, transitions)
7575

7676

77-
def vehicle_type_counts_definition(definitions=None):
77+
def vehicle_type_counts_definition(definitions):
7878
"""
7979
Generate a definition for a dict of vehicle_type: int.
8080
"""
8181
vehicle_type_counts = {}
8282
def_name = "vehicle_type_counts"
8383
def_id = definition_id(def_name)
84-
vehicle_types = definitions["vehicle_type"] if definitions else load_definitions("vehicle_type")
84+
vehicle_types = definitions["vehicle_type"]
8585

8686
for vehicle_type in vehicle_types["enum"]:
8787
vehicle_type_counts[vehicle_type] = {
@@ -131,7 +131,7 @@ def point_definition():
131131
}
132132

133133

134-
def mds_feature_point_definition():
134+
def mds_feature_point_definition(definitions):
135135
"""
136136
Create a customized definition of the GeoJSON Feature schema for MDS Points.
137137
"""
@@ -164,6 +164,10 @@ def mds_feature_point_definition():
164164
}
165165
}
166166

167+
# merge telemetry props
168+
telemetry = definitions["telemetry"]
169+
f_properties["properties"].update(telemetry["properties"])
170+
167171
return {MDS_FEATURE_POINT: feature}
168172

169173

@@ -216,7 +220,7 @@ def load_definitions(*args, allow_null=False):
216220
common_definitions = common["definitions"]
217221

218222
# MDS specific geography definition
219-
mds_feature = mds_feature_point_definition()
223+
mds_feature = mds_feature_point_definition(common_definitions)
220224
common_definitions.update(mds_feature)
221225

222226
# vehicle_type -> count definition

0 commit comments

Comments
 (0)