PyOTP 1.3.1__tar.gz → 1.4__tar.gz

{pyotp-1.3.1 → pyotp-1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: pyotp
-Version: 1.3.1
+Version: 1.4
 Summary: https://github.com/nathforge/pyotp
 Home-page: https://github.com/nathforge/pyotp
 Author: Mark Percival (ported by Nathan Reynolds to Python)

{pyotp-1.3.1 → pyotp-1.4}/setup.py

@@ -3,7 +3,7 @@
 from distutils.core import setup
 
 setup(
-    version='1.3.1',
+    version='1.4',
     url='https://github.com/nathforge/pyotp',
     name='pyotp',
     description='https://github.com/nathforge/pyotp',

{pyotp-1.3.1 → pyotp-1.4}/src/pyotp/hotp.py

@@ -10,15 +10,15 @@ class HOTP(OTP):
         @returns [Integer] OTP
         """
         return self.generate_otp(count)
-    
+
     def verify(self, otp, counter):
         """
         Verifies the OTP passed in against the current time OTP
         @param [String/Integer] otp the OTP to check against
         @param [Integer] counter the counter of the OTP
         """
-        return otp == self.at(counter)
-    
+        return unicode(otp) == unicode(self.at(counter))
+
     def provisioning_uri(self, name, initial_count=0):
         """
         Returns the provisioning URI for the OTP

{pyotp-1.3.1 → pyotp-1.4}/src/pyotp/totp.py

@@ -7,12 +7,12 @@ import urllib
 class TOTP(OTP):
     def __init__(self, *args, **kwargs):
         """
-        @option options [Integer] interval (30) the time interval in seconds for OTP
-            This defaults to 30 which is standard.
+        @option options [Integer] interval (30) the time interval in seconds
+            for OTP This defaults to 30 which is standard.
         """
         self.interval = kwargs.pop('interval', 30)
         super(TOTP, self).__init__(*args, **kwargs)
-    
+
     def at(self, for_time):
         """
         Accepts either a Unix timestamp integer or a Time object.
@@ -22,14 +22,14 @@ class TOTP(OTP):
         if not isinstance(for_time, datetime.datetime):
             for_time = datetime.datetime.fromtimestamp(int(for_time))
         return self.generate_otp(self.timecode(for_time))
-    
+
     def now(self):
         """
         Generate the current time OTP
         @return [Integer] the OTP as an integer
         """
         return self.generate_otp(self.timecode(datetime.datetime.now()))
-    
+
     def verify(self, otp, for_time=None):
         """
         Verifies the OTP passed in against the current time OTP
@@ -37,8 +37,9 @@ class TOTP(OTP):
         """
         if for_time is None:
             for_time = datetime.datetime.now()
-        return otp == self.at(for_time)
-    
+
+        return unicode(otp) == unicode(self.at(for_time))
+
     def provisioning_uri(self, name):
         """
         Returns the provisioning URI for the OTP
@@ -51,7 +52,7 @@ class TOTP(OTP):
             'name': urllib.quote(name, safe='@'),
             'secret': self.secret,
         }
-    
+
     def timecode(self, for_time):
         i = time.mktime(for_time.timetuple())
         return int(i / self.interval)

{pyotp-1.3.1 → pyotp-1.4}/test.py

@@ -23,12 +23,13 @@ class HOTPExampleValuesFromTheRFC(unittest.TestCase):
         self.assertEqual(hotp.at(7), 162583)
         self.assertEqual(hotp.at(8), 399871)
         self.assertEqual(hotp.at(9), 520489)
-    
+
     def testVerifyAnOTPAndNowAllowReuse(self):
         hotp = pyotp.HOTP('GEZDGNBVGY3TQOJQGEZDGNBVGY3TQOJQ')
         self.assertTrue(hotp.verify(520489, 9))
         self.assertFalse(hotp.verify(520489, 10))
-    
+        self.assertFalse(hotp.verify("520489", 10))
+
     def testProvisioningURI(self):
         hotp = pyotp.HOTP('wrn3pqx5uqxqvnqr')
         self.assertEqual(hotp.provisioning_uri('mark@percival'), 'otpauth://hotp/mark@percival?secret=wrn3pqx5uqxqvnqr&counter=0')
@@ -40,19 +41,20 @@ class TOTPExampleValuesFromTheRFC(unittest.TestCase):
         self.assertEqual(totp.at(1111111111), 50471)
         self.assertEqual(totp.at(1234567890), 5924)
         self.assertEqual(totp.at(2000000000), 279037)
-    
+
     def testMatchTheGoogleAuthenticatorOutput(self):
         totp = pyotp.TOTP('wrn3pqx5uqxqvnqr')
         with Timecop(1297553958):
             self.assertEqual(totp.now(), 102705)
-    
+
     def testValidateATimeBasedOTP(self):
         totp = pyotp.TOTP('wrn3pqx5uqxqvnqr')
         with Timecop(1297553958):
             self.assertTrue(totp.verify(102705))
+            self.assertTrue(totp.verify("102705"))
         with Timecop(1297553958 + 30):
             self.assertFalse(totp.verify(102705))
-    
+
     def testProvisioningURI(self):
         totp = pyotp.TOTP('wrn3pqx5uqxqvnqr')
         self.assertEqual(totp.provisioning_uri('mark@percival'), 'otpauth://totp/mark@percival?secret=wrn3pqx5uqxqvnqr')
@@ -62,23 +64,23 @@ class Timecop(object):
     """
     Half-assed clone of timecop.rb, just enough to pass our tests.
     """
-    
+
     def __init__(self, freeze_timestamp):
         self.freeze_timestamp = freeze_timestamp
-    
+
     def __enter__(self):
         self.real_datetime = datetime.datetime
         datetime.datetime = self.frozen_datetime()
-    
+
     def __exit__(self, type, value, traceback):
         datetime.datetime = self.real_datetime
-    
+
     def frozen_datetime(self):
         class FrozenDateTime(datetime.datetime):
             @classmethod
             def now(cls):
                 return cls.fromtimestamp(timecop.freeze_timestamp)
-        
+
         timecop = self
         return FrozenDateTime