add XML import class

django_diu/import_command.py

@@ -8,6 +8,8 @@ from django.core.management import BaseCommand, CommandError
 import mysql.connector
 import requests_cache
 import click
+import xml.etree.ElementTree as ET
+
 
 class MyConverter(mysql.connector.conversion.MySQLConverter):
 
@@ -158,6 +160,21 @@ class DataFrameImportTask(ImportTask):
         self.source_count = self.dataframe.shape[0]
 
 
+class XMLImportTask(ImportTask):
+
+    description = "Abstract XML import task"
+
+    xmlFile = None
+    xpathSelector = "*"
+
+    def open_data_source(self):
+        tree = ET.parse(argv[1])
+        root = tree.getroot()
+        self.rows = root.findall(self.xpathSelector)
+
+    def count_source(self):
+        self.source_count = len(self.rows)
+
 class ImportCommand(BaseCommand):
     help = "Generic command to import data into a django database"