def setUp(self): self.user = User() self.user.save() token = Token.objects.create(user=self.user) token.save() self.valid_token = 'Token ' + str(self.user.auth_token) AddressBaseBasicImporter().import_csv( self._sample_data_file('addressbase_basic_barnet_sample.csv')) PostcodeGssCodeImporter().import_postcode_gss_codes( self._sample_data_file('NSPL_barnet_sample.csv')) LocalAuthoritiesImporter().import_local_authorities( self._sample_data_file('local_authorities_sample.nt'))
def rerun_nspl_import(apps, schema_editor): if settings.DEBUG or settings.TESTING: print 'not in production - skipping re-import of NSPL' else: destination_dir = dest_dir('/tmp/postcode_gss_codes/') filepath = PostcodeGssCodeDownloader().download(destination_dir) if isinstance(filepath, list): filepath = filepath[0] files = ZipExtractor(filepath).unzip_if_needed('.*NSPL.*\.csv') for path in files: print 'importing ' + path PostcodeGssCodeImporter().import_postcode_gss_codes(path)
def _import(self, downloaded_file): importer = PostcodeGssCodeImporter() importer.import_postcode_gss_codes(downloaded_file)
def _importer(self): return PostcodeGssCodeImporter()
def import_csv(filename): if not os.access(filename, os.R_OK): raise CommandError('CSV file could not be read') importer = PostcodeGssCodeImporter() importer.import_postcode_gss_codes(filename)