# -*- coding: utf-8 -*-

"""

    rhodecode.model.users_group

    ~~~~~~~~~~~~~~~~~~~~~~~~~~~

    users group model for RhodeCode

    :created_on: Oct 1, 2011

    :author: nvinot

    :copyright: (C) 2011-2011 Nicolas Vinot <aeris@imirhil.fr>

    :license: GPLv3, see COPYING for more details.

"""

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging

import traceback

from rhodecode.lib.caching_query import FromCache

from rhodecode.model.db import UsersGroupMember, UsersGroup

log = logging.getLogger(__name__)

class UsersGroupModel(BaseModel):

    def get(self, users_group_id, cache = False):

        users_group = UsersGroup.query()

        if cache:

            users_group = users_group.options(FromCache("sql_cache_short",

                                          "get_users_group_%s" % users_group_id))

        return users_group.get(users_group_id)

    def get_by_name(self, name, cache = False, case_insensitive = False):

        users_group = UsersGroup.query()

        if case_insensitive:

            users_group = users_group.filter(UsersGroup.users_group_name.ilike(name))

        else:

            users_group = users_group.filter(UsersGroup.users_group_name == name)

        if cache:

            users_group = users_group.options(FromCache("sql_cache_short",

                                          "get_users_group_%s" % name))

        return users_group.scalar()

    def create(self, form_data):

        try:

            new_users_group = UsersGroup()

            for k, v in form_data.items():

                setattr(new_users_group, k, v)

            self.sa.add(new_users_group)

            self.sa.commit()

            return new_users_group

        except:

            log.error(traceback.format_exc())

            self.sa.rollback()

            raise

    def add_user_to_group(self, users_group, user):

        for m in users_group.members:

            u = m.user

            if u.user_id == user.user_id:

                return m

        try:

            users_group_member = UsersGroupMember()

            users_group_member.user = user

            users_group_member.users_group = users_group