Guest User

Untitled

a guest
Sep 22nd, 2018
71
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.41 KB | None | 0 0
  1.        def _get_role(self):                                                
  2.         """                                                              
  3.        """                                                              
  4.         session = get_session()                                          
  5.         roles = session.query(UserRole).filter(                          
  6.             UserRole.user_id == self.username).all()                    
  7.         for r in roles:                                                  
  8.             if r.role in ['Approved', 'Rejected', 'Pending', 'Manager']:
  9.                 return r.role  
  10.  
  11. def _set_role(self, role):                                          
  12.         """                                                              
  13.        Make it work as a single value instead of a list - setting a role
  14.        removes all other roles (it doesn't make sence to have both      
  15.        Approved and Rejected, for example                              
  16.        """                                                              
  17.         ### Ha! There's a simpler way!                                  
  18.         if role == 'Pending':                                            
  19.             self.roles = [UserRole(role)]                                
  20.         else:                                                            
  21.             self.roles = [UserRole(role), UserRole('Member')]
Add Comment
Please, Sign In to add comment