mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-22 08:21:13 -05:00
f98aca05e9
Other systems we’re working with won’t easily be adapted to emit JSON instead of CAP, so it’s less work for us to do that conversion. This commit adds to code to parse the XML and turn it into a dict that we can work with, including converting the polygon string into native Python lists.
32 lines
905 B
Python
32 lines
905 B
Python
from bs4 import BeautifulSoup
|
||
|
||
|
||
def cap_xml_to_dict(cap_xml):
|
||
# This function assumes that it’s being passed valid CAP XML
|
||
cap = BeautifulSoup(cap_xml, "xml")
|
||
return {
|
||
"reference": cap.alert.identifier.text,
|
||
"category": cap.alert.info.category.text,
|
||
"expires": cap.alert.info.expires.text,
|
||
"content": cap.alert.info.description.text,
|
||
"areas": [
|
||
{
|
||
"name": area.areaDesc.text,
|
||
"polygons": [
|
||
cap_xml_polygon_to_list(polygon.text)
|
||
for polygon in area.find_all('polygon')
|
||
]
|
||
}
|
||
for area in cap.alert.info.find_all('area')
|
||
]
|
||
}
|
||
|
||
|
||
def cap_xml_polygon_to_list(polygon_string):
|
||
return [
|
||
[
|
||
float(coordinate) for coordinate in pair.split(',')
|
||
]
|
||
for pair in polygon_string.split(' ')
|
||
]
|