diff options
| author | David Salvisberg <david.salvisberg@seantis.ch> | 2025-03-13 14:15:54 +0100 |
|---|---|---|
| committer | Arthur de Jong <arthur@arthurdejong.org> | 2025-05-05 11:51:03 +0200 |
| commit | 0e857fb8c5ba500a80cd20553bff488bf46b7643 (patch) | |
| tree | d50fc324a77f3d142b7b3fec3dcf6bb72dcc6932 /stdnum/gr | |
| parent | bc689fdddfb9dfa5ba60ad7bd08fb2533086d3cb (diff) | |
Add type hints
Closes https://github.com/arthurdejong/python-stdnum/pull/467
Closes https://github.com/arthurdejong/python-stdnum/issues/433
Diffstat (limited to 'stdnum/gr')
| -rw-r--r-- | stdnum/gr/amka.py | 12 | ||||
| -rw-r--r-- | stdnum/gr/vat.py | 10 |
2 files changed, 13 insertions, 9 deletions
diff --git a/stdnum/gr/amka.py b/stdnum/gr/amka.py index e04efeb..bfb1ae3 100644 --- a/stdnum/gr/amka.py +++ b/stdnum/gr/amka.py @@ -41,6 +41,8 @@ datetime.date(1930, 1, 1) 'M' """ +from __future__ import annotations + import datetime from stdnum import luhn @@ -48,13 +50,13 @@ from stdnum.exceptions import * from stdnum.util import clean, isdigits -def compact(number): +def compact(number: str) -> str: """Convert the number to the minimal representation. This strips the number of any valid separators and removes surrounding whitespace.""" return clean(number, ' -').strip() -def get_birth_date(number): +def get_birth_date(number: str) -> datetime.date: """Split the date parts from the number and return the date of birth. Since only two digits are used for the year, the century may be incorrect.""" @@ -71,7 +73,7 @@ def get_birth_date(number): raise InvalidComponent() -def get_gender(number): +def get_gender(number: str) -> str: """Get the gender (M/F) from the person's AMKA.""" number = compact(number) if int(number[9]) % 2: @@ -80,7 +82,7 @@ def get_gender(number): return 'F' -def validate(number): +def validate(number: str) -> str: """Check if the number is a valid AMKA. This checks the length, formatting and check digit.""" number = compact(number) @@ -93,7 +95,7 @@ def validate(number): return number -def is_valid(number): +def is_valid(number: str) -> bool: """Check if the number is a valid AMKA.""" try: return bool(validate(number)) diff --git a/stdnum/gr/vat.py b/stdnum/gr/vat.py index dc13c0c..b83117e 100644 --- a/stdnum/gr/vat.py +++ b/stdnum/gr/vat.py @@ -32,11 +32,13 @@ Traceback (most recent call last): InvalidChecksum: ... """ +from __future__ import annotations + from stdnum.exceptions import * from stdnum.util import clean, isdigits -def compact(number): +def compact(number: str) -> str: """Convert the number to the minimal representation. This strips the number of any valid separators and removes surrounding whitespace.""" number = clean(number, ' -./:').upper().strip() @@ -47,7 +49,7 @@ def compact(number): return number -def calc_check_digit(number): +def calc_check_digit(number: str) -> str: """Calculate the check digit. The number passed should not have the check digit included.""" checksum = 0 @@ -56,7 +58,7 @@ def calc_check_digit(number): return str(checksum * 2 % 11 % 10) -def validate(number): +def validate(number: str) -> str: """Check if the number is a valid VAT number. This checks the length, formatting and check digit.""" number = compact(number) @@ -69,7 +71,7 @@ def validate(number): return number -def is_valid(number): +def is_valid(number: str) -> bool: """Check if the number is a valid VAT number.""" try: return bool(validate(number)) |