Initial work on #101 - Create a filter that just quietly removes checksums since the core api doesn't care for them, update get_balances and find_transactions with that filter

Author: plenarius

iota/commands/core/find_transactions.py

@@ -7,7 +7,7 @@
 
 from iota import Address, Tag, TransactionHash
 from iota.commands import FilterCommand, RequestFilter, ResponseFilter
-from iota.filters import Trytes
+from iota.filters import Trytes, AddressNoChecksum
 
 __all__ = [
   'FindTransactionsCommand',
@@ -43,7 +43,7 @@ def __init__(self):
             f.Array
           | f.FilterRepeater(
                 f.Required
-              | Trytes(result_type=Address)
+              | AddressNoChecksum(result_type=Address)
               | f.Unicode(encoding='ascii', normalize=False)
             )
         ),

iota/commands/core/get_balances.py

@@ -6,7 +6,7 @@
 
 from iota import Address
 from iota.commands import FilterCommand, RequestFilter, ResponseFilter
-from iota.filters import Trytes
+from iota.filters import Trytes, AddressNoChecksum
 
 __all__ = [
   'GetBalancesCommand',
@@ -37,7 +37,7 @@ def __init__(self):
           | f.Array
           | f.FilterRepeater(
                 f.Required
-              | Trytes(result_type=Address)
+              | AddressNoChecksum(result_type=Address)
               | f.Unicode(encoding='ascii', normalize=False)
             )
         ),

iota/filters.py

@@ -9,7 +9,6 @@
 
 from iota import Address, TryteString, TrytesCompatible
 
-
 class GeneratedAddress(f.BaseFilter):
   """
   Validates an incoming value as a generated :py:class:`Address` (must
@@ -69,10 +68,12 @@ class Trytes(f.BaseFilter):
   """
   Validates a sequence as a sequence of trytes.
   """
-  CODE_NOT_TRYTES   = 'not_trytes'
-  CODE_WRONG_FORMAT = 'wrong_format'
+  ADDRESS_BAD_CHECKSUM   = 'address_bad_checksum'
+  CODE_NOT_TRYTES        = 'not_trytes'
+  CODE_WRONG_FORMAT      = 'wrong_format'
 
   templates = {
+    ADDRESS_BAD_CHECKSUM: 'The checksum for this address is invalid.',
     CODE_NOT_TRYTES: 'This value is not a valid tryte sequence.',
     CODE_WRONG_FORMAT: 'This value is not a valid {result_type}.',
   }
@@ -142,3 +143,67 @@ def _apply(self, value):
           'result_type': self.result_type.__name__,
         },
       )
+
+
+class AddressNoChecksum(Trytes):
+  """
+  Validates a sequence as a sequence of trytes.
+  """
+
+  def _apply(self, value):
+    # noinspection PyTypeChecker
+    value =\
+      self._filter(
+        filter_chain = f.Type((binary_type, bytearray, text_type, TryteString)),
+        value = value,
+      ) # type: TrytesCompatible
+
+    if self._has_errors:
+      return None
+
+    # If the incoming value already is an Address then we just make sure it has no checksum
+    if isinstance(value, Address):
+      if value.checksum and not value.is_checksum_valid():
+        return self._invalid_value(
+        value     = value,
+        reason    = self.ADDRESS_BAD_CHECKSUM,
+        exc_info  = True,
+
+        template_vars = {
+          'result_type': self.ADDRESS_BAD_CHECKSUM,
+        },
+      )
+      return Address(value.address)
+
+    # First convert to a generic TryteString, to make sure that the
+    # sequence doesn't contain any invalid characters.
+    try:
+      value = TryteString(value)
+    except ValueError:
+      return self._invalid_value(value, self.CODE_NOT_TRYTES, exc_info=True)
+
+    # Now coerce to the expected type and verify that there are no
+    # type-specific errors.
+    try:
+      addy = Address(value)
+      if addy.checksum and not addy.is_checksum_valid():
+        return self._invalid_value(
+        value     = addy,
+        reason    = self.ADDRESS_BAD_CHECKSUM,
+        exc_info  = True,
+
+        template_vars = {
+          'result_type': self.ADDRESS_BAD_CHECKSUM,
+        },
+      )
+      return Address(addy.address)
+    except ValueError:
+      return self._invalid_value(
+        value     = value,
+        reason    = self.CODE_WRONG_FORMAT,
+        exc_info  = True,
+
+        template_vars = {
+          'result_type': self.result_type.__name__,
+        },
+      )

test/filters_test.py

@@ -6,7 +6,7 @@
 from filters.test import BaseFilterTestCase
 
 from iota import Address, TryteString, TransactionHash
-from iota.filters import GeneratedAddress, NodeUri, Trytes
+from iota.filters import GeneratedAddress, NodeUri, Trytes, AddressNoChecksum
 
 
 class GeneratedAddressTestCase(BaseFilterTestCase):
@@ -201,3 +201,48 @@ def test_fail_wrong_type(self):
       [TryteString(b'RBTC9D9DCDQAEASBYBCCKBFA')],
       [f.Type.CODE_WRONG_TYPE],
     )
+
+
+# noinspection SpellCheckingInspection
+class AddressNoChecksumTestCase(BaseFilterTestCase):
+  filter_type = AddressNoChecksum
+
+  # noinspection SpellCheckingInspection
+  def setUp(self):
+    super(AddressNoChecksumTestCase, self).setUp()
+
+    # Define some addresses that we can reuse between tests
+    """
+    Incoming value is not an :py:class:`Address` instance.
+    """
+    self.tryte1 = (
+      b'TESTVALUE9DONTUSEINPRODUCTION99999FBFFTG'
+      b'QFWEHEL9KCAFXBJBXGE9HID9XCOHFIDABHDG9AHDR'
+    )
+    self.checksum = b'ENXYJOBP9'
+    self.address = Address(self.tryte1)
+    self.address_with_checksum = Address(self.tryte1 + self.checksum)
+    self.address_with_bad_checksum = Address(self.tryte1 + b'DEADBEEF9')
+
+  def test_pass_checksumless_addy(self):
+    """
+    Incoming value is tryte in address form or Address object
+    """
+    self.assertFilterPasses(self.tryte1)
+    self.assertFilterPasses(self.address)
+
+  def test_pass_withchecksum_addy(self):
+    """
+    After passing through the filter an address with a checksum should
+    return the address without
+    """
+    self.assertFilterPasses(self.address_with_checksum, self.address)
+
+  def test_fail_badchecksum_addy(self):
+    """
+    If they've got a bad checksum in their address we should probably tell
+    them so they don't wonder why something works in one place and not another
+    """
+    self.assertFilterErrors(
+      self.address_with_bad_checksum,
+      [AddressNoChecksum.ADDRESS_BAD_CHECKSUM])