import factory
from api import models
from api import db


class DesignerFactory(factory.Factory):
    class Meta:
        model = models.Designer

    name = "Spiderman"
    username = "spiderman"
    territory = "US"
    currency = "USD"


class ContractCustomerFactory(factory.Factory):
    class Meta:
        model = models.ContractCustomer

    pricepremium = 0


class InquiryFactory(factory.Factory):
    class Meta:
        model = models.Inquiry

    price_retouch = None
    pricepremium = 0


class MarketFactory(factory.Factory):
    class Meta:
        model = models.Market