from lxml import etree

from optparse import OptionParser #@UnresolvedImport

def get_options():

    parser = OptionParser()

    parser.add_option("-f", "--file", dest="outputfile",

                      help="destination filename", metavar="FILE", default="twitter_export_conf.xml")

    parser.add_option("-i", "--input", dest="inputfile", 

                      help="inputfile", metavar="INPUT", default=None)

    return parser.parse_args()

if __name__ == "__main__":

    (options, args) = get_options()

    dest_filename = options.outputfile

    path_list = []

    if options.inputfile is None:

        path_list = args

    else:

        with open(options.inputfile, 'r') as fi:

            path_list = fi

    root = etree.Element("twitter_export")

    for path in path_list:

        iri_doc = etree.parse(path)

        media_nodes = iri_doc.xpath("/iri/body/medias/media[@id='video']/video")

        duration = int(media_nodes[0].get("dur"))/1000

        file_elem = etree.SubElement(root, "file")

        etree.SubElement(file_elem, "path").text = path

        etree.SubElement(file_elem, "start_date")

        etree.SubElement(file_elem, "duration").text = unicode(duration)

    tree = etree.ElementTree(root)

    tree.write(dest_filename, encoding="utf-8", pretty_print=True, xml_declaration=True)