app/broadcast_message/translators.py

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(' ')
    ]
-												Add translators from CAP to dict

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.

											
										
										
											2021-01-18 10:01:45 +00:00
+								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 {
-												Don’t accept cancel or update via broadcast API

We don’t support these methods at the moment. Instead we were just
ignoring the `msgType` field, so issuing one of these commands would
cause a new alert to be broadcast 🙃

We might want to support `Cancel` in the future, but for now let’s
reject anything that isn’t `Alert` (CAP terminology for the initial
broadcast).

											
										
										
											2021-02-15 09:32:33 +00:00
+								        "msgType": cap.alert.msgType.text,
-												Add translators from CAP to dict

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.

											
										
										
											2021-01-18 10:01:45 +00:00
+								        "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(' ')
 								    ]