mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 08:51:30 -05:00
33 lines
948 B
Python
33 lines
948 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 {
|
||
"msgType": cap.alert.msgType.text,
|
||
"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(' ')
|
||
]
|