from __future__ import absolute_import
import argparse
from xml.dom.minidom import parse
import csv


def main():
    """
    Take as options: Xml file, CSV File

    Parse the XML and write each value get from it inside the CSV file provided.
    :return: None
    """
    parser = argparse.ArgumentParser(description="Convert XML file to CSV file")
    parser.add_argument("xml", metavar='xml', type=str, help="The XML file to parse")
    parser.add_argument("csv", metavar='csv', type=str, help="The CSV file to generate")
    args = parser.parse_args()

    sitemap_to_csv(args.xml, args.csv)


def sitemap_to_csv(xml_file, csv_file):
    try:
        datasource = open(xml_file, "r")
        dom = parse(datasource)
    except IOError:
        raise IOError("Bad XML file provided")

    with open(csv_file, 'wb') as opencsv:
        urls = dom.getElementsByTagName('url')
        for url in urls:
            writer = csv.writer(opencsv, delimiter=';', quotechar='|', quoting=csv.QUOTE_MINIMAL)
            writer.writerow([url.getElementsByTagName('loc')[0].firstChild.nodeValue])

from __future__ import absolute_import
import argparse
from xml.dom.minidom import parse
import csv


def main():
    """
    Take as options: Xml file, CSV File

    Parse the XML and write each value get from it inside the CSV file provided.
    :return: None
    """
    parser = argparse.ArgumentParser(description="Convert XML file to CSV file")
    parser.add_argument("xml", metavar='xml', type=str, help="The XML file to parse")
    parser.add_argument("csv", metavar='csv', type=str, help="The CSV file to generate")
    args = parser.parse_args()

    sitemap_to_csv(args.xml, args.csv)


def sitemap_to_csv(xml_file, csv_file):
    try:
        datasource = open(xml_file, "r+")
        dom = parse(datasource)
    except IOError:
        raise IOError("Bad XML file provided")

    with open(csv_file, 'w+') as opencsv:
        urls = dom.getElementsByTagName('url')
        for url in urls:
            writer = csv.writer(opencsv, delimiter=';', quotechar='|', quoting=csv.QUOTE_MINIMAL)
            writer.writerow([url.getElementsByTagName('loc')[0].firstChild.nodeValue])

fix an error in xmltocsv for py3k support