From bcaa85a1424180604e13488e75fc81d4986c375e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20M=C3=A9nager?= <herve.menager@gmail.com> Date: Wed, 7 Oct 2020 15:05:10 +0200 Subject: [PATCH] use lxml instead of ElementTree for XML parsing --- django_diu/import_command.py | 4 ++-- requirements.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/django_diu/import_command.py b/django_diu/import_command.py index 42940e4..2dc255e 100644 --- a/django_diu/import_command.py +++ b/django_diu/import_command.py @@ -9,6 +9,7 @@ import mysql.connector import requests_cache import click import xml.etree.ElementTree as ET +from lxml import etree class MyConverter(mysql.connector.conversion.MySQLConverter): @@ -168,8 +169,7 @@ class XMLImportTask(ImportTask): xpathSelector = "*" def open_data_source(self): - tree = ET.parse(self.xmlFile) - self.root = tree.getroot() + self.root = etree.parse(self.xmlFile) self.rows = self.root.findall(self.xpathSelector) def count_source(self): diff --git a/requirements.txt b/requirements.txt index ea4c949..00dbccd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,6 @@ django requests_cache click +lxml pandas mysql-connector \ No newline at end of file -- GitLab