def do_import(self):
""" Imports objects into database """
self.stdout.write(self.style.MIGRATE_HEADING('\nBeginning import:'))
if settings.DEFAULT_FILE_STORAGE == 'storages.backends.s3boto3.S3Boto3Storage':
with default_storage.open('data/tStudyAreas.csv', 'r') as areas_csv_bin, \
default_storage.open('data/Kea.csv', 'r') as birds_csv_bin, \
default_storage.open('data/Transmitter actions.csv', 'r') as transmitters_csv_bin:
# boto3 opens files as binary, hence the need to convert
areas_csv = io.StringIO(areas_csv_bin.read().decode('utf-8'))
birds_csv = io.StringIO(birds_csv_bin.read().decode('utf-8'))
transmitters_csv = io.StringIO(transmitters_csv_bin.read().decode('utf-8'))
synchronise_StudyArea(self, areas_csv)
synchronise_Bird(self, birds_csv)
synchronise_BandCombo(self, transmitters_csv)
else:
with open('../data/tStudyAreas.csv', 'rt') as areas_csv, \
open('../data/Kea.csv', 'rt') as birds_csv, \
open('../data/Transmitter actions.csv', 'rt') as transmitters_csv:
synchronise_StudyArea(self, areas_csv)
synchronise_Bird(self, birds_csv)
synchronise_BandCombo(self, transmitters_csv)
self.stdout.write(self.style.SUCCESS('\nImport complete'))
评论列表
文章目录