diff options
Diffstat (limited to 'lib/svg/svg.py')
| -rw-r--r-- | lib/svg/svg.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/lib/svg/svg.py b/lib/svg/svg.py index 3715def8..464a2a18 100644 --- a/lib/svg/svg.py +++ b/lib/svg/svg.py @@ -9,7 +9,11 @@ def get_document(node): def generate_unique_id(document_or_element, prefix="path"): - document = get_document(document_or_element) + if isinstance(document_or_element, etree._ElementTree): + document = document_or_element.getroot() + else: + document = get_document(document_or_element) + doc_ids = {node.get('id') for node in document.iterdescendants() if 'id' in node.attrib} i = 1 |