From 73ff1c00b3d9d9ff7a45684d83649ae55581980a Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Wed, 11 Sep 2013 01:10:16 -0400 Subject: [PATCH] Fixes #146 --- flask_security/datastore.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/flask_security/datastore.py b/flask_security/datastore.py index 2adc7a7..b454dc8 100644 --- a/flask_security/datastore.py +++ b/flask_security/datastore.py @@ -266,7 +266,10 @@ class PeeweeUserDatastore(PeeweeDatastore, UserDatastore): return None def find_role(self, role): - return self.role_model.filter(name=role).get() + try: + return self.role_model.filter(name=role).get() + except self.role_model.DoesNotExist: + return None def create_user(self, **kwargs): """Creates and returns a new user from the given parameters."""