is_polish_national_id.py

defis_polish_national_id(input_str: str) ->bool:
"""
 Verification of the correctness of the PESEL number.
 www-gov-pl.translate.goog/web/gov/czym-jest-numer-pesel?_x_tr_sl=auto&_x_tr_tl=en

 PESEL can start with 0, that's why we take str as input,
 but convert it to int for some calculations.


 >>> is_polish_national_id(123)
 Traceback (most recent call last):
 ...
 ValueError: Expected str as input, found <class 'int'>

 >>> is_polish_national_id("abc")
 Traceback (most recent call last):
 ...
 ValueError: Expected number as input

 >>> is_polish_national_id("02070803628") # correct PESEL
 True

 >>> is_polish_national_id("02150803629") # wrong month
 False

 >>> is_polish_national_id("02075503622") # wrong day
 False

 >>> is_polish_national_id("-99012212349") # wrong range
 False

 >>> is_polish_national_id("990122123499999") # wrong range
 False

 >>> is_polish_national_id("02070803621") # wrong checksum
 False
 """

# check for invalid input type
ifnotisinstance(input_str, str):
msg=f"Expected str as input, found {type(input_str)}"
raiseValueError(msg)

# check if input can be converted to int
try:
input_int=int(input_str)
exceptValueError:
msg="Expected number as input"
raiseValueError(msg)

# check number range
ifnot10100000<=input_int<=99923199999:
returnFalse

# check month correctness
month=int(input_str[2:4])

if (
monthnotinrange(1, 13) # year 1900-1999
andmonthnotinrange(21, 33) # 2000-2099
andmonthnotinrange(41, 53) # 2100-2199
andmonthnotinrange(61, 73) # 2200-2299
andmonthnotinrange(81, 93) # 1800-1899
 ):
returnFalse

# check day correctness
day=int(input_str[4:6])

ifdaynotinrange(1, 32):
returnFalse

# check the checksum
multipliers= [1, 3, 7, 9, 1, 3, 7, 9, 1, 3]
subtotal=0

digits_to_check=str(input_str)[:-1] # cut off the checksum

forindex, digitinenumerate(digits_to_check):
# Multiply corresponding digits and multipliers.
# In case of a double-digit result, add only the last digit.
subtotal+= (int(digit) *multipliers[index]) %10

checksum=10-subtotal%10

returnchecksum==input_int%10


if__name__=="__main__":
fromdoctestimporttestmod

testmod()