added importer user to TnsCsvVisitImportReader

smash/web/importer/csv_tns_visit_import_reader.py

@@ -10,7 +10,7 @@ import pytz
 from django.conf import settings
 
 from warning_counter import MsgCounterHandler
-from web.models import StudySubject, Study, Visit, Appointment, AppointmentType, Location, AppointmentTypeLink, Subject
+from web.models import StudySubject, Study, Visit, Appointment, AppointmentType, Location, AppointmentTypeLink, Subject, User, Worker, Provenance
 from web.models.constants import GLOBAL_STUDY_ID
 
 CSV_DATE_FORMAT = "%d/%m/%Y"
@@ -33,6 +33,16 @@ class TnsCsvVisitImportReader:
         self.processed_count = 0
         self.warning_count = 0
 
+        self.importer_user = None
+
+        importer_user_name = getattr(settings, "IMPORTER_USER", None)
+        if importer_user_name is not None:
+            user = User.objects.filter(username=importer_user_name)
+            if user is None:
+                logger.warn("User does not exist: " + importer_user_name)
+            else:
+                self.importer_user = Worker.objects.filter(user=user)
+
     def load_data(self, filename):
         warning_counter = MsgCounterHandler()
         logging.getLogger('').addHandler(warning_counter)