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