formencode-cvs Mailing List for FormEncode for Python (Page 27)

SourceForge Headquarters 1320 Columbia Street Suite 310 San Diego, CA 92101 +1 (858) 422-6466

Author: rflosi
Date: 2005-11-29 23:04:57 +0000 (Tue, 29 Nov 2005)
New Revision: 1327

Added:
   FormEncode/trunk/tests/test_cc_expires.py
Modified:
   FormEncode/trunk/formencode/validators.py
Log:
added a CreditCardExpires validator and cooresponding test file; usage is like CreditCardValidator; default field names are: ccExpiresMonth, ccExpiresYear; Currently only supports 4 digit years; errors could be improved if desired

Modified: FormEncode/trunk/formencode/validators.py
===================================================================

--- FormEncode/trunk/formencode/validators.py	2005-11-28 09:16:53 UTC (rev 1326)
+++ FormEncode/trunk/formencode/validators.py	2005-11-29 23:04:57 UTC (rev 1327)
@@ -2159,6 +2159,80 @@
                 ('1800', 15)],
             }
 
+class CreditCardExpires(FormValidator):
+    """
+    Checks that credit card expiration date is valid relative to 
+    the current date.
+
+    You pass in the name of the field that has the credit card
+    expiration month and the field with the credit card expiration 
+    year.  
+
+    ::
+
+        >>> ed = CreditCardExpires()
+        >>> ed.to_python({'ccExpiresMonth': '11', 'ccExpiresYear': '2250'})
+        {'ccExpiresYear': '2250', 'ccExpiresMonth': '11'}
+        >>> ed.to_python({'ccExpiresMonth': '10', 'ccExpiresYear': '2005'})
+        Traceback (most recent call last):
+            ...
+        Invalid: ccExpiresMonth: Invalid Expiration Date<br>
+        ccExpiresYear: Invalid Expiration Date
+    """
+
+    validate_partial_form = True
+
+    cc_expires_month_field = 'ccExpiresMonth'
+    cc_expires_year_field = 'ccExpiresYear'
+    __unpackargs__ = ('cc_expires_month_field', 'cc_expires_year_field')
+
+    datetime_module = None
+
+    messages = {
+        'notANumber': "Please enter numbers only for month and year",
+        'invalidNumber': "Invalid Expiration Date",
+        }
+
+    def validate_partial(self, field_dict, state):
+        if not field_dict.get(self.cc_expires_month_field, None) \
+           or not field_dict.get(self.cc_expires_year_field, None):
+            return None
+        self.validate_python(field_dict, state)
+
+    def validate_python(self, field_dict, state):
+        errors = self._validateReturn(field_dict, state)
+        if errors:
+            error_list = errors.items()
+            error_list.sort()
+            raise Invalid(
+                '<br>\n'.join(["%s: %s" % (name, value)
+                               for name, value in error_list]),
+                field_dict, state, error_dict=errors)
+        
+    def _validateReturn(self, field_dict, state):
+        ccExpiresMonth = field_dict[self.cc_expires_month_field].strip()
+        ccExpiresYear = field_dict[self.cc_expires_year_field].strip()
+
+        try:
+            ccExpiresMonth = int(ccExpiresMonth)
+            ccExpiresYear = int(ccExpiresYear)
+            dt_mod = import_datetime(self.datetime_module)
+            now = datetime_now(dt_mod)
+            today = datetime_makedate(dt_mod, now.year, now.month, now.day)
+            next_month = (ccExpiresMonth % 12) + 1
+            if next_month == 1:
+                next_month_year = ccExpiresYear + 1
+            else:
+                next_month_year = ccExpiresYear
+            expires_date = datetime_makedate(dt_mod, next_month_year, next_month, 1)
+            assert expires_date > today
+        except ValueError:
+            return {self.cc_expires_month_field: self.message('notANumber', state),
+                    self.cc_expires_year_field: self.message('notANumber', state)}
+        except AssertionError:
+            return {self.cc_expires_month_field: self.message('invalidNumber', state),
+                    self.cc_expires_year_field: self.message('invalidNumber', state)}
+
 __all__ = []
 for name, value in globals().items():
     if isinstance(value, type) and issubclass(value, Validator):

Added: FormEncode/trunk/tests/test_cc_expires.py
===================================================================
--- FormEncode/trunk/tests/test_cc_expires.py	2005-11-28 09:16:53 UTC (rev 1326)
+++ FormEncode/trunk/tests/test_cc_expires.py	2005-11-29 23:04:57 UTC (rev 1327)
@@ -0,0 +1,19 @@
+from formencode.validators import CreditCardExpires, Invalid
+
+ed = CreditCardExpires()
+
+def validate(month, year):
+    try:
+        ed.validate_python({'ccExpiresMonth': month,
+                            'ccExpiresYear': year}, None)
+    except Invalid, e:
+        return e.unpack_errors()['ccExpiresMonth']
+
+messages = CreditCardExpires._messages
+
+def test_ed():
+    assert validate('11', '2250') is None
+    assert validate('11', 'test') == messages['notANumber']
+    assert validate('test', '2250') == messages['notANumber']
+    assert validate('10', '2005') == messages['invalidNumber']
+    assert validate('10', '05') == messages['invalidNumber']





2004	Jan	Feb	Mar (1)	Apr	May	Jun	Jul	Aug	Sep	Oct	Nov (10)	Dec (4)
2005	Jan (1)	Feb (8)	Mar (8)	Apr (4)	May (19)	Jun (1)	Jul (1)	Aug (18)	Sep (18)	Oct (19)	Nov (75)	Dec (80)
2006	Jan (86)	Feb (61)	Mar (60)	Apr (47)	May (39)	Jun (16)	Jul (30)	Aug (13)	Sep (13)	Oct (21)	Nov (1)	Dec (10)
2007	Jan (2)	Feb (7)	Mar (9)	Apr (3)	May (9)	Jun (4)	Jul (1)	Aug (2)	Sep	Oct (12)	Nov (1)	Dec (7)
2008	Jan	Feb (2)	Mar (14)	Apr (9)	May (23)	Jun (4)	Jul	Aug (13)	Sep (8)	Oct (15)	Nov (40)	Dec (14)
2009	Jan	Feb (4)	Mar (10)	Apr (2)	May (2)	Jun (1)	Jul	Aug	Sep	Oct	Nov	Dec
2010	Jan	Feb	Mar	Apr	May (1)	Jun	Jul	Aug	Sep	Oct	Nov	Dec
2011	Jan	Feb	Mar	Apr	May (1)	Jun	Jul	Aug	Sep	Oct	Nov	Dec

formencode-cvs Mailing List for FormEncode for Python (Page 27)

formencode-cvs — CVS update notifications