Move roles to models file

Any time we can keep code out of the view files is good. This user stuff
seems like a sensible place to put it, since we’re going to use it
there.
This commit is contained in:
Chris Hill-Scott
2018-01-11 22:12:30 +00:00
parent d9a63c07a9
commit f2418c171c
2 changed files with 9 additions and 8 deletions

View File

@@ -20,17 +20,10 @@ from app.main.forms import (
PermissionsForm
)
from app import (user_api_client, current_service, service_api_client, invite_api_client)
from app.notify_client.models import roles
from app.utils import user_has_permissions
roles = {
'send_messages': ['send_texts', 'send_emails', 'send_letters'],
'manage_templates': ['manage_templates'],
'manage_service': ['manage_users', 'manage_settings'],
'manage_api_keys': ['manage_api_keys']
}
@main.route("/services/<service_id>/users")
@login_required
@user_has_permissions('view_activity', admin_override=True)

View File

@@ -2,6 +2,14 @@ from flask_login import UserMixin, AnonymousUserMixin
from flask import session
roles = {
'send_messages': ['send_texts', 'send_emails', 'send_letters'],
'manage_templates': ['manage_templates'],
'manage_service': ['manage_users', 'manage_settings'],
'manage_api_keys': ['manage_api_keys']
}
class User(UserMixin):
def __init__(self, fields, max_failed_login_count=3):
self._id = fields.get('id')