def new_customer(cls, email, password, creator_id, detailed_info=None, comment="New customer",
withdraw_period=None, make_prod=False, customer_type=None, promo_code=None, locale=None):
from model import Tariff, Account, CustomerHistory
customer = cls()
customer.email = email
customer.password = cls.password_hashing(password) if password else ""
customer.deleted = None
customer.created = utcnow().datetime
customer.email_confirmed = False
customer.blocked = False
customer.customer_mode = cls.CUSTOMER_PRODUCTION_MODE if make_prod else cls.CUSTOMER_TEST_MODE
customer.customer_type = customer_type if customer_type else cls.CUSTOMER_TYPE_PRIVATE_PERSON
customer.withdraw_period = withdraw_period or conf.customer.default_withdraw_period
customer.auto_withdraw_enabled = conf.payments.auto_withdraw_enable
customer.auto_withdraw_balance_limit = conf.payments.auto_withdraw_balance_limit
customer.auto_withdraw_amount = conf.payments.auto_withdraw_amount
customer.locale = locale or conf.customer.default_locale
customer.os_dashboard = conf.customer.default_openstack_dashboard
default_tariff = Tariff.get_default().first()
if not default_tariff:
default_tariff = Tariff.query.filter_by(mutable=False).first() or Tariff.query.filter_by().first()
if not default_tariff:
raise errors.TariffNotFound()
customer.tariff_id = default_tariff.tariff_id
customer.balance_limit = conf.customer.balance_limits.get(default_tariff.currency,
conf.customer.balance_limits.default)
db.session.add(customer)
db.session.flush()
customer.init_subscriptions()
template = 'customer' if make_prod else 'test_customer'
customer.quota_init(template)
customer.accounts.append(Account.create(default_tariff.currency, customer, creator_id, comment))
if promo_code is not None:
PromoCode.new_code(promo_code, customer.customer_id)
CustomerHistory.new_customer(customer, creator_id, comment)
if detailed_info:
customer.create_info(customer.customer_type, detailed_info)
auto_report_task = ScheduledTask(cls.AUTO_REPORT_TASK, customer.customer_id, customer.withdraw_period)
db.session.add(auto_report_task)
logbook.info("New customer created: {}", customer)
if not make_prod:
currency = customer.tariff.currency.upper()
initial_balance = conf.customer.test_customer.balance.get(currency)
logbook.debug("Initial balance for customer {}: {} {}", customer, initial_balance, currency)
if initial_balance:
customer.modify_balance(Decimal(initial_balance), currency, None, "Initial test balance")
return customer
def get_default(self):
"""
Get description of default tariff
:return dict tariff_info: Returns dict with tariff description.
"""
tariff = Tariff.get_default().first()
if not tariff:
raise errors.TariffNotFound()
return {"tariff_info": display(tariff)}
def check_default_tariff():
from model import Tariff
return bool(Tariff.get_default().first())
def check_default_tariff():
from model import Tariff
return bool(Tariff.get_default().first())