b4w dummy data improved
This commit is contained in:
@@ -1,9 +1,45 @@
|
||||
from decimal import Decimal
|
||||
|
||||
from faker import Faker
|
||||
from polyfactory.factories.pydantic_factory import ModelFactory
|
||||
from polyfactory.fields import Use
|
||||
|
||||
from src.models import B4wExport
|
||||
|
||||
DEFAULT_CATEGORY = "root_category:parent_category:leaf_category"
|
||||
|
||||
|
||||
def _amount() -> Decimal:
|
||||
return B4wExportFactory.__faker__.pydecimal(
|
||||
left_digits=6,
|
||||
right_digits=2,
|
||||
positive=True,
|
||||
min_value=Decimal("0.01"),
|
||||
)
|
||||
|
||||
|
||||
def _sentence() -> str:
|
||||
return B4wExportFactory.__faker__.sentence(
|
||||
nb_words=8,
|
||||
variable_nb_words=False,
|
||||
)
|
||||
|
||||
|
||||
def _names() -> str:
|
||||
return B4wExportFactory.__faker__.company()
|
||||
|
||||
|
||||
def _iban() -> str:
|
||||
return B4wExportFactory.__faker__.iban()
|
||||
|
||||
|
||||
class B4wExportFactory(ModelFactory[B4wExport]):
|
||||
__faker__ = Faker("de_DE")
|
||||
__use_examples__ = True
|
||||
|
||||
Amt = Use(_amount)
|
||||
RmtInf = Use(_sentence)
|
||||
Notes = Use(_sentence)
|
||||
RmtdNm = Use(_names)
|
||||
OwnrAcctIBAN = Use(_iban)
|
||||
RmtdAcctIBAN = Use(_iban)
|
||||
|
||||
Reference in New Issue
Block a user