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 in gcbq_client.list_datasets():
            tables = list(dataset.list_tables())
            if len(tables) == 0:
                dataset.delete()
Exemplo n.º 2
0
    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

        datasets = list(gcbq_client.list_datasets())

        for dataset_list_item in 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)