get root properly

pull/399/head
Lex Neva 2019-03-08 20:50:48 -05:00
rodzic fb3c8186d2
commit d1c001857d
1 zmienionych plików z 5 dodań i 1 usunięć

Wyświetl plik

@ -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