#!/awips2/python/bin/python2

##
# DR 6518 - This script will update any found site-level overrides to 
# cave_static/base/textws/gui/QualityControlCfg.xml with new configuration
# entries required to support the DSW and SQW WarnGen products.
##

import logging
import glob
import re
import shutil
import xml.dom.minidom as minidom
import xml.etree.ElementTree as ET


logging.basicConfig(format='%(asctime)-15s %(levelname)s:  %(message)s', 
                    datefmt='%Y-%m-%d %H:%M:%S',
                    level=logging.INFO)
log = logging.getLogger("updateQualityControlCfg.py")

QC_CONFIG_PATH = "/awips2/edex/data/utility/cave_static/site/*/textws/gui/QualityControlCfg.xml"



def main():
    log.info("Starting delta script for DR #6518: updating QualityControlCfg.xml...")
    
    for file in glob.iglob(QC_CONFIG_PATH):
        log.info("Updating file [%s]...", file)
        
        tree = ET.parse(file)
        root = tree.getroot()
        
        product_map = root.find("productTypeMap")
        if product_map is not None:
            if product_map.find("./item[key='DSW']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "DSW"
                ET.SubElement(new_item, "value").text = "Dust Storm Warning"
            
            if product_map.find("./item[key='SQW']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "SQW"
                ET.SubElement(new_item, "value").text = "Snow Squall Warning"
            
            if product_map.find("./item[key='DS.W']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "DS.W"
                ET.SubElement(new_item, "value").text = "Dust Storm Warning"
            
            if product_map.find("./item[key='DS.Y']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "DS.Y"
                ET.SubElement(new_item, "value").text = "Dust Advisory"
            
            if product_map.find("./item[key='SQ.W']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "SQ.W"
                ET.SubElement(new_item, "value").text = "Snow Squall Warning"
                
            if product_map.find("./item[key='FA.Y']") is None:
                new_item = ET.SubElement(product_map, "item")
                ET.SubElement(new_item, "key").text = "FA.Y"
                ET.SubElement(new_item, "value").text = "Flood Advisory"
        else:
            log.error("Could not find productTypeMap in file [%s].", file)
            continue
        
        followup_nnn = root.find("followupNNN")
        if followup_nnn is not None:
            if followup_nnn.find("./item[key='DS.W']") is None:
                new_item = ET.SubElement(followup_nnn, "item")
                ET.SubElement(new_item, "key").text = "DS.W"
                ET.SubElement(new_item, "value").text = "DSW"
            
            if followup_nnn.find("./item[key='DS.Y']") is None:
                new_item = ET.SubElement(followup_nnn, "item")
                ET.SubElement(new_item, "key").text = "DS.Y"
                ET.SubElement(new_item, "value").text = "DSW"
            
            if followup_nnn.find("./item[key='SQ.W']") is None:
                new_item = ET.SubElement(followup_nnn, "item")
                ET.SubElement(new_item, "key").text = "SQ.W"
                ET.SubElement(new_item, "value").text = "SQW"
        else:
            log.error("Could not find followupNNN in file [%s].", file)
            continue
             
        nnn_ident = root.find("nnnOfIdent")
        if nnn_ident is not None:
            if nnn_ident.find("./item[key='DS.W']") is None:
                new_item = ET.SubElement(nnn_ident, "item")
                ET.SubElement(new_item, "key").text = "DS.W"
                ET.SubElement(new_item, "value").text = "DSW"
            
            if nnn_ident.find("./item[key='DS.Y']") is None:
                new_item = ET.SubElement(nnn_ident, "item")
                ET.SubElement(new_item, "key").text = "DS.Y"
                ET.SubElement(new_item, "value").text = "DSW"
            
            if nnn_ident.find("./item[key='SQ.W']") is None:
                new_item = ET.SubElement(nnn_ident, "item")
                ET.SubElement(new_item, "key").text = "SQ.W"
                ET.SubElement(new_item, "value").text = "SQW"
        else:
            log.error("Could not find nnnOfIdent in file [%s].", file)
            continue
        
        bullet_map = root.find("bulletTypeMap")
        if bullet_map is not None:
            if bullet_map.find("./item[key='DS.Y']") is None:
                new_item = ET.SubElement(bullet_map, "item")
                ET.SubElement(new_item, "key").text = "DS.Y"
                ET.SubElement(new_item, "value").text = "Dust Advisory"
        else:
            log.error("Could not find bulletTypeMap in file [%s].", file)
            continue
        
        # make backup
        shutil.copy(file, file + ".bak_dr_6518")
        
        # Cleanup junk new-lines and extraneous spaces from output
        pretty_xml = minidom.parseString(ET.tostring(root, 'utf-8')).toprettyxml(indent=' '*4, encoding='UTF-8')
        pretty_xml = '\n'.join([line for line in pretty_xml.split('\n') if line.strip()])
        text_re = re.compile('>\n\s+([^<>\s].*?)\n\s+</', re.DOTALL)
        pretty_xml = text_re.sub('>\g<1></', pretty_xml)
        
        # because of how the SegmentedNNN tags are defined, if we used ET to
        # insert these new entries they would just be placed at the end of the
        # file and not with the block of existing entries
        try:
            for pil in ['DSW', 'SQW']:
                segmented_nnn_entry = '<SegmentedNNN>{}</SegmentedNNN>'.format(pil)
                if segmented_nnn_entry not in pretty_xml:
                    index = pretty_xml.rindex('</SegmentedNNN>')
                    index += len('</SegmentedNNN>') + 1
                    segmented_nnn_entry = ' '*4 + segmented_nnn_entry + '\n'
                    pretty_xml = pretty_xml[:index] + segmented_nnn_entry + pretty_xml[index:]
        except ValueError:
            log.error("Could not find SegmentedNNN in file [%s].", file)
            continue
        except:
            log.exception("Failed to update SegmentedNNN entries in file [%s].", file)
            continue
        
        with open(file, 'w') as out_file:
            out_file.write(pretty_xml)

    log.info("Delta script complete.")



if __name__ == '__main__':
    main()