Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def _get_role(self):
- """
- """
- session = get_session()
- roles = session.query(UserRole).filter(
- UserRole.user_id == self.username).all()
- for r in roles:
- if r.role in ['Approved', 'Rejected', 'Pending', 'Manager']:
- return r.role
- def _set_role(self, role):
- """
- Make it work as a single value instead of a list - setting a role
- removes all other roles (it doesn't make sence to have both
- Approved and Rejected, for example
- """
- ### Ha! There's a simpler way!
- if role == 'Pending':
- self.roles = [UserRole(role)]
- else:
- self.roles = [UserRole(role), UserRole('Member')]
Add Comment
Please, Sign In to add comment