synchronise.py 文件源码

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'))