mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-05 16:38:59 -04:00
Merge pull request #121 from alphagov/user-cleanup
Inherit from flask usermixin for default implementation of
This commit is contained in:
@@ -4,6 +4,8 @@ from client.errors import (
|
|||||||
InvalidResponse
|
InvalidResponse
|
||||||
)
|
)
|
||||||
|
|
||||||
|
from flask.ext.login import UserMixin
|
||||||
|
|
||||||
|
|
||||||
class UserApiClient(BaseAPIClient):
|
class UserApiClient(BaseAPIClient):
|
||||||
def __init__(self, base_url=None, client_id=None, secret=None):
|
def __init__(self, base_url=None, client_id=None, secret=None):
|
||||||
@@ -84,7 +86,7 @@ class UserApiClient(BaseAPIClient):
|
|||||||
raise e
|
raise e
|
||||||
|
|
||||||
|
|
||||||
class User(object):
|
class User(UserMixin):
|
||||||
def __init__(self, fields, max_failed_login_count=3):
|
def __init__(self, fields, max_failed_login_count=3):
|
||||||
self._id = fields.get('id')
|
self._id = fields.get('id')
|
||||||
self._name = fields.get('name')
|
self._name = fields.get('name')
|
||||||
@@ -98,9 +100,6 @@ class User(object):
|
|||||||
def get_id(self):
|
def get_id(self):
|
||||||
return self.id
|
return self.id
|
||||||
|
|
||||||
def is_authenticated(self):
|
|
||||||
return True
|
|
||||||
|
|
||||||
def is_active(self):
|
def is_active(self):
|
||||||
return self.state == 'active'
|
return self.state == 'active'
|
||||||
|
|
||||||
@@ -160,9 +159,6 @@ class User(object):
|
|||||||
def failed_login_count(self, num):
|
def failed_login_count(self, num):
|
||||||
self._failed_login_count += num
|
self._failed_login_count += num
|
||||||
|
|
||||||
def is_anonymous(self):
|
|
||||||
return False
|
|
||||||
|
|
||||||
def is_locked(self):
|
def is_locked(self):
|
||||||
return self.failed_login_count >= self.max_failed_login_count
|
return self.failed_login_count >= self.max_failed_login_count
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user