Add field for greek Social Security Number (AMKA)

docs/changelog.rst

@@ -11,6 +11,7 @@ New flavors:
 New fields for existing flavors:
 
 - `NLLicensePlateField` in NL flavor.
+- `GRSocialSecurityNumberField` (AMKA) in GR flavore
 
 Modifications to existing flavors:
 

localflavor/gr/forms.py

@@ -1,11 +1,14 @@
 """Greek-specific forms helpers."""
+import datetime
 import re
 
 from django.core.validators import EMPTY_VALUES
 from django.forms import Field, RegexField, ValidationError
 from django.utils.encoding import force_text
 from django.utils.translation import ugettext_lazy as _
 
+from localflavor.generic.checksums import luhn
+
 
 class GRPostalCodeField(RegexField):
     """
@@ -60,3 +63,44 @@ def clean(self, value):
         if mod != check:
             raise ValidationError(self.error_messages['invalid'])
         return val
+
+
+class GRSocialSecurityNumberCodeField(RegexField):
+    """
+    Greek social security number (AMKA) field.
+
+    The allow_test_value option can be used to enable the usage of the
+    non valid 00000000000 (11 zeros) value for testing and development
+    """
+
+    default_error_messages = {
+        'invalid': _('Enter a valid greek social security number (AMKA - 11 digits).'),
+    }
+
+    def __init__(self, allow_test_value=False, *args, **kwargs):
+        self.allow_test_value = allow_test_value
+        super(GRSocialSecurityNumberCodeField, self).__init__(r'^[0-9\s\-]+$', *args, **kwargs)
+
+    def check_date(self, val):
+        try:
+            datetime.datetime.strptime(val[:6], '%d%m%y')
+        except:
+            raise ValidationError(self.error_messages['invalid'])
+
+    def clean(self, value):
+        super(GRSocialSecurityNumberCodeField, self).clean(value)
+        if value in self.empty_values:
+            return self.empty_value
+        val = re.sub('[\-\s]', '', force_text(value))
+        if not val or len(val) < 11:
+            raise ValidationError(self.error_messages['invalid'])
+        if self.allow_test_value and val == '00000000000':
+            return val
+        if not all(char.isdigit() for char in val):
+            raise ValidationError(self.error_messages['invalid'])
+
+        self.check_date(val)
+        if not luhn(val):
+            raise ValidationError(self.error_messages['invalid'])
+
+        return val

tests/test_gr.py

@@ -1,6 +1,6 @@
 from django.test import SimpleTestCase
 
-from localflavor.gr.forms import GRPostalCodeField, GRTaxNumberCodeField
+from localflavor.gr.forms import GRPostalCodeField, GRSocialSecurityNumberCodeField, GRTaxNumberCodeField
 
 
 class GRLocalFlavorTests(SimpleTestCase):
@@ -52,3 +52,25 @@ def test_GRPostalCodeField(self):
             'b231a': error,
         }
         self.assertFieldOutput(GRPostalCodeField, valid, invalid)
+
+    def test_GRSocialSecurityNumberCodeField(self):
+        """It's not easy finding valid AMKAs in the internet, these were created by hand"""
+        error = ['Enter a valid greek social security number (AMKA - 11 digits).']
+
+        valid = {
+            '20107201921': '20107201921',
+            '14059202672': '14059202672',
+            '03059302467': '03059302467',
+        }
+
+        invalid = {
+            '20207201920': error,
+            '12345678901': error,
+            'ab345678901': error,
+            '00000000000': error,
+            '1': error,
+            '123': error,
+            'aaa': error,
+        }
+        self.assertFieldOutput(GRSocialSecurityNumberCodeField, valid, invalid)
+        self.assertFieldOutput(GRSocialSecurityNumberCodeField, {'00000000000': '00000000000',}, {}, field_kwargs={'allow_test_value': True})