Source code for stalker.models.department

# -*- coding: utf-8 -*-
# Stalker a Production Asset Management System
# Copyright (C) 2009-2013 Erkan Ozgur Yilmaz
# 
# This file is part of Stalker.
# 
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation;
# version 2.1 of the License.
# 
# This library 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
# Lesser General Public License for more details.
# 
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA

from sqlalchemy import Column, Integer, ForeignKey
from sqlalchemy.orm import relationship, validates, synonym

from stalker import defaults

from stalker.models.auth import User
from stalker.models.entity import Entity

from stalker.log import logging_level
import logging
logger = logging.getLogger(__name__)
logger.setLevel(logging_level)


[docs]class Department(Entity): """The departments that forms the studio itself. The information that a Department object holds is like: * The members of the department * The lead of the department * and all the other things those are inherited from the AuditEntity class Two Department object considered the same if they have the same name, the the members list nor the lead info is important, a "Modeling" department should of course be the same with another department which has the name "Modeling" again. so creating a department object needs the following parameters: :param members: it can be an empty list, so one department can be created without any member in it. But this parameter should be a list of User objects. :param lead: this is a User object, that holds the lead information, a lead could be in this department but it is not forced to be also a member of the department. So another departments member can be a lead for another department. Lead attribute can not be empty or None. """ __auto_name__ = False __tablename__ = "Departments" __mapper_args__ = {"polymorphic_identity": "Department"} department_id = Column( "id", Integer, ForeignKey("Entities.id"), primary_key=True ) lead_id = Column( "lead_id", Integer, ForeignKey("Users.id", use_alter=True, name="x") ) lead = relationship( "User", uselist=False, primaryjoin="Department.lead_id==User.user_id", post_update=True, doc="""The lead of this department, it is a User object""", ) members = relationship( "User", secondary='User_Departments', back_populates="departments", doc="""List of users representing the members of this department.""", ) users = synonym('members')
[docs] def __init__(self, members=None, lead=None, **kwargs): super(Department, self).__init__(**kwargs) if members is None: members = [] self.members = members self.lead = lead
def __eq__(self, other): """the equality operator """ return super(Department, self).__eq__(other) and\ isinstance(other, Department) @validates("members") def _validate_members(self, key, member): """validates the given member attribute """ if not isinstance(member, User): raise TypeError("Every element in the %s.members list should be " "an instance of stalker.models.auth.User class " "not %s" % (self.__class__.__name__, member.__class__.__name__)) return member @validates("lead") def _validate_lead(self, key, lead): """validates the given lead attribute """ from stalker.models.auth import User if lead is not None: # the lead should be an instance of User class if not isinstance(lead, User): raise TypeError("%s.lead should be an instance of " "stalker.models.auth.User not %s" % (self.__class__.__name__, lead.__class__.__name__)) return lead @property
[docs] def to_tjp(self): """outputs a TaskJuggler compatible string """ from jinja2 import Template temp = Template(defaults.tjp_department_template) return temp.render({'department': self})