mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
dbfb47d6fb
* marketplace: change redhatsubscriptions model to return a list of customer ids (PROJQUAY-7129) * Use list comprehension in model and expect to return None if no rows are returned * fix behavior for reconciler and fix unit test for reconciler
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import logging
|
|
|
|
from data import model
|
|
from data.database import RedHatSubscriptions
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_web_customer_ids(user_id):
|
|
try:
|
|
customer_ids = [
|
|
customer.account_number
|
|
for customer in RedHatSubscriptions.select().where(
|
|
RedHatSubscriptions.user_id == user_id
|
|
)
|
|
]
|
|
if len(customer_ids) == 0:
|
|
return None
|
|
return customer_ids
|
|
except RedHatSubscriptions.DoesNotExist:
|
|
return None
|
|
|
|
|
|
def save_web_customer_id(user, web_customer_id):
|
|
try:
|
|
return RedHatSubscriptions.create(user_id=user.id, account_number=web_customer_id)
|
|
except model.DataModelException as ex:
|
|
logger.error("Problem saving account number for %s: %s", user.username, ex)
|
|
|
|
|
|
def update_web_customer_id(user, web_customer_id):
|
|
try:
|
|
query = RedHatSubscriptions.update(
|
|
{RedHatSubscriptions.account_number: web_customer_id}
|
|
).where(RedHatSubscriptions.user_id == user.id)
|
|
query.execute()
|
|
except model.DataModelException as ex:
|
|
logger.error("Problem updating customer id for %s: %s", user.username, ex)
|
|
|
|
|
|
def remove_web_customer_id(user, web_customer_id):
|
|
try:
|
|
customer_id = RedHatSubscriptions.get(
|
|
RedHatSubscriptions.user_id == user.id,
|
|
RedHatSubscriptions.account_number == web_customer_id,
|
|
)
|
|
return customer_id.delete_instance()
|
|
except model.DataModelException as ex:
|
|
logger.error("Problem removing customer id for %s: %s", user.username, ex)
|