OpenProject is the leading open source project management software.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
openproject/lib/open_project/costs/patches/user_patch.rb

111 lines
3.3 KiB

#-- copyright
# OpenProject Costs Plugin
#
# Copyright (C) 2009 - 2014 the OpenProject Foundation (OPF)
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# version 3.
#
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#++
module OpenProject::Costs::Patches::UserPatch
def self.included(base) # :nodoc:
base.send(:include, InstanceMethods)
base.class_eval do
unloadable
has_many :rates, :class_name => 'HourlyRate'
has_many :default_rates, :class_name => 'DefaultHourlyRate'
before_save :save_rates
end
end
module InstanceMethods
def allowed_to_condition_with_project_id(permission, projects = nil)
ids = Project.all(:select => :id,
:conditions => Project.allowed_to_condition(self, permission, :project => projects)).map(&:id)
ids.empty? ?
"1=0" :
"(#{Project.table_name}.id in (#{ids.join(", ")}))"
end
def current_rate(project = nil, include_default = true)
rate_at(Date.today, project, include_default)
end
# kept for backwards compatibility
def rate_at(date, project = nil, include_default = true)
::HourlyRate.at_date_for_user_in_project(date, self.id, project, include_default)
end
def current_default_rate()
::DefaultHourlyRate.at_for_user(Date.today, self.id)
end
# kept for backwards compatibility
def default_rate_at(date)
::DefaultHourlyRate.at_for_user(date, self.id)
end
def add_rates(project, rate_attributes)
# set project to nil to set the default rates
return unless rate_attributes
rate_attributes.each do |index, attributes|
attributes[:rate] = Rate.clean_currency(attributes[:rate])
if project.nil?
default_rates.build(attributes)
else
attributes[:project] = project
rates.build(attributes)
end
end
end
def set_existing_rates (project, rates_attributes)
if project.nil?
default_rates.reject(&:new_record?).each do |rate|
update_rate(rate, rates_attributes[rate.id.to_s], false)
end
else
rates.reject{|r| r.new_record? || r.project_id != project.id}.each do |rate|
update_rate(rate, rates_attributes[rate.id.to_s], true)
end
end
end
def save_rates
(default_rates + rates).each do |rate|
return false if !rate.save
end
end
private
def update_rate(rate, attributes, project_rate = true)
if attributes && attributes[:rate].present?
attributes[:rate] = Rate.clean_currency(attributes[:rate])
rate.attributes = attributes
else
# TODO: this is surprising
# as it actually deletes the rate right away
# as opposed to the behaviour when changing the attributes
project_rate ? rates.delete(rate) : default_rates.delete(rate)
end
end
end
end