def ensure_plans(self, dry_run=False): edition_to_features = self.ensure_features(dry_run=dry_run) for product_type in self.product_types: for edition in self.editions: role_slug = self.BOOTSTRAP_EDITION_TO_ROLE[edition] try: role = Role.objects.get(slug=role_slug) except ObjectDoesNotExist: logging.info("Could not find the role '%s'. Did you forget to run cchq_prbac_bootstrap?") logging.info("Aborting. You should figure this out.") return software_plan_version = SoftwarePlanVersion(role=role) product, product_rates = self.ensure_product_and_rate(product_type, edition, dry_run=dry_run) feature_rates = self.ensure_feature_rates(edition_to_features[edition], edition, dry_run=dry_run) software_plan = SoftwarePlan( name='%s Edition' % product.name, edition=edition, visibility=SoftwarePlanVisibility.PUBLIC ) if dry_run: logging.info("[DRY RUN] Creating Software Plan: %s" % software_plan) else: try: software_plan = SoftwarePlan.objects.get(name=software_plan.name) logging.info("Plan '%s' already exists. Using existing plan to add version." % software_plan.name) except ObjectDoesNotExist: software_plan.save() logging.info("Creating Software Plan: %s" % software_plan) software_plan_version.plan = software_plan software_plan_version.save() for product_rate in product_rates: software_plan_version.product_rates.add(product_rate) for feature_rate in feature_rates: software_plan_version.feature_rates.add(feature_rate) software_plan_version.save() default_product_plan = DefaultProductPlan(product_type=product.product_type, edition=edition) if dry_run: logging.info("[DRY RUN] Setting plan as default for product '%s' and edition '%s'." % (product.product_type, default_product_plan.edition)) else: try: default_product_plan = DefaultProductPlan.objects.get(product_type=product.product_type, edition=edition) logging.info("Default for product '%s' and edition '%s' already exists." % (product.product_type, default_product_plan.edition)) except ObjectDoesNotExist: default_product_plan.plan = software_plan default_product_plan.save() logging.info("Setting plan as default for product '%s' and edition '%s'." % (product.product_type, default_product_plan.edition))
def ensure_plans(self, dry_run=False): edition_to_features = self.ensure_features(dry_run=dry_run) for product_type in self.product_types: for edition in self.editions: role_slug = self.BOOTSTRAP_EDITION_TO_ROLE[edition] try: role = Role.objects.get(slug=role_slug) except ObjectDoesNotExist: logger.info("Could not find the role '%s'. Did you forget to run cchq_prbac_bootstrap?") logger.info("Aborting. You should figure this out.") return software_plan_version = SoftwarePlanVersion(role=role) product, product_rates = self.ensure_product_and_rate(product_type, edition, dry_run=dry_run) feature_rates = self.ensure_feature_rates(edition_to_features[edition], edition, dry_run=dry_run) software_plan = SoftwarePlan( name='%s Edition' % product.name, edition=edition, visibility=SoftwarePlanVisibility.PUBLIC ) if dry_run: logger.info("[DRY RUN] Creating Software Plan: %s" % software_plan.name) else: try: software_plan = SoftwarePlan.objects.get(name=software_plan.name) if self.verbose: logger.info("Plan '%s' already exists. Using existing plan to add version." % software_plan.name) except SoftwarePlan.DoesNotExist: software_plan.save() if self.verbose: logger.info("Creating Software Plan: %s" % software_plan.name) software_plan_version.plan = software_plan software_plan_version.save() for product_rate in product_rates: product_rate.save() software_plan_version.product_rates.add(product_rate) for feature_rate in feature_rates: feature_rate.save() software_plan_version.feature_rates.add(feature_rate) software_plan_version.save() if edition == SoftwarePlanEdition.ADVANCED: trials = [True, False] else: trials = [False] for is_trial in trials: default_product_plan = DefaultProductPlan(product_type=product.product_type, edition=edition, is_trial=is_trial) if dry_run: logger.info("[DRY RUN] Setting plan as default for product '%s' and edition '%s'." % (product.product_type, default_product_plan.edition)) else: try: default_product_plan = DefaultProductPlan.objects.get(product_type=product.product_type, edition=edition, is_trial=is_trial) if self.verbose: logger.info("Default for product '%s' and edition " "'%s' already exists." % ( product.product_type, default_product_plan.edition )) except ObjectDoesNotExist: default_product_plan.plan = software_plan default_product_plan.save() if self.verbose: logger.info("Setting plan as default for product '%s' and edition '%s'." % (product.product_type, default_product_plan.edition))
def ensure_plans(config, verbose, apps): DefaultProductPlan = apps.get_model('accounting', 'DefaultProductPlan') SoftwarePlan = apps.get_model('accounting', 'SoftwarePlan') SoftwarePlanVersion = apps.get_model('accounting', 'SoftwarePlanVersion') Role = apps.get_model('django_prbac', 'Role') for plan_key, plan_deets in config.items(): edition, is_trial, is_report_builder_enabled = plan_key features = _ensure_features(edition, verbose, apps) try: role = _ensure_role(plan_deets['role'], apps) except Role.DoesNotExist: return product, product_rate = _ensure_product_rate( plan_deets['product_rate_monthly_fee'], edition, verbose=verbose, apps=apps, ) feature_rates = _ensure_feature_rates( plan_deets['feature_rates'], features, edition, verbose=verbose, apps=apps, ) software_plan = SoftwarePlan( name=( (('%s Trial' % product_rate.name) if is_trial else ('%s Edition' % product_rate.name)) if product is None else product.name # TODO - remove after squashing migrations ), edition=edition, visibility=SoftwarePlanVisibility.PUBLIC ) if is_report_builder_enabled: software_plan.name = '%s - Report Builder (5 Reports)' % software_plan.name try: software_plan = SoftwarePlan.objects.get(name=software_plan.name) if verbose: log_accounting_info( "Plan '%s' already exists. Using existing plan to add version." % software_plan.name ) except SoftwarePlan.DoesNotExist: software_plan.save() if verbose: log_accounting_info("Creating Software Plan: %s" % software_plan.name) product_rate.save() software_plan_version = SoftwarePlanVersion(role=role, plan=software_plan, product_rate=product_rate) software_plan_version.save() for feature_rate in feature_rates: feature_rate.save() software_plan_version.feature_rates.add(feature_rate) software_plan_version.save() try: default_product_plan = DefaultProductPlan.objects.get( edition=edition, is_trial=is_trial, is_report_builder_enabled=is_report_builder_enabled, ) if verbose: log_accounting_info( "Default for edition '%s' with is_trial='%s' already exists." % (default_product_plan.edition, is_trial) ) except DefaultProductPlan.DoesNotExist: default_product_plan = DefaultProductPlan( edition=edition, is_trial=is_trial, is_report_builder_enabled=is_report_builder_enabled, ) finally: default_product_plan.plan = software_plan default_product_plan.save() if verbose: log_accounting_info( "Setting plan as default for edition '%s' with is_trial='%s'." % (default_product_plan.edition, is_trial) ) _clear_cache(SoftwarePlan.objects.all(), DefaultProductPlan.objects.all())