correct counting rows in source dataframe

django_diu/import_command.py

@@ -152,6 +152,9 @@ class DataFrameImportTask(ImportTask):
     def open_data_source(self):
         self.rows = self.dataframe.iterrows()
 
+    def count_source(self):
+        self.source_count = self.dataframe.shape[0]
+
 
 class ImportCommand(BaseCommand):
     help = "Generic command to import data into a django database"