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