diff options
| author | Lex Neva <github.com@lexneva.name> | 2019-03-08 20:50:48 -0500 |
|---|---|---|
| committer | Lex Neva <github.com@lexneva.name> | 2019-03-08 20:51:58 -0500 |
| commit | d1c001857d1e389950b3f346c1a0413d82899ca4 (patch) | |
| tree | d004d6ece7f217e17a6513233ac83dcbfed0e06b /lib/svg/svg.py | |
| parent | fb3c8186d275afa18c8146a453654afaf879ed34 (diff) | |
get root properly
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 |