def test_get_element_by_id(self):
idml_file = IDMLPackage(os.path.join(IDMLFILES_DIR, "4-pages.idml"), mode="r")
stories = idml_file.stories
story = Story(idml_file, stories[1]) # u11b
elem = story.get_element_by_id("di2i3i2")
self.assertEqual(elem.get("MarkupTag"), "XMLTag/content")
elem = story.get_element_by_id("di2i3i2", tag="*")
self.assertEqual(elem.get("MarkupTag"), "XMLTag/content")
def append_childs(source_node, destination_node):
"""Recursive function to discover node structure from a story to another. """
for elt in source_node.iterchildren():
if not elt.tag == "XMLElement":
append_childs(elt, destination_node)
if elt.get("Self") == source_node.get("Self"):
continue
if not elt.get("MarkupTag"):
continue
elt = XMLElement(elt)
new_destination_node = elt.to_xml_structure_element()
destination_node.append(new_destination_node)
if elt.get("XMLContent"):
xml_content_value = elt.get("XMLContent")
story_name = "Stories/Story_%s.xml" % xml_content_value
story = Story(self, name=story_name)
try:
new_source_node = story.get_element_by_id(elt.get("Self"))
# The story does not exists.
except KeyError:
continue
else:
append_childs(new_source_node, new_destination_node)
else:
append_childs(elt, new_destination_node)
def append_childs(source_node, destination_node):
"""Recursive function to discover node structure from a story to another. """
for elt in source_node.iterchildren():
if not elt.tag == "XMLElement":
append_childs(elt, destination_node)
if elt.get("Self") == source_node.get("Self"):
continue
if not elt.get("MarkupTag"):
continue
elt = XMLElement(elt)
new_destination_node = elt.to_xml_structure_element()
destination_node.append(new_destination_node)
if elt.get("XMLContent"):
xml_content_value = elt.get("XMLContent")
story_name = "Stories/Story_%s.xml" % xml_content_value
story = Story(self, name=story_name)
try:
new_source_node = story.get_element_by_id(
elt.get("Self"))
# The story does not exists.
except KeyError:
continue
else:
append_childs(new_source_node,
new_destination_node)
else:
append_childs(elt, new_destination_node)
def _add_stories_from_idml(self, idml_package, at, only):
"""Add all idml_package stories and insert `only' refence at `at' position in self.
What we have:
=============
o The Story file in self containing "at" (/Root/article[3] marked by (A)) [1]:
<XMLElement Self="di2" MarkupTag="XMLTag/Root">
<XMLElement Self="di2i3" MarkupTag="XMLTag/article" XMLContent="u102"/>
<XMLElement Self="di2i4" MarkupTag="XMLTag/article" XMLContent="udb"/>
<XMLElement Self="di2i5" MarkupTag="XMLTag/article" XMLContent="udd"/> (A)
<XMLElement Self="di2i6" MarkupTag="XMLTag/advertise" XMLContent="udf"/>
</XMLElement>
o The idml_package Story file containing "only" (/Root/module[1] marked by (B)) [2]:
<XMLElement Self="prefixeddi2" MarkupTag="XMLTag/Root">
<XMLElement Self="prefixeddi2i3" MarkupTag="XMLTag/module" XMLContent="prefixedu102"/> (B)
</XMLElement>
What we want:
=============
o if XMLContent is not related to a story file, we create it and insert `only' XML content:
<XMLElement Self="di2i5" MarkupTag="XMLTag/article" XMLContent="udd">
<XMLElement Self="prefixeddi2i3" MarkupTag="XMLTag/module" XMLContent="prefixedu102"/> (A)
</XMLElement>
o The Story_udd.xml is created.
o The Spread page item 'udd' is updated.
o The designmap.xml file is updated.
"""
xml_element_src_id = idml_package.xml_structure.xpath(only)[0].get("Self")
story_src_filename = idml_package.get_story_by_xpath(only)
story_src = Story(idml_package, story_src_filename)
story_src_elt = story_src.get_element_by_id(xml_element_src_id).element
xml_element_dest = self.xml_structure.xpath(at)[0]
xml_element_dest_id = xml_element_dest.get("Self")
content_ref = xml_element_dest.get("XMLContent")
# We don't want to lose the XMLContent referencing the spread page item.
# Neither we want to wipe the page item out from the spread.
# To keep the document valid, the solution is to create a proxy story.
if content_ref and (content_ref not in self.story_ids):
self.add_story_with_content(content_ref, xml_element_dest_id, xml_element_dest.tag)
self.xml_element_leaf_to_node(at, content_ref)
xml_element_dest = self.xml_structure.xpath(at)[0]
story_dest_filename = self.get_story_by_xpath(at)
story_dest = Story(self, story_dest_filename, self.working_copy_path)
story_dest_elt = story_dest.get_element_by_id(xml_element_dest_id)
story_src_elt_copy = copy.copy(story_src_elt)
if story_src_elt_copy.get("XMLContent"):
for child in story_src_elt_copy.iterchildren():
story_src_elt_copy.remove(child)
story_dest_elt.append(story_src_elt_copy)
story_dest.synchronize()
# Add Story files.
# `Stories' directory may not be present in the destination package.
stories_dirname = os.path.join(self.working_copy_path, STORIES_DIRNAME)
if not os.path.exists(stories_dirname):
os.mkdir(stories_dirname)
for filename in idml_package.stories_for_node(only):
story_cp = open(os.path.join(self.working_copy_path, filename), mode="wb+")
story_cp.write(idml_package.open(filename, mode="r").read())
story_cp.close()
# Update designmap.xml.
self.designmap.add_stories(idml_package.story_ids_for_node(only))
self.designmap.synchronize()
# BackingStory.xml ??
self.init_lazy_references()
def _add_stories_from_idml(self, idml_package, at, only):
"""Add all idml_package stories and insert `only' refence at `at' position in self.
What we have:
=============
o The Story file in self containing "at" (/Root/article[3] marked by (A)) [1]:
<XMLElement Self="di2" MarkupTag="XMLTag/Root">
<XMLElement Self="di2i3" MarkupTag="XMLTag/article" XMLContent="u102"/>
<XMLElement Self="di2i4" MarkupTag="XMLTag/article" XMLContent="udb"/>
<XMLElement Self="di2i5" MarkupTag="XMLTag/article" XMLContent="udd"/> (A)
<XMLElement Self="di2i6" MarkupTag="XMLTag/advertise" XMLContent="udf"/>
</XMLElement>
o The idml_package Story file containing "only" (/Root/module[1] marked by (B)) [2]:
<XMLElement Self="prefixeddi2" MarkupTag="XMLTag/Root">
<XMLElement Self="prefixeddi2i3" MarkupTag="XMLTag/module" XMLContent="prefixedu102"/> (B)
</XMLElement>
What we want:
=============
o if XMLContent is not related to a story file, we create it and insert `only' XML content:
<XMLElement Self="di2i5" MarkupTag="XMLTag/article" XMLContent="udd">
<XMLElement Self="prefixeddi2i3" MarkupTag="XMLTag/module" XMLContent="prefixedu102"/> (A)
</XMLElement>
o The Story_udd.xml is created.
o The Spread page item 'udd' is updated.
o The designmap.xml file is updated.
"""
xml_element_src_id = idml_package.xml_structure.xpath(only)[0].get(
"Self")
story_src_filename = idml_package.get_story_by_xpath(only)
story_src = Story(idml_package, story_src_filename)
story_src_elt = story_src.get_element_by_id(xml_element_src_id).element
xml_element_dest = self.xml_structure.xpath(at)[0]
xml_element_dest_id = xml_element_dest.get("Self")
content_ref = xml_element_dest.get("XMLContent")
# We don't want to lose the XMLContent referencing the spread page item.
# Neither we want to wipe the page item out from the spread.
# To keep the document valid, the solution is to create a proxy story.
if content_ref and (content_ref not in self.story_ids):
self.add_story_with_content(content_ref, xml_element_dest_id,
xml_element_dest.tag)
self.xml_element_leaf_to_node(at, content_ref)
xml_element_dest = self.xml_structure.xpath(at)[0]
story_dest_filename = self.get_story_by_xpath(at)
story_dest = Story(self, story_dest_filename, self.working_copy_path)
story_dest_elt = story_dest.get_element_by_id(xml_element_dest_id)
story_src_elt_copy = copy.copy(story_src_elt)
if story_src_elt_copy.get("XMLContent"):
for child in story_src_elt_copy.iterchildren():
story_src_elt_copy.remove(child)
story_dest_elt.append(story_src_elt_copy)
story_dest.synchronize()
# Add Story files.
# `Stories' directory may not be present in the destination package.
stories_dirname = os.path.join(self.working_copy_path, STORIES_DIRNAME)
if not os.path.exists(stories_dirname):
os.mkdir(stories_dirname)
for filename in idml_package.stories_for_node(only):
story_cp = open(os.path.join(self.working_copy_path, filename),
mode="w+")
story_cp.write(idml_package.open(filename, mode="r").read())
story_cp.close()
# Update designmap.xml.
self.designmap.add_stories(idml_package.story_ids_for_node(only))
self.designmap.synchronize()
# BackingStory.xml ??
self.init_lazy_references()
