import datetime

import logging

log = logging.getLogger(__name__)

class CubicWebAuthTktAuthenticationPolicy(AuthTktAuthenticationPolicy):

    """An authentication policy that update the user last_login_time.

    The update is done in the 'remember' method, which is called on login,

    and each time the authentication ticket is reissued.

    Meaning, the last_login_time is updated reissue_time seconds (maximum)

    before the last request by the user.

    """

    def remember(self, request, principal, **kw):

        headers = super(CubicWebAuthTktAuthenticationPolicy, self).remember(

            request, principal, **kw)

        try:

            repo = request.registry['cubicweb.repository']

            with repo.internal_session() as cnx:

                cnx.execute(

                    "SET U last_login_time %(now)s WHERE U eid %(user)s", {

                        'now': datetime.datetime.now(),

                        'user': principal})

                cnx.commit()

        except:

            log.exception("Failed to update last_login_time")

        return headers

        CubicWebAuthTktAuthenticationPolicy(