def subscribe_item(context, region=None, item=None, item_type=None, payment_type=None, resource_uuid=None, resource_name=None, **kwargs): """ """ # values of product values = { 'project_id': context.project_id, 'resource_uuid': resource_uuid, 'resource_name': resource_name, } try: # filter to get product_id products = _product_get_all(context, region=region, item=item, item_type=item_type, payment_type=payment_type) # TODO(lzyeval): check if products size is not 1 values['product_id'] = products[0]['id'] subscription_ref = db.subscription_create(context, values) db.subscription_extend(context, subscription_ref['id'], subscription_ref['created_at']) except Exception, e: # TODO(lzyeval): report raise
def subscribe_item(context, region=None, item=None, item_type=None, payment_type=None, resource_uuid=None, resource_name=None, **kwargs): """ """ # values of product values = { 'project_id': context.project_id, 'resource_uuid': resource_uuid, 'resource_name': resource_name, } try: # filter to get product_id products = _product_get_all(context, region=region, item=item, item_type=item_type, payment_type=payment_type) # TODO(lzyeval): check if products size is not 1 values['product_id'] = products[0]['id'] values['status'] = "verified" print "subscription_create", item, payment_type, values app = context.app app.info("subscribe_item:proj_id=" + str(context.project_id) + \ " name:" + str(resource_name) + \ "/" + str(item) + \ "/" + str(payment_type) + \ "/" + str(resource_uuid)) subscription_ref = db.subscription_create(context, values) db.subscription_extend(context, subscription_ref['id'], subscription_ref['created_at']) print "item subscribed." except Exception, e: # TODO(lzyeval): report print "subscribe failed:", Exception, e raise
def creating(context, subscription_id, tenant_id, item_name, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, interval_unit, interval_size, is_prepaid): app = context.app conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): app.info("wait:%s creating, but %s not running." % (str(subscription_id), item_name)) if created_at + relativedelta(minutes=10) < utils.utcnow(): app.info("%s(%s) status creating-->error" % (str(subscription_id), item_name)) db.subscription_error(context, subscription_id) # TODO(lzyeval): report else: interval_info = { interval_unit: interval_size, } app.info("%s(%s) status creating-->verify" % (str(subscription_id), item_name)) db.subscription_verify(context, subscription_id) if is_prepaid: quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) print "creating and is running", tenant_id, subscription_id, \ quantity, order_size, "\033[1;33m", price, "\033[0m" app.info("creating %s:subid=%s,tid=%s,price=%s" % (item_name, subscription_id, tenant_id, str(price))) charge(context, tenant_id, subscription_id, quantity, order_size, price) else: app.info("%s/%s/%s is_prepaid" % (tenant_id, str(subscription_id), item_name)) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def creating(context, subscription_id, tenant_id, item_name, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, interval_unit, interval_size, is_prepaid): app = context.app conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): app.info("wait:%s creating, but %s not running." % (str(subscription_id), item_name)) if created_at + relativedelta(minutes=10) < utils.utcnow(): app.info("%s(%s) status creating-->error" % (str(subscription_id), item_name)) db.subscription_error(context, subscription_id) # TODO(lzyeval): report else: interval_info = { interval_unit: interval_size, } app.info("%s(%s) status creating-->verify" % (str(subscription_id), item_name)) db.subscription_verify(context, subscription_id) if is_prepaid: quantity = conn.get_usage( resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) print "creating and is running", tenant_id, subscription_id, \ quantity, order_size, "\033[1;33m", price, "\033[0m" app.info("creating %s:subid=%s,tid=%s,price=%s" % (item_name, subscription_id, tenant_id, str(price))) charge(context, tenant_id, subscription_id, quantity, order_size, price) else: app.info("%s/%s/%s is_prepaid" % (tenant_id, str(subscription_id), item_name)) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def verified(context, subscription_id, tenant_id, resource_uuid, created_at, deleted_at, expires_at, order_unit, order_size, price, currency, region_name, item_name, interval_unit, interval_size, is_prepaid): conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): raise Exception() interval_info = { interval_unit: interval_size, } quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def verified(context, subscription_id, tenant_id, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, item_name, interval_unit, interval_size, is_prepaid): conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): # FIXME(lzyeval): raise Exception() return interval_info = { interval_unit: interval_size, } quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def creating(context, subscription_id, tenant_id, resource_uuid, created_at, deleted_at, expires_at, order_unit, order_size, price, currency, region_name, item_name, interval_unit, interval_size, is_prepaid): conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): if created_at + relativedelta(minutes=10) < utils.utcnow(): db.subscription_error(context, subscription_id) # TODO(lzyeval): report else: interval_info = { interval_unit: interval_size, } db.subscription_verify(context, subscription_id) if is_prepaid: quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def test_subscription_extend(self): self.truncate_table("regions") self.truncate_table("items") self.truncate_table("item_types") self.truncate_table("payment_types") self.truncate_table("products") self.truncate_table("subscriptions") region_ref = db.region_create(self.context, {'name': 'a_region'}) item_ref = db.item_create(self.context, {'name': 'an_item'}) item_type_ref = db.item_type_create(self.context, {'name': 'an_item_type'}) payment_type_ref = db.payment_type_create( self.context, { 'name': 'a_payment_type', 'interval_unit': 'days', 'interval_size': 12345, 'is_prepaid': False }) values = { 'region_id': region_ref.id, 'item_id': item_ref.id, 'item_type_id': item_type_ref.id, 'payment_type_id': payment_type_ref.id, 'order_unit': 'some_measure', 'order_size': 1, 'price': 11.24, 'currency': 'CNY', } product_ref = db.product_create(self.context, values) values = { 'project_id': self.context.project_id, 'product_id': product_ref.id, 'resource_uuid': self.resource_uuid, 'resource_name': self.resource_name, } expect = db.subscription_create(self.context, values) datetime_to = datetime.datetime.now() + datetime.timedelta(days=1) db.subscription_extend(self.context, expect.id, datetime_to) actual = db.subscription_get(self.context, expect.id) self.assertEqual(datetime_to.day, actual['expires_at'].day)
def verified(context, subscription_id, tenant_id, item_name, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, interval_unit, interval_size, is_prepaid): app = context.app conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid, tenant_id=tenant_id): # FIXME(lzyeval): raise Exception() app.info("%s verified, but %s not running." % (str(subscription_id), item_name)) return interval_info = { interval_unit: interval_size, } quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) print "verified", tenant_id, subscription_id, \ quantity, order_size, "\033[1;33m", price, "\033[0m" app.info("verified %s:subid=%s,tid=%s,price=%s" % (item_name, subscription_id, tenant_id, str(price))) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def creating(context, subscription_id, tenant_id, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, item_name, interval_unit, interval_size, is_prepaid): conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid): if created_at + relativedelta(minutes=10) < utils.utcnow(): db.subscription_error(context, subscription_id) # TODO(lzyeval): report else: interval_info = { interval_unit: interval_size, } db.subscription_verify(context, subscription_id) if is_prepaid: quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def test_subscription_extend(self): self.truncate_table("regions") self.truncate_table("items") self.truncate_table("item_types") self.truncate_table("payment_types") self.truncate_table("products") self.truncate_table("subscriptions") region_ref = db.region_create(self.context, {'name': 'a_region'}) item_ref = db.item_create(self.context, {'name': 'an_item'}) item_type_ref = db.item_type_create(self.context, {'name': 'an_item_type'}) payment_type_ref = db.payment_type_create(self.context, {'name': 'a_payment_type', 'interval_unit': 'days', 'interval_size': 12345, 'is_prepaid': False}) values = { 'region_id': region_ref.id, 'item_id': item_ref.id, 'item_type_id': item_type_ref.id, 'payment_type_id': payment_type_ref.id, 'order_unit': 'some_measure', 'order_size': 1, 'price': 11.24, 'currency': 'CNY', } product_ref = db.product_create(self.context, values) values = { 'project_id': self.context.project_id, 'product_id': product_ref.id, 'resource_uuid': self.resource_uuid, 'resource_name': self.resource_name, } expect = db.subscription_create(self.context, values) datetime_to = datetime.datetime.now() + datetime.timedelta(days=1) db.subscription_extend(self.context, expect.id, datetime_to) actual = db.subscription_get(self.context, expect.id) self.assertEqual(datetime_to.day, actual['expires_at'].day)
def verified(context, subscription_id, tenant_id, item_name, resource_uuid, created_at, updated_at, expires_at, order_unit, order_size, price, currency, region_name, interval_unit, interval_size, is_prepaid): app = context.app conn = driver.get_connection(item_name) if not conn.is_running(resource_uuid, tenant_id=tenant_id): # FIXME(lzyeval): raise Exception() app.info("%s verified, but %s not running." % (str(subscription_id), item_name)) return interval_info = { interval_unit: interval_size, } quantity = conn.get_usage(resource_uuid, expires_at - relativedelta(**interval_info), expires_at, order_size) print "verified", tenant_id, subscription_id, \ quantity, order_size, "\033[1;33m", price, "\033[0m" app.info("verified %s(%s/%s/%s)" \ % (subscription_id, item_name, str(price), str(expires_at))) charge(context, tenant_id, subscription_id, quantity, order_size, price) db.subscription_extend(context, subscription_id, expires_at + relativedelta(**interval_info))
def test_subscribe_item(self): self.mox.StubOutWithMock(db, 'region_get_by_name') self.mox.StubOutWithMock(db, 'item_get_by_name') self.mox.StubOutWithMock(db, 'item_type_get_by_name') self.mox.StubOutWithMock(db, 'payment_type_get_by_name') self.mox.StubOutWithMock(db, 'product_get_all') self.mox.StubOutWithMock(db, 'subscription_create') self.mox.StubOutWithMock(db, 'subscription_extend') db.region_get_by_name(self.context, self.region_name).\ AndReturn(self.region) db.item_get_by_name(self.context, self.item_name).AndReturn(self.item) db.item_type_get_by_name(self.context, self.item_type_name).\ AndReturn(self.item_type) db.payment_type_get_by_name(self.context, self.payment_type_name).\ AndReturn(self.payment_type) db.product_get_all(self.context, filters={'region_id': self.region_id, 'item_id': self.item_id, 'item_type_id': self.item_type_id, 'payment_type_id': self.payment_type_id}).\ AndReturn(self.products) db.subscription_create(self.context, {'project_id': self.context.project_id, 'product_id': self.product_id, 'resource_uuid': self.resource_uuid, 'resource_name': self.resource_name}).\ AndReturn(self.subscription) db.subscription_extend(self.context, self.subscription_id, self.created_at).AndReturn(None) self.mox.ReplayAll() result = api.subscribe_item(self.context, self.region_name, self.item_name, self.item_type_name, self.payment_type_name, self.resource_uuid, self.resource_name) self.mox.VerifyAll() self.assertEqual(result, {})