mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
f69716b867
* marketplace: splittable sku for MW02702 (PROJQUAY-8151) * Alembic migration to drop unique constraint on the orgrhsubscriptions table * Can split sub quantities of MW02702 across multiple orgs * Can specify quantity for the MW02702 SKU across orgs on react UI * Update angular UI to allow user to specify quantities for MW02702
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
import logging
|
|
|
|
import peewee
|
|
|
|
from data import model
|
|
from data.database import OrganizationRhSkus, db_transaction
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_org_subscriptions(org_id):
|
|
try:
|
|
query = OrganizationRhSkus.select().where(OrganizationRhSkus.org_id == org_id)
|
|
return query
|
|
except OrganizationRhSkus.DoesNotExist:
|
|
return None
|
|
|
|
|
|
def bind_subscription_to_org(subscription_id, org_id, user_id, quantity=1):
|
|
try:
|
|
return OrganizationRhSkus.create(
|
|
subscription_id=subscription_id, org_id=org_id, user_id=user_id, quantity=quantity
|
|
)
|
|
except model.DataModelException as ex:
|
|
logger.error("Problem binding subscription to org %s: %s", org_id, ex)
|
|
except peewee.IntegrityError:
|
|
raise model.OrgSubscriptionBindingAlreadyExists()
|
|
|
|
|
|
def subscription_bound_to_org(subscription_id):
|
|
# lookup row in table matching subscription_id, if there is no row return false, otherwise return true
|
|
# this function is used to check if a subscription is bound to an org or
|
|
try:
|
|
query = (
|
|
OrganizationRhSkus.select()
|
|
.where(OrganizationRhSkus.subscription_id == subscription_id)
|
|
.dicts()
|
|
)
|
|
if query.__len__() > 0:
|
|
return True, query
|
|
return False, None
|
|
except OrganizationRhSkus.DoesNotExist:
|
|
return False, None
|
|
|
|
|
|
def remove_subscription_from_org(org_id, subscription_id):
|
|
query = OrganizationRhSkus.delete().where(
|
|
OrganizationRhSkus.org_id == org_id,
|
|
OrganizationRhSkus.subscription_id == subscription_id,
|
|
)
|
|
query.execute()
|
|
|
|
|
|
def remove_all_owner_subscriptions_from_org(user_id, org_id):
|
|
try:
|
|
query = OrganizationRhSkus.delete().where(
|
|
OrganizationRhSkus.user_id == user_id,
|
|
OrganizationRhSkus.org_id == org_id,
|
|
)
|
|
query.execute()
|
|
except model.DataModelException as ex:
|
|
raise model.DataModelException(ex)
|
|
|
|
|
|
def get_bound_subscriptions(subscription_id):
|
|
try:
|
|
query = OrganizationRhSkus.select().where(
|
|
OrganizationRhSkus.subscription_id == subscription_id
|
|
)
|
|
return query
|
|
except OrganizationRhSkus.DoesNotExist:
|
|
return None
|