def get_by_user_name_or_email(cls, request, user_name, email):
session = get_session(request)
return session.query(cls).filter(
or_(
func.lower(cls.user_name) == user_name.lower(),
cls.email == email
)
).first()
def test_get_session(self):
from horus.interfaces import IHorusSession
from horus.lib import get_session
request = testing.DummyRequest()
request.registry = Mock()
session = Mock()
getUtility = Mock()
getUtility.return_value = session
request.registry.getUtility = getUtility
new_session = get_session(request)
getUtility.assert_called_with(IHorusSession)
assert new_session == session
def create_activation(request, user):
db = get_session(request)
Activation = request.registry.getUtility(IHorusActivationClass)
activation = Activation()
db.add(activation)
user.activation = activation
db.flush()
body = pystache.render(_("Please activate your e-mail address by visiting {{ link }}"),
{
'link': request.route_url('activate', user_pk=user.pk, code=user.activation.code)
}
)
subject = _("Please activate your e-mail address!")
message = Message(subject=subject, recipients=[user.email], body=body)
mailer = get_mailer(request)
mailer.send(message)
def __init__(self, request):
self._request = request
self.settings = request.registry.settings
self.User = request.registry.getUtility(IHorusUserClass)
self.Activation = request.registry.getUtility(IHorusActivationClass)
self.db = get_session(request)
def get_by_pk(cls, request, pk):
"""Gets an object by its primary key"""
session = get_session(request)
return session.query(cls).filter(cls.pk == pk).first()
def get_all(cls, request):
session = get_session(request)
return session.query(cls).all()
def get_by_activation(cls, request, activation):
session = get_session(request)
user = session.query(cls).filter(cls.activation_pk == activation.pk).first()
return user
def get_by_user_name(cls, request, user_name):
session = get_session(request)
return session.query(cls).filter(
func.lower(cls.user_name) == user_name.lower(),
).first()
def get_by_email(cls, request, email):
session = get_session(request)
return session.query(cls).filter(
func.lower(cls.email) == email.lower()
).first()
def get_by_code(cls, request, code):
session = get_session(request)
return session.query(cls).filter(cls.code == code).first()
