aboutsummaryrefslogtreecommitdiffstats
path: root/src/cryptography/hazmat/primitives/twofactor
diff options
context:
space:
mode:
authorJiangge Zhang <tonyseek@gmail.com>2015-06-03 17:30:11 +0800
committerJiangge Zhang <tonyseek@gmail.com>2015-06-03 17:39:59 +0800
commitd4e0a839ef676a2ed05140f98ef3630deb0fdd61 (patch)
tree4786265848ce18dbda22f6daa3b3a28e6c48d561 /src/cryptography/hazmat/primitives/twofactor
parente9f8eb9af0c4f42dd4e68ae05580f77dc352aaf4 (diff)
downloadcryptography-d4e0a839ef676a2ed05140f98ef3630deb0fdd61.tar.gz
cryptography-d4e0a839ef676a2ed05140f98ef3630deb0fdd61.tar.bz2
cryptography-d4e0a839ef676a2ed05140f98ef3630deb0fdd61.zip
Remove a default argument and rename a private function.
Diffstat (limited to 'src/cryptography/hazmat/primitives/twofactor')
-rw-r--r--src/cryptography/hazmat/primitives/twofactor/hotp.py6
-rw-r--r--src/cryptography/hazmat/primitives/twofactor/totp.py6
-rw-r--r--src/cryptography/hazmat/primitives/twofactor/utils.py4
3 files changed, 8 insertions, 8 deletions
diff --git a/src/cryptography/hazmat/primitives/twofactor/hotp.py b/src/cryptography/hazmat/primitives/twofactor/hotp.py
index f59f551c..8c0cec14 100644
--- a/src/cryptography/hazmat/primitives/twofactor/hotp.py
+++ b/src/cryptography/hazmat/primitives/twofactor/hotp.py
@@ -15,7 +15,7 @@ from cryptography.hazmat.backends.interfaces import HMACBackend
from cryptography.hazmat.primitives import constant_time, hmac
from cryptography.hazmat.primitives.hashes import SHA1, SHA256, SHA512
from cryptography.hazmat.primitives.twofactor import InvalidToken
-from cryptography.hazmat.primitives.twofactor.utils import generate_uri
+from cryptography.hazmat.primitives.twofactor.utils import _generate_uri
class HOTP(object):
@@ -61,7 +61,7 @@ class HOTP(object):
p = hmac_value[offset:offset + 4]
return struct.unpack(">I", p)[0] & 0x7fffffff
- def get_provisioning_uri(self, account_name, counter, issuer=None):
- return generate_uri(self, 'hotp', account_name, issuer, [
+ def get_provisioning_uri(self, account_name, counter, issuer):
+ return _generate_uri(self, 'hotp', account_name, issuer, [
('counter', int(counter)),
])
diff --git a/src/cryptography/hazmat/primitives/twofactor/totp.py b/src/cryptography/hazmat/primitives/twofactor/totp.py
index 9c8eddad..98493b6d 100644
--- a/src/cryptography/hazmat/primitives/twofactor/totp.py
+++ b/src/cryptography/hazmat/primitives/twofactor/totp.py
@@ -11,7 +11,7 @@ from cryptography.hazmat.backends.interfaces import HMACBackend
from cryptography.hazmat.primitives import constant_time
from cryptography.hazmat.primitives.twofactor import InvalidToken
from cryptography.hazmat.primitives.twofactor.hotp import HOTP
-from cryptography.hazmat.primitives.twofactor.utils import generate_uri
+from cryptography.hazmat.primitives.twofactor.utils import _generate_uri
class TOTP(object):
@@ -33,7 +33,7 @@ class TOTP(object):
if not constant_time.bytes_eq(self.generate(time), totp):
raise InvalidToken("Supplied TOTP value does not match.")
- def get_provisioning_uri(self, account_name, issuer=None):
- return generate_uri(self._hotp, 'totp', account_name, issuer, [
+ def get_provisioning_uri(self, account_name, issuer):
+ return _generate_uri(self._hotp, 'totp', account_name, issuer, [
('period', int(self._time_step)),
])
diff --git a/src/cryptography/hazmat/primitives/twofactor/utils.py b/src/cryptography/hazmat/primitives/twofactor/utils.py
index 89d38ff2..ae32058b 100644
--- a/src/cryptography/hazmat/primitives/twofactor/utils.py
+++ b/src/cryptography/hazmat/primitives/twofactor/utils.py
@@ -1,11 +1,11 @@
-from __future__ import unicode_literals
+from __future__ import absolute_import, division, print_function
import base64
from six.moves.urllib.parse import quote, urlencode
-def generate_uri(hotp, type_name, account_name, issuer, extra_parameters):
+def _generate_uri(hotp, type_name, account_name, issuer, extra_parameters):
parameters = [
('digits', hotp._length),
('secret', base64.b32encode(hotp._key)),