from typing import Iterable, Union
from django.contrib.auth import get_user_model
from shuup.apps.provides import get_provide_objects
from shuup.core.models import Product, Shop, Supplier
from shuup.core.pricing import TaxfulPrice, TaxlessPrice
UserModel = get_user_model()
[docs]
class ProductSubscriptionOption:
value = None
label = None
price = None
description = None
[docs]
def __init__(
self,
label: str,
value: str,
price: Union[TaxfulPrice, TaxlessPrice],
description: str = None,
):
self.label = label
self.value = value
self.price = price
self.description = description
[docs]
class ProductSubscriptionContext:
shop = None
product = None
supplier = None
user = None
[docs]
def __init__(
self,
shop: Shop,
product: Product,
supplier: Supplier = None,
user: UserModel = None,
**kwargs,
):
self.shop = shop
self.product = product
self.supplier = supplier
self.user = user
[docs]
class BaseProductSubscriptionOptionProvider:
[docs]
@classmethod
def get_subscription_options(cls, context: ProductSubscriptionContext) -> Iterable[ProductSubscriptionOption]:
raise NotImplementedError()
[docs]
def get_product_subscription_options(
context: ProductSubscriptionContext, **kwargs
) -> Iterable[ProductSubscriptionOption]:
for product_subscription_option_provider in get_provide_objects("product_subscription_option_provider"):
if not issubclass(product_subscription_option_provider, BaseProductSubscriptionOptionProvider):
continue
yield from product_subscription_option_provider.get_subscription_options(context)