generators.py 513 B

12345678910111213
  1. from model_bakery import baker
  2. from model_bakery.random_gen import gen_slug, gen_datetime, gen_integer
  3. def currency_code():
  4. return "USD"
  5. baker.generators.add("organizations.fields.SlugField", gen_slug)
  6. baker.generators.add("djstripe.fields.StripeCurrencyCodeField", currency_code)
  7. baker.generators.add("djstripe.fields.StripeIdField", gen_slug)
  8. baker.generators.add("djstripe.fields.StripeDateTimeField", gen_datetime)
  9. baker.generators.add("djstripe.fields.StripeQuantumCurrencyAmountField", gen_integer)