def _delete_enrollment_table()

in jetstream/cli.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

    def _delete_enrollment_table(self, config: AnalysisConfiguration) -> None:
        """Deletes all enrollment table associated with the experiment."""
        print(f"Delete enrollment table for {config.experiment.normandy_slug}")

        if config.experiment.is_private or config.experiment.dataset_id is not None:
            # private experiments must override the dataset and set it to a private dataset
            dataset_id = config.experiment.dataset_id
        else:
            dataset_id = self.dataset_id

        client = BigQueryClient(project=self.project_id, dataset=dataset_id)
        normalized_slug = bq_normalize_name(config.experiment.normandy_slug)
        enrollments_table = f"{self.project_id}.{dataset_id}.enrollments_{normalized_slug}"
        client.delete_table(enrollments_table)