Making credit card numbers match the credit card type in a schema

[DamaniN]

I've written the piece of python code below to generate some PII data. I'd like the credit card number to match the credit card type field when creating this schema. I can't find a way to specify the card_type option and reference the credit_card_network field for the record.

Code:

from mimesis.locales import Locale
from mimesis.schema import Field, Schema
from mimesis.builtins import USASpecProvider

usa = USASpecProvider()


_ = Field(locale=Locale.EN)
schema = Schema(schema=lambda: {
    "pk": _("increment"),
    "first_name": _("first_name"),
    "last_name": _("last_name"),
    "gender": _("gender"),
    "email": _("email"),
    "mobile_number": _("telephone", mask='1-###-###-####'),
    "home_number": _("telephone", mask='1-###-###-####'),
    "social_security_number": usa.ssn(),
    "credit_card_number": _("credit_card_number", card_type=_("credit_card_network")),
    "credit_card_network": _("credit_card_network"),
    "credit_card_expiration_date": _("credit_card_expiration_date"),
    "cvv": _("cvv"),
    "version": _("version"),
    "timestamp": _("timestamp", posix=False),
})
schema.to_csv(file_path='data.csv', iterations=1000)
schema.to_json(file_path='data.json', iterations=1000)
schema.to_pickle(file_path='data.obj', iterations=1000)

Error:

Traceback (most recent call last):
  File "/home/ubuntu/./gendata.py", line 25, in <module>
    schema.to_csv(file_path='data.csv', iterations=1000)
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/schema.py", line 253, in to_csv
    data = self.create(iterations)
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/schema.py", line 303, in create
    return [self._schema() for _ in range(iterations)]
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/schema.py", line 303, in <listcomp>
    return [self._schema() for _ in range(iterations)]
  File "/home/ubuntu/./gendata.py", line 18, in <lambda>
    "credit_card_number": _("credit_card_number", card_type=_("credit_card_network")),
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/schema.py", line 162, in __call__
    return self.perform(*args, **kwargs)
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/schema.py", line 124, in perform
    result = self._table[name](**kwargs)
  File "/home/ubuntu/.local/lib/python3.10/site-packages/mimesis/providers/payment.py", line 126, in credit_card_number
    raise NonEnumerableError(CardType)
mimesis.exceptions.NonEnumerableError: You should use one item of: «CardType.VISA, CardType.MASTER_CARD, CardType.AMERICAN_EXPRESS» of the object mimesis.enums.CardType

[lk-geimfari]

Instead of doing this:

"credit_card_number": _("credit_card_number", card_type=_("credit_card_network")),

do this:

from mimesis.enums import CardType
_("credit_card_number", card_type=CardType.VISA),

See: https://mimesis.name/en/master/api.html#mimesis.Payment.credit_card_number

[DamaniN]

@lk-geimfari, thanks for the suggestion, however, this will make all credit cards Visa cards. I still want random credit card numbers as part of this data; however, I want the credit_card_network field to match the type of credit card that was randomly selected. So if the schema selects a Visa card, the credit_card_network should indicate Visa. If the schema selects a MasterCard, the credit_card_network filed should be MasterCard.

How do I make the credit_card_number field match what the credit_card_network filed is set to in the same schema?

[lk-geimfari]

If you need random creadit card network just pass None: _("credit_card_number", card_type=None)

[DamaniN]

I need to have the credit card number and the type of credit card in the data. I'd like these two fields to match, since the number patterns are different for each type of credit card.