python 读写XML

# coding:utf-8
import os
import sys
from lxml import etree

sys.path.append(os.path.dirname(os.path.dirname(__file__)))
from abbr_extractor.pgconnect import PgConnect
from common_tool.common_utils import get_latest_database_name, region_version
from db_options import get_all_host_options, get_all_database, select_u_turn_info


def collect_all_u_turn_info():
    host_option_list = get_all_host_options()
    # get all database name
    database_host_dic = {}
    database_name_list = []
    for host_option in host_option_list:
        pg_conn = PgConnect(host_option)
        if pg_conn.init_db():
            host_database_list = get_all_database(pg_conn)
            database_name_list += host_database_list
            for name in host_database_list:
                database_host_dic[name] = host_option
    # get the latest database
    latest_name_list = get_latest_database_name(database_name_list)
    print latest_name_list

    # sql uturn
    u_turn_info_list_all = []
    for database_name in latest_name_list:
        region, version = region_version(database_name)
        host_option = database_host_dic.get(database_name)
        host_option.dbname = database_name
        pg_conn = PgConnect(host_option)
        if pg_conn.init_db():
            u_turn_info_list = select_u_turn_info(pg_conn)
            for u_turn_info in u_turn_info_list:
                setattr(u_turn_info, "region", region)
            u_turn_info_list_all += u_turn_info_list

    return u_turn_info_list_all


def save_u_turn_to_xml(u_turn_info_list, save_xml_filename):
    root = etree.Element(‘u-turn_config‘, nsmap={"xmlns": ‘http://www.w3.org/2001/XMLSchema-instance‘})

    for u_turn in u_turn_info_list:
        record_node = etree.Element(‘record‘)
        root.append(record_node)

        admin_place_id_node = etree.Element(‘admin_place_id‘)
        record_node.append(admin_place_id_node)
        admin_place_id_node.text = "%s" % u_turn.admin_place_id

        admin_type_node = etree.Element(‘admin_type‘)
        record_node.append(admin_type_node)
        admin_type_node.text = "%s" % u_turn.admin_type

        admin_name_node = etree.Element("admin_name")
        record_node.append(admin_name_node)
        if isinstance(u_turn.admin_name, str):
            u_turn.admin_name = unicode(u_turn.admin_name, ‘utf-8‘)
        admin_name_node.text = u_turn.admin_name

        country_node = etree.Element("country")
        record_node.append(country_node)
        country_node.text = u_turn.country

        region_node = etree.Element("region")
        record_node.append(region_node)
        region_node.text = u_turn.region.lower()

    tree = etree.ElementTree(root)
    tree.write(save_xml_filename, pretty_print=True, xml_declaration=True, encoding=‘utf-8‘)
    print "write content to [%s] finish." % save_xml_filename


def read_u_turn_xml(xml_file_path):
    xml = etree.parse(xml_file_path)
    root = xml.getroot()
    u_turn_info_list = []
    for recode_node in root.getchildren():
        from db_options import Options
        u_turn_info = Options()
        for tag_node in recode_node.getchildren():
            tag_name = tag_node.tag
            tag_text = tag_node.text
            setattr(u_turn_info, tag_name, tag_text)
        u_turn_info_list.append(u_turn_info)

    return u_turn_info_list


if __name__ == ‘__main__‘:
    u_turn_list = collect_all_u_turn_info()
    save_u_turn_to_xml(u_turn_list, ‘uturn_auto.xml‘)

    # u_turn_list = read_u_turn_xml(‘uturn.xml‘)
    # u_turn_list.sort(key=lambda d: d.admin_place_id)
    # save_u_turn_to_xml(u_turn_list, ‘uturn_sort.xml‘)