Cleaned some code/gitignore and commented the classes

This commit is contained in:
Felipe Martin 2013-07-21 16:25:58 +02:00
parent 6e8c9130a3
commit 37cc29fad7
2 changed files with 20 additions and 7 deletions

3
.gitignore vendored
View File

@ -1,3 +1,6 @@
.c9revisions/
tests.py
*.pyc
build/
dist/
*.egg-info

View File

@ -8,6 +8,10 @@ import requests
class YubicoWS(object):
"""
Yubico Web Service class that interacts with the Yubico API
"""
register_ws = 'https://upgrade.yubico.com/getapikey/?format=json'
api_ws = None
@ -152,6 +156,10 @@ class YubicoWS(object):
class Yubikey(object):
"""
Yubikey object wrapper
"""
id = None
key = None
prefix = None
@ -197,7 +205,11 @@ class Yubikey(object):
self.prefix = str.lower(otp[:-32])
###
# Custom exceptions
###
class YubicoWSError(Exception):
"Web service error. Defined by yubico documentation."
def __init__(self, message=None):
self.msg = "Web Service responded with an error: %s" % message
@ -206,17 +218,15 @@ class YubicoWSError(Exception):
class YubicoWSInvalidResponse(Exception):
"Exception if the web service answers without same otp/nonce parameters"
msg = 'Response from the server is invalid'
class WSResponseError(Exception):
def __str__(self):
return repr(self.msg)
class YubicoWSResponseBadSignature(Exception):
"Exception if the web service answers with a invalid signature"
pass
class OTPIncorrectFormat(Exception):
pass
class YubicoWSResponseBadSignature(Exception):
"Exception raised if the OTP provided is incorrect"
pass