insee_number_translator/main.py
2019-07-10 23:13:38 +02:00

124 lines
3.5 KiB
Python

import calendar
import datetime
from departments import DEPARTMENTS
from cities import CITIES
from countries import COUNTRIES, CONTINENTS
class InseeData:
def __init__(self, insee_number: str):
self.insee_number = insee_number
if len(insee_number) != 15:
raise ValueError("insee number must be 15 characters")
self._gender = insee_number[0]
self._year = insee_number[1:3]
self.month = int(insee_number[3:5])
department = insee_number[5:7]
city = insee_number[7:10]
if department in ["97", "98"]:
department = insee_number[5:8]
city = insee_number[8:10]
self.foreign = False
if department == "99":
self.foreign = True
self._country = city
else:
self._department = department
self._city = city
self.order_of_birth = int(insee_number[10:13])
self.control_key = int(insee_number[13:])
@property
def is_valid(self):
code = 97 - (
int(self.insee_number.replace("2A", "19").replace("2B", "18")[:-2]) % 97
)
return code == self.control_key
@property
def year(self):
year = int("20" + self._year)
date = datetime.date(year, self.month, 1)
if date < datetime.date.today():
return year
return year - 100
@property
def gender(self):
if self._gender == "1":
return "Male"
return "Female"
@property
def city(self):
if self.foreign:
return "unknown"
return self._city
@property
def country(self):
if self.foreign:
return self._country
return "FR"
@property
def department(self):
if self.foreign:
return "unknown"
return self._department
def __str__(self):
message = []
if self.is_valid:
message.append("The number is valid")
else:
message.append("The number is invalid")
message.append(
f"You're a {self.gender}, born in {calendar.month_name[self.month]}, probably in {self.year}."
)
if self.foreign:
word = "country"
line = "You're born outside France, "
country = COUNTRIES.get("99" + self.country, [])
continent = CONTINENTS.get(self.country[0])
if len(country) > 1:
line += f"probably in one of these countries/territories: {', '.join(country)} ({continent})"
elif len(country) == 1:
line += f"in {country[0]} ({continent})"
else:
line += f"in an unknown country numbered {self.country}"
message.append(line)
else:
word = "city"
city = CITIES.get(self.department + self.city)
department_nice = DEPARTMENTS[self.department]
if city:
message.append(
f"You're born in {city['name']}, {city['zip_code']} ({department_nice}), France."
)
else:
message.append(
f"You're born in a unknown city numbered {self.city}, which is located in {department_nice}, in France."
)
message.append(
f"You're the {self.order_of_birth}th to be born in this {word} on this month."
)
return "\n".join(message)
def main():
# insee_number = "168127982980507"
insee_number = "269059913116714"
data = InseeData(insee_number)
print(data)
main()