#!/usr/bin/env python

import requests

from lxml import etree

import sys

import zipfile

import re

import slugify

project_id = sys.argv[1]

dlurl = "http://ldt.iri.centrepompidou.fr/ldtplatform/ldt/space/ldt/project/" + project_id

r = requests.get(dlurl)

ldt_root = etree.fromstring(r.content)

project_title = ldt_root.find("project").get("title")

#create zipfile

if project_title:

    basename = slugify.slugify(project_title)

else:

    basename = project_id

zf = zipfile.ZipFile(basename+".zip", "w")

try:

    for i,media_node in enumerate(ldt_root.iterfind("medias/media")):

        iri_url = media_node.get("src")

        r_iri = requests.get(iri_url)

        iriname = "iri/%s_%d.iri" % (basename, i)

        zf.writestr(basename+"/"+iriname, r_iri.content)

        media_node.set("src", iriname)

        media_node.set("video", "")

    zf.writestr(basename+'/'+basename+".ldt", etree.tostring(ldt_root, xml_declaration=True, encoding='utf-8'))

finally:

    zf.close()