def setUp(self): self.postgresql = testing.postgresql.Postgresql() self.database = Database(url=self.postgresql.url()) self.database.engine.execute('create extension postgis;') self.database.engine.execute('create extension postgis_topology;') self.database.create() self.root_dir = os.path.dirname(__file__) file_location = os.path.join(self.root_dir, 'data/sample-feed.zip') md5 = hashlib.md5(open(file_location, 'rb').read()).hexdigest() feed_file=FeedFile(md5sum=md5, file_url=file_location) database_load_versioned(feed_file, self.postgresql.url())
def serial_run(sources, database): for source in sources: database_load_versioned(db_url=database, feed_file=source)