user.py

# -*- coding: utf-8 -*-
from trytond.model import ModelSQL, fields
from trytond.pool import PoolMeta
from nereid import login_required, request, flash, redirect, url_for, jsonify, \
     route, current_user

__all__ = ['NereidUser', 'NereidUserParty', 'Party']
__metaclass__ = PoolMeta


class NereidUser:
    "Nereid User"
    __name__ = "nereid.user"

    parties = fields.Many2Many(
        'nereid.user-party.party', 'nereid_user', 'party', 'Parties',
        help="Parties to which this user is related"
    )

    @classmethod
    def __setup__(cls):
        super(NereidUser, cls).__setup__()
        cls._error_messages.update({
            "party_not_in_parties": "Party missing from Parties",
        })

    @classmethod
    def validate(cls, users):
        super(NereidUser, cls).validate(users)

        for user in users:
            user.validate_party()

    def validate_party(self):
        if self.parties and self.party not in self.parties:
            self.raise_user_error('party_not_in_parties')

    @classmethod
    def create(cls, vlist):
        """
        Add the current party of the user to the list of parties allowed for
        the user automatically
        """
        users = super(NereidUser, cls).create(vlist)
        for user in users:
            if user.party not in user.parties:
                cls.write([user], {'parties': [('add', [user.party.id])]})
        return users

    @classmethod
    @login_required
    @route('/change-current-party/<int:party_id>', readonly=False)
    def change_party(cls, party_id):
        """
        Updates the current party of the nereid_user to the new party_id if
        it is one of the parties in the list of parties of the user

        :param party_id: ID of the party
        """
        for party in current_user.parties:
            if party.id == party_id:
                cls.write([current_user], {'party': party.id})
                break
        else:
            if request.is_xhr:
                return jsonify(
                    error='The party is not valid'
                ), 400
            flash("The party is not valid")
        if request.is_xhr:
            return jsonify(
                success='Party has been changed successfully'
            )
        return redirect(
            request.args.get('next', url_for('nereid.website.home'))
        )


class NereidUserParty(ModelSQL):
    "Nereid User - Party"
    __name__ = "nereid.user-party.party"

    nereid_user = fields.Many2One(
        'nereid.user', 'Nereid User', ondelete='CASCADE',
        select=True, required=True
    )
    party = fields.Many2One(
        'party.party', 'Party', ondelete='CASCADE',
        select=True, required=True
    )


class Party:
    __name__ = "party.party"

    @classmethod
    def __setup__(cls):
        super(Party, cls).__setup__()

        cls.nereid_users = fields.Many2Many(
            'nereid.user-party.party', 'party', 'nereid_user', 'Nereid Users',
            help="Nereid Users which have access to this party"
        )