Skip to content

Commit

Permalink
addressing PR comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Marcusk19 committed Jan 29, 2024
1 parent 855b1e9 commit 5e06830
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,7 @@

def upgrade(op, tables, tester):
op.add_column("organizationrhskus", sa.Column("quantity", sa.Integer(), nullable=True))
pass


def downgrade(op, tables, tester):
op.drop_column("organizationrhskus", "quantity")
pass
7 changes: 3 additions & 4 deletions data/model/organization_skus.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,9 @@ def get_org_subscriptions(org_id):

def bind_subscription_to_org(subscription_id, org_id, user_id, quantity=1):
try:
with db_transaction():
return OrganizationRhSkus.create(
subscription_id=subscription_id, org_id=org_id, user_id=user_id, quantity=quantity
)
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:
Expand Down
4 changes: 1 addition & 3 deletions endpoints/api/billing.py
Original file line number Diff line number Diff line change
Expand Up @@ -984,9 +984,6 @@ def post(self, orgname):
subscription_id = subscription.get("subscription_id")
if subscription_id is None:
break
# quantity = subscription.get("quantity")
# if quantity is None:
# quantity = 1
user = get_authenticated_user()
account_number = marketplace_users.get_account_number(user)
subscriptions = marketplace_subscriptions.get_list_of_subscriptions(account_number)
Expand All @@ -1000,6 +997,7 @@ def post(self, orgname):
for subscription in subscriptions:
if subscription["id"] == subscription_id:
quantity = subscription["quantity"]
break
try:
model.organization_skus.bind_subscription_to_org(
user_id=user.id,
Expand Down

0 comments on commit 5e06830

Please sign in to comment.