diff --git a/django_diu/import_command.py b/django_diu/import_command.py index 42940e4e3465e56e2c64344b7104b35a7890fda8..2dc255e2d613ef03e724ac596ea6e78d11fcce4f 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 ea4c949bf25ff2b895f28718e64cea500ee1ce74..00dbccdce2d97f8228b6b8e6ab6c97f2a58583c5 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