~openerp-commiter/openobject-addons/trunk-extra-addons

##############################################################################

#

# Copyright (c) 2007 TINY SPRL. (http://tiny.be) All Rights Reserved.

#

# WARNING: This program as such is intended to be used by professional

# programmers who take the whole responsability of assessing all potential

# consequences resulting from its eventual inadequacies and bugs

# End users who are looking for a ready-to-use solution with commercial

# garantees and support are strongly adviced to contract a Free Software

# Service Company

#

# 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 2

# 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 impli(ed 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

#

##############################################################################

from osv import fields, osv

import time

from datetime import date,timedelta

import datetime

import pooler

import netsvc

STATE = [

	('none', 'Non Member'),

	('canceled', 'Canceled Member'),

	('old', 'Old Member'),

	('waiting', 'Waiting Member'),

	('invoiced', 'Invoiced Member'),

	('associated', 'Associated Member'),

	('free', 'Free Member'),

	('paid', 'Paid Member'),

]

class res_partner(osv.osv):

	_inherit = 'res.partner'

	_description = 'res.partner'

	_columns = {

		'asker_name': fields.char('Asker Name',size=50),

		'asker_address': fields.char('Asker Address',size=50),

		'asker_zip_id': fields.many2one('res.partner.zip','Asker Zip Code'),

		'sender_name': fields.char('Sender Name',size=50),

		'insurer_id' : fields.char('Insurer ID',size=50),

	}

res_partner()

class cci_missions_site(osv.osv):

	_name = 'cci_missions.site'

	_description = 'cci_missions.site'

	_columns = {

		'name' : fields.char('Name of the Site',size=50,required=True),

		'official_name_1' : fields.char('Official Name of the Site',size=50,required=True),

		'official_name_2' : fields.char('Official Name of the Site',size=50),

		'official_name_3' : fields.char('Official Name of the Site',size=50),

		'official_name_4' : fields.char('Official Name of the Site',size=50),

		'embassy_sequence_id' : fields.many2one('ir.sequence','Sequence for Embassy Folder'),

	}

cci_missions_site()

class cci_missions_embassy_folder(osv.osv):

	_name = 'cci_missions.embassy_folder'

	_description = 'cci_missions.embassy_folder'

	_inherits = {'crm.case': 'crm_case_id'}

	def _cci_mission_send(self, cr, uid, ids, *args):

		self.write(cr, uid, ids, {'state':'pending',})

		cases = self.browse(cr, uid, ids)

		self._history(cr, uid, cases, 'Send', history=True,)

		return True

	def _cci_mission_got_back(self,cr,uid,ids,*args):

		self.write(cr, uid, ids, {'state':'open',})

		cases = self.browse(cr, uid, ids)

		self._history(cr, uid, cases, 'Got Back', history=True)

		for id in self.browse(cr, uid, ids):

			data = {}

			obj_folder_line = self.pool.get('cci_missions.embassy_folder_line')

			temp = obj_folder_line.search(cr, uid, [('folder_id','=',id.id),('type','=','Translation')])

			if temp:

				translation_line = obj_folder_line.browse(cr, uid, [temp[0]])[0]

				if translation_line.awex_eligible and id.partner_id.awex_eligible == 'yes':

					#look for an existing credit line in the current time

					credit_line = self.pool.get('credit.line').search(cr, uid, [('from_date','<=',time.strftime('%Y-%m-%d')), ('to_date', '>=', time.strftime('%Y-%m-%d'))])

					if credit_line:

						#if there is one available: get available amount from it

						amount = self.pool.get('credit.line').browse(cr, uid,[credit_line[0]])[0].get_available_amount(cr, uid, credit_line[0], translation_line.customer_amount, id.partner_id.id)

						if amount > 0:

							data['awex_amount'] = amount

							data['credit_line_id'] =  credit_line[0]

						else:

							data['awex_eligible'] = False

					obj_folder_line.write(cr, uid, [translation_line.id], data)

		return True

	def _cci_mission_done_folder(self,cr,uid,ids,*args):

		self.write(cr, uid, ids, {'state':'done','invoice_date': time.strftime('%Y-%m-%d %H:%M:%S')})

		cases = self.browse(cr, uid, ids)

		self._history(cr, uid, cases, 'Invoiced', history=True)

		return True

	def _history(self, cr, uid,ids,keyword, history=False, email=False, context={}):

		for case in ids:

			data = {

				'name': keyword,

				'som': case.som.id,

				'canal_id': case.canal_id.id,

				'user_id': uid,

				'case_id': case.crm_case_id.id

			}

			obj = self.pool.get('crm.case.log')

			obj.create(cr, uid, data, context)

		return True

	def create(self, cr, uid, vals, *args, **kwargs):

#		Overwrite the name field to set next sequence according to the sequence in for the embassy folder related in the site_id

		if vals['site_id']:

			data = self.pool.get('cci_missions.site').browse(cr, uid,vals['site_id'])

			seq = self.pool.get('ir.sequence').get(cr, uid,data.embassy_sequence_id.code)

			if seq:

				vals.update({'name': seq})

		temp = super(osv.osv,self).create(cr, uid, vals, *args, **kwargs)

		self._history(cr, uid,self.browse(cr, uid, [temp]), 'Created', history=True)

		return temp

	def onchange_partner_id(self, cr, uid, ids, part):

		if not part:

			return {'value':{'partner_address_id': False}}

		part_obj = self.pool.get('res.partner')

		data_partner = part_obj.browse(cr,uid,part)

		if data_partner.alert_legalisations:

				raise osv.except_osv('Error!',data_partner.alert_explanation or 'Partner is not valid')

		addr = part_obj.address_get(cr, uid, [part], ['contact'])

		data = {'partner_address_id':addr['contact']}

		return {'value':data}

	def check_folder_line(self, cr, uid, ids):

		#CONSTRAINT: For each embassy Folder, it can only be one embassy_folder_line of each type.

		data_folder = self.browse(cr,uid,ids)

		list = []

		for folder in data_folder:

			for line in folder.embassy_folder_line_ids:

				if line.type and line.type in list:

					return False

				list.append(line.type)

		return True

	_columns = {

		'crm_case_id' : fields.many2one('crm.case','Case'),

		'member_price' : fields.boolean('Member Price Allowed'),

		'customer_reference' : fields.char('Folders Reference for the Customer',size=30),

		'destination_id' : fields.many2one('res.country','Destination Country'),

		'link_ids': fields.one2many('cci_missions.dossier','embassy_folder_id','Linked Documents'),

		'internal_note': fields.text('Internal Note'),

		'invoice_note':fields.text('Note to Display on the Invoice',help='to display as the last embassy_folder_line of this embassy_folder.'),

		'embassy_folder_line_ids' : fields.one2many('cci_missions.embassy_folder_line','folder_id','Details'),

		'site_id': fields.many2one('cci_missions.site','Site', required=True),

		'invoice_date' : fields.datetime('Invoice Date', readonly=True) ,

		"invoice_id":fields.many2one("account.invoice","Invoice"),

	}

	_defaults = {

		'section_id': lambda obj, cr, uid, context: obj.pool.get('crm.case.section').search(cr, uid, [('name','=','Embassy Folder')])[0],

		'invoice_date': lambda *a: False,

		'name': lambda *args: '/',

		'state' :  lambda *a : 'draft'

	}

	_constraints = [(check_folder_line, 'Error: Only One Embassy Folder line allowed for each type!', ['embassy_folder_line_ids'])]

cci_missions_embassy_folder()

class cci_missions_embassy_folder_line (osv.osv):

	_name = 'cci_missions.embassy_folder_line'

	_description = 'cci_missions.embassy_folder_line '

	def create(self, cr, uid, vals, *args, **kwargs):

		prod_name= vals['type'] + str(' Product')

		cr.execute('select id from product_template where name='"'%s'"''%str(prod_name))

		prod=cr.fetchone()

		if prod:

			product_id=prod[0]

			prod_info = self.pool.get('product.product').browse(cr, uid,product_id)

			account =  prod_info.product_tmpl_id.property_account_income.id

			if not account:

				account = prod_info.categ_id.property_account_income_categ.id

			vals['account_id']=account

		return super(osv.osv,self).create(cr, uid, vals, *args, **kwargs)

	def write(self, cr, uid, ids, vals, *args, **kwargs):

		if vals.has_key('type'):

			prod_name = vals['type'] + str(' Product')

			cr.execute('select id from product_template where name='"'%s'"''%str(prod_name))

			prod=cr.fetchone()

			if prod:

				product_id=prod[0]

				prod_info = self.pool.get('product.product').browse(cr, uid,product_id)

				account =  prod_info.product_tmpl_id.property_account_income.id

				if not account:

					account = prod_info.categ_id.property_account_income_categ.id

				vals['account_id']=account

		return super(osv.osv,self).write( cr, uid, ids,vals, *args, **kwargs)

	def onchange_line_type(self,cr,uid,ids,type):

		data={}

		data['courier_cost']=data['customer_amount']=data['account_id']=data['name']=False

		if not type:

			return {'value' : data }

		data['name']=type

		prod_name= str(type) + str(' Product')

		cr.execute('select id from product_template where name='"'%s'"''%str(prod_name))

		prod=cr.fetchone()

		if not prod:

			return {'value' : data }

		product_id=prod[0]

		prod_info = self.pool.get('product.product').browse(cr, uid,product_id)

		data['courier_cost']=prod_info.standard_price

		data['customer_amount']=prod_info.list_price

		account =  prod_info.product_tmpl_id.property_account_income.id

		if not account:

			account = prod_info.categ_id.property_account_income_categ.id

		data['account_id']=account

		return {'value' : data }

	_columns = {

		'name' : fields.char('Description',size=50,required=True),

		'folder_id' : fields.many2one('cci_missions.embassy_folder','Related Embassy Folder',required=True),

		'courier_cost' : fields.float('Couriers Costs'),

		'customer_amount' : fields.float('Invoiced Amount'),

		'tax_rate': fields.many2one('account.tax','Tax Rate'),

		'type' : fields.selection([('CBA','CBA'),('Ministry','Ministry'),('Embassy Consulate','Embassy Consulate'),('Translation','Translation'),('Administrative','Administrative'),('Travel Costs','Travel Costs'),('Others','Others')],'Type',required=True),

		'account_id' : fields.many2one('account.account', 'Account',required=True),

		'awex_eligible':fields.boolean('AWEX Eligible'),

		'awex_amount':fields.float('AWEX Amount', readonly=True),

		'credit_line_id':fields.many2one('credit.line', 'Credit Line', readonly=True),

	}

cci_missions_embassy_folder_line()

class cci_missions_dossier_type(osv.osv):

	_name = 'cci_missions.dossier_type'

	_description = 'cci_missions.dossier_type'

	_columns = {

		'code' : fields.char('Code',size=3,required=True),

		'name' : fields.char('Description',size=50,required=True),

		'original_product_id' : fields.many2one('product.product','Reference for Original Copies',required=True,help='for the association with a pricelist'),

		'copy_product_id' : fields.many2one('product.product','Reference for Copies',required=True,help='for the association with a pricelist'),

		'site_id' : fields.many2one('cci_missions.site','Site',required=True),

		'sequence_id' : fields.many2one('ir.sequence','Sequence',required=True,help='for association with a sequence'),

		'section' : fields.selection([('certificate','Certificate'),('legalization','Legalization'),('ATA','ATA Carnet')],'Type',required=True),

		'warranty_product_1': fields.many2one('product.product', 'Warranty product for ATA carnet if Own Risk'),

		'warranty_product_2': fields.many2one('product.product', 'Warranty product for ATA carnet if not own Risk'),

		'id_letter' : fields.char('ID Letter', size=1, help='for identify the type of certificate by the federation' ), 

	}

cci_missions_dossier_type()

class cci_missions_dossier(osv.osv):

	_name = 'cci_missions.dossier'

	_description = 'cci_missions.dossier'

	def create(self, cr, uid, vals, *args, **kwargs):

		#overwrite the create: if the text_on_invoice field is empty then fill it with name + destination_id.name + (quantity_original)

		if not vals['text_on_invoice']:

			invoice_text = vals['name']

			if vals['destination_id']:

				destination_data = self.pool.get('res.country').browse(cr,uid,vals['destination_id'])

				invoice_text = vals['name'] + ' ' + destination_data.name + ' (' + str(vals['quantity_original'])  + ')'

			vals.update({'text_on_invoice': invoice_text})

		return super(osv.osv,self).create(cr, uid, vals, *args, **kwargs)

	def get_partner_details(self, cr, uid, ids,order_partner_id):

		result={}

		asker_name=False

		sender_name=False

		if order_partner_id:

			partner_info = self.pool.get('res.partner').browse(cr, uid,order_partner_id)

			if not partner_info.asker_name:

				asker_name=partner_info.name

			else:

				asker_name=partner_info.asker_name

			if not partner_info.sender_name:

				sender_name=partner_info.name

			else:

				sender_name=partner_info.sender_name

		result = {'value': {

			'asker_name': asker_name,

			'sender_name': sender_name}

		}

		return result

	def _amount_subtotal(self, cr, uid, ids, name, args, context=None):

		res={}

		data_dosseir = self.browse(cr,uid,ids)

		for data in data_dosseir:

			sum=0.00

			for product in data.product_ids:

				sum += product.price_subtotal

			res[data.id]=sum

		return res

	_columns = {

		'id': fields.integer('ID', readonly=True),

		'name' : fields.char('Reference',size=20,required=True),

		'type_id' : fields.many2one('cci_missions.dossier_type','Dossier Type',required=True),

		'date' : fields.date('Creation Date',required=True),

		'order_partner_id': fields.many2one('res.partner','Billed Customer',required=True),

		'asker_name':fields.char('Asker Name',size=50),

		'sender_name':fields.char('Sender Name',size=50),

		'to_bill':fields.boolean('To Be Billed'),

		'state':fields.selection([('draft','Confirmed'),('invoiced','Invoiced'),('cancel_customer','Canceled by Customer'),('cancel_cci','Canceled by the CCI')],'State',),

		'goods':fields.char('Goods Description',size=100),

		'goods_value':fields.float('Value of the Sold Goods'),#Monetary; must be greater than zero

		'destination_id':fields.many2one('res.country','Destination Country'),

		'embassy_folder_id':fields.many2one('cci_missions.embassy_folder','Related Embassy Folder'),

		'quantity_copies':fields.integer('Number of Copies'),

		'quantity_original' : fields.integer('Quantity of Originals',required=True),

		'sub_total':fields.function(_amount_subtotal, method=True, string='Sub Total for Extra Products', store=True),

		'text_on_invoice':fields.text('Text to Display on the Invoice'),

		'product_ids': fields.one2many('product.lines', 'dossier_product_line_id', 'Products'),

		'invoice_id':fields.many2one("account.invoice","Invoice"),

		'invoiced_amount': fields.float('Total'),

	}

	_defaults = {

		'name': lambda *args: '/',

		'date': lambda *a: time.strftime('%Y-%m-%d'),

		'to_bill' : lambda *b : True,

		'state' : lambda *a : 'draft',

		'quantity_original' : lambda *a : 1

	}

cci_missions_dossier()

class cci_missions_custom_code(osv.osv):

	_name= 'cci_missions.custom_code'

	_desctiption = 'cci_missions.custom_code'

	_columns = {

		'name' : fields.char('Name',size=8,required=True),

		'meaning' : fields.char('Meaning',size=250,required=True),

		'official' : fields.boolean('Official Code'),

	}

	_defaults = {

		'official': lambda *a: False,

	}

cci_missions_custom_code()

class cci_missions_certificate(osv.osv):

	_name = 'cci_missions.certificate'

	_description = 'cci_missions.certificate'

	_inherits = {'cci_missions.dossier': 'dossier_id' }

	def _amount_total(self, cr, uid, ids, name, args, context=None):

		res ={}

		data_dosseir = self.browse(cr,uid,ids)

		for data in data_dosseir:

			if data.state =='draft':

				data_partner = self.pool.get('res.partner').browse(cr,uid,data.order_partner_id.id)

				context.update({'partner_id':data_partner})

				context.update({'force_member':False})

				context.update({'force_non_member':False})

				context.update({'date':data.date})

				context.update({'value_goods':data.goods_value})

				context.update({'pricelist':data_partner.property_product_pricelist.id})

				price_org = self.pool.get('product.product')._product_price(cr, uid, [data.type_id.original_product_id.id], False, False, context)

				price_copy = self.pool.get('product.product')._product_price(cr, uid, [data.type_id.copy_product_id.id], False, False, context)

				cost_org=price_org[data.type_id.original_product_id.id]

				cost_copy=price_copy[data.type_id.copy_product_id.id]

				qty_org = data.quantity_original

				qty_copy = data.quantity_copies

				subtotal =  data.sub_total

				if qty_org < 0 or qty_copy < 0:

					raise osv.except_osv('Input Error!','No. of Copies and Quantity of Originals should be positive.')

				total = ((cost_org * qty_org ) + (cost_copy * qty_copy) + subtotal)

				res[data.id] = total

			else :

				res[data.id]=data.invoiced_amount

		return res

	def cci_dossier_cancel_cci(self, cr, uid, ids, *args):

		data=self.browse(cr,uid,ids[0])

		if data.invoice_id.state == 'paid':

			new_ids = self.pool.get('account.invoice').refund(cr, uid,[data.invoice_id.id])

			self.write(cr, uid,ids, {'invoice_id' : new_ids[0]})

		else:

			wf_service = netsvc.LocalService('workflow')

			wf_service.trg_validate(uid, 'account.invoice', data.invoice_id.id, 'invoice_cancel', cr)

		self.write(cr, uid, ids, {'state':'cancel_cci',})

		return True

	def get_certification_details(self, cr, uid, ids,order_partner_id):

		result={}

		asker_name=False

		sender_name=False

		asker_address=False

		zip=False

		if order_partner_id:

			partner_info = self.pool.get('res.partner').browse(cr, uid,order_partner_id)

			if partner_info.alert_legalisations:

				raise osv.except_osv('Error!',partner_info.alert_explanation or 'Partner is not valid')

			if not partner_info.asker_name:

				asker_name=partner_info.name

			else:

				asker_name=partner_info.asker_name

			if not partner_info.sender_name:

				sender_name=partner_info.name

			else:

				sender_name=partner_info.sender_name

			if not partner_info.asker_address:

				if partner_info.address!=[]:

					for add in partner_info.address:

						if add.type=='default':

							asker_address=add.street

							break

			else:

				asker_address=partner_info.asker_address

			if not partner_info.asker_zip_id.id:

				if partner_info.address!=[]:

					for add in partner_info.address:

						if add.type=='default':

							zip=add.zip_id.id

							break

			else:

				zip=partner_info.asker_zip_id.id

		result = {'value': {

			'asker_name': asker_name,

			'asker_address': asker_address,

			'asker_zip_id': zip,

			'sender_name': sender_name}

		}

		return result

	def create(self, cr, uid, vals, *args, **kwargs):

#		Overwrite the name fields to set next sequence according to the sequence in the certification type (type_id)

		if vals['type_id']:

			data = self.pool.get('cci_missions.dossier_type').browse(cr, uid,vals['type_id'])

			seq = self.pool.get('ir.sequence').get(cr, uid,data.sequence_id.code)

			if seq:

				vals.update({'name': seq})

		return super(osv.osv,self).create(cr, uid, vals, *args, **kwargs)

	_columns = {

		'dossier_id' : fields.many2one('cci_missions.dossier','Dossier'),

		'total':fields.function(_amount_total, method=True, string='Total', store=True),# sum of the price for copies, originals and extra_products

		'asker_address' : fields.char('Asker Address',size=50),#by default, res.partner->asker_adress or, res_partner.address[default]->street

		'asker_zip_id' : fields.many2one('res.partner.zip','Asker Zip Code'),#by default, res.partner->asker_zip_id or, res_partner.address[default]->zip_id

		'special_reason' : fields.selection([('none','None'),('Commercial Reason','Commercial Reason'),('Substitution','Substitution')],'For special cases'),

		'legalization_ids' : fields.one2many('cci_missions.legalization','certificate_id','Related Legalizations'),

		'customs_ids' : fields.many2many('cci_missions.custom_code','certificate_custome_code_rel','certificate_id','custom_id','Custom Codes'),

		'sending_spf': fields.date('SPF Sending Date',help='Date of the sending of this record to the external database'),

		'origin_ids' : fields.many2many('res.country','certificate_country_rel','certificate_id','country_id','Origin Countries')

	}

	_defaults = {

		'special_reason': lambda *a: 'none',

	}

cci_missions_certificate()

class cci_missions_legalization(osv.osv):

	_name = 'cci_missions.legalization'

	_description = 'cci_missions.legalization'

	_inherits = {'cci_missions.dossier': 'dossier_id'}

	def _amount_total(self, cr, uid, ids, name, args, context=None):

		res ={}

		data_dosseir = self.browse(cr,uid,ids)

		for data in data_dosseir:

			if data.state =='draft':

				data_partner = self.pool.get('res.partner').browse(cr,uid,data.order_partner_id.id)

				force_member=force_non_member=False

				if data.member_price==1:

					force_member=True

				else:

					force_non_member=True

				context.update({'partner_id':data_partner})

				context.update({'force_member':force_member})

				context.update({'force_non_member':force_non_member})

				context.update({'date':data.date})

				context.update({'value_goods':data.goods_value})

				context.update({'pricelist':data_partner.property_product_pricelist.id})

				price_org = self.pool.get('product.product')._product_price(cr, uid, [data.type_id.original_product_id.id], False, False, context)

				price_copy = self.pool.get('product.product')._product_price(cr, uid, [data.type_id.copy_product_id.id], False, False, context)

				cost_org=price_org[data.type_id.original_product_id.id]

				cost_copy=price_copy[data.type_id.copy_product_id.id]

				qty_org = data.quantity_original

				qty_copy = data.quantity_copies

				subtotal =  data.sub_total

				if qty_org < 0 or qty_copy < 0:

					raise osv.except_osv('Input Error!','No. of Copies and Quantity of Originals should be positive.')

				total = ((cost_org * qty_org ) + (cost_copy * qty_copy) + subtotal)

				res[data.id] = total

			else :

				res[data.id]=data.invoiced_amount

		return res

	def cci_dossier_cancel_cci(self, cr, uid, ids, *args):

		data=self.browse(cr,uid,ids[0])

		if data.invoice_id.state == 'paid':

			new_ids = self.pool.get('account.invoice').refund(cr, uid,[data.invoice_id.id])

			self.write(cr, uid,ids, {'invoice_id' : new_ids[0]})

		else:

			wf_service = netsvc.LocalService('workflow')

			wf_service.trg_validate(uid, 'account.invoice', data.invoice_id.id, 'invoice_cancel', cr)