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)