def tearDown(self): client = Client('test') client.delete_dataset() client = StorageClient() bucket = client.bucket() for blob in bucket.list_blobs(prefix=self.storage_prefix): blob.delete()
def handle(self, *args, **kwargs): if os.environ['DJANGO_SETTINGS_MODULE'] != \ 'openprescribing.settings.test': raise CommandError('Command must run with test settings') gcbq_client = Client().gcbq_client for dataset_list_item in gcbq_client.list_datasets(): dataset_ref = dataset_list_item.reference tables = list(gcbq_client.list_tables(dataset_ref)) if len(tables) == 0: gcbq_client.delete_dataset(dataset_ref)