Source code for mimesis.builtins.pl

# -*- coding: utf-8 -*-

"""Specific data provider for Poland (pl)."""

from typing import Optional

from mimesis.builtins.base import BaseSpecProvider
from mimesis.enums import Gender
from mimesis.locales import Locale
from mimesis.providers import Datetime
from mimesis.typing import DateTime, Seed

__all__ = ["PolandSpecProvider"]


[docs]class PolandSpecProvider(BaseSpecProvider): """Class that provides special data for Poland (pl)."""
[docs] def __init__(self, seed: Optional[Seed] = None) -> None: """Initialize attributes.""" super().__init__(locale=Locale.PL, seed=seed)
[docs] class Meta: """The name of the provider.""" name = "poland_provider"
[docs] def nip(self) -> str: """Generate random valid 10-digit NIP. :return: Valid 10-digit NIP """ nip_digits = [int(d) for d in str(self.random.randint(101, 998))] nip_digits += [self.random.randint(0, 9) for _ in range(6)] nip_coefficients = (6, 5, 7, 2, 3, 4, 5, 6, 7) sum_v = sum([nc * nd for nc, nd in zip(nip_coefficients, nip_digits)]) checksum_digit = sum_v % 11 if checksum_digit > 9: return self.nip() nip_digits.append(checksum_digit) return "".join(str(d) for d in nip_digits)
[docs] def pesel( self, birth_date: Optional[DateTime] = None, gender: Optional[Gender] = None ) -> str: """Generate random 11-digit PESEL. :param birth_date: Initial birth date (optional) :param gender: Gender of person :return: Valid 11-digit PESEL """ date_object = birth_date if not date_object: date_object = Datetime().datetime(1940, 2018) year = date_object.date().year month = date_object.date().month day = date_object.date().day pesel_digits = [int(d) for d in str(year)][-2:] if 1800 <= year <= 1899: month += 80 elif 2000 <= year <= 2099: month += 20 elif 2100 <= year <= 2199: month += 40 elif 2200 <= year <= 2299: month += 60 pesel_digits += [int(d) for d in "{:02d}".format(month)] pesel_digits += [int(d) for d in "{:02d}".format(day)] series_number = self.random.randint(0, 999) pesel_digits += [int(d) for d in "{:03d}".format(series_number)] if gender == Gender.MALE: gender_digit = self.random.choice((1, 3, 5, 7, 9)) elif gender == Gender.FEMALE: gender_digit = self.random.choice((0, 2, 4, 6, 8)) else: gender_digit = self.random.choice(range(10)) pesel_digits.append(gender_digit) pesel_coeffs = (9, 7, 3, 1, 9, 7, 3, 1, 9, 7) sum_v = sum([nc * nd for nc, nd in zip(pesel_coeffs, pesel_digits)]) checksum_digit = sum_v % 10 pesel_digits.append(checksum_digit) return "".join(str(d) for d in pesel_digits)
[docs] def regon(self) -> str: """Generate random valid 9-digit REGON. :return: Valid 9-digit REGON """ regon_coeffs = (8, 9, 2, 3, 4, 5, 6, 7) regon_digits = [self.random.randint(0, 9) for _ in range(8)] sum_v = sum([nc * nd for nc, nd in zip(regon_coeffs, regon_digits)]) checksum_digit = sum_v % 11 if checksum_digit > 9: checksum_digit = 0 regon_digits.append(checksum_digit) return "".join(str(d) for d in regon_digits)