def setUp(self):
self.tsk = CSVTask(
{
"csv_headers": {
"edilizia": {
"a": ["1", "2", "3"],
"b": ["4", "5", "6"]
},
"easyroom": {
"a": ["1", "2", "7"],
"b": ["4", "5", "8"]
}
},
"folders": {
"data_csv_sources": "folder"
}
}, MagicMock())
def run_csv(self, files):
"""
Process csv files.
Arguments:
- files: files we want process.
Returns: None
Instantiates a MongoDBPersistenceManager and a CSVTask to process the csv
files whit the apposite procedure.
"""
persistence = MongoDBPersistenceManager(self._config)
ODMModel.set_pm(persistence)
task = CSVTask(self._config)
task.perform_updates_on_files(files)
self.run_lookup()
