158 lines
6.1 KiB
Python
158 lines
6.1 KiB
Python
##############################################################################
|
|
#
|
|
# Copyright (c) 2004-2008 TINY SPRL. (http://tiny.be) All Rights Reserved.
|
|
#
|
|
# $Id$
|
|
#
|
|
# 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 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
##############################################################################
|
|
|
|
from osv import fields,osv
|
|
import tools
|
|
|
|
class res_company(osv.osv):
|
|
_name = "res.company"
|
|
|
|
_columns = {
|
|
'name': fields.char('Company Name', size=64, required=True),
|
|
'parent_id': fields.many2one('res.company', 'Parent Company', select=True),
|
|
'child_ids': fields.one2many('res.company', 'parent_id', 'Childs Company'),
|
|
'partner_id': fields.many2one('res.partner', 'Partner', required=True),
|
|
'rml_header1': fields.char('Report Header', size=200),
|
|
'rml_footer1': fields.char('Report Footer 1', size=200),
|
|
'rml_footer2': fields.char('Report Footer 2', size=200),
|
|
'rml_header' : fields.text('Rml Header'),
|
|
'logo' : fields.binary('Logo'),
|
|
'currency_id': fields.many2one('res.currency', 'Currency', required=True),
|
|
}
|
|
|
|
def _get_child_ids(self, cr, uid, uid2, context={}):
|
|
company = self.pool.get('res.users').company_get(cr, uid, uid2)
|
|
ids = self._get_company_children(cr, uid, company)
|
|
return ids
|
|
|
|
def _get_company_children(self, cr, uid=None, company=None):
|
|
if not company:
|
|
return []
|
|
ids = self.search(cr, uid, [('parent_id','child_of',[company])])
|
|
return ids
|
|
_get_company_children = tools.cache()(_get_company_children)
|
|
|
|
def _get_partner_hierarchy(self, cr, uid, company_id, context={}):
|
|
if company_id:
|
|
parent_id = self.browse(cr, uid, company_id)['parent_id']
|
|
if parent_id:
|
|
return self._get_partner_hierarchy(cr, uid, parent_id.id, context)
|
|
else:
|
|
return self._get_partner_descendance(cr, uid, company_id, [], context)
|
|
return []
|
|
|
|
def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}):
|
|
descendance.append(self.browse(cr, uid, company_id).partner_id.id)
|
|
for child_id in self._get_company_children(cr, uid, company_id):
|
|
if child_id != company_id:
|
|
descendance = self._get_partner_descendance(cr, uid, child_id, descendance)
|
|
return descendance
|
|
|
|
def __init__(self, *args, **argv):
|
|
return super(res_company, self).__init__(*args, **argv)
|
|
|
|
#
|
|
# This function restart the cache on the _get_company_children method
|
|
#
|
|
def cache_restart(self, uid=None):
|
|
self._get_company_children()
|
|
|
|
def create(self, *args, **argv):
|
|
self.cache_restart()
|
|
return super(res_company, self).create(*args, **argv)
|
|
|
|
def write(self, *args, **argv):
|
|
self.cache_restart()
|
|
# Restart the cache on the company_get method
|
|
self.pool.get('ir.rule').domain_get()
|
|
return super(res_company, self).write(*args, **argv)
|
|
|
|
def _get_euro(self, cr, uid, context={}):
|
|
try:
|
|
return self.pool.get('res.currency').search(cr, uid, [('rate', '=', 1.0),])[0]
|
|
except:
|
|
return 1
|
|
|
|
def _check_recursion(self, cr, uid, ids):
|
|
level = 100
|
|
while len(ids):
|
|
cr.execute('select distinct parent_id from res_company where id in ('+','.join(map(str,ids))+')')
|
|
ids = filter(None, map(lambda x:x[0], cr.fetchall()))
|
|
if not level:
|
|
return False
|
|
level -= 1
|
|
return True
|
|
|
|
def _get_header(self,cr,uid,ids):
|
|
try :
|
|
return tools.file_open('custom/corporate_rml_header.rml').read()
|
|
except:
|
|
return """<header>
|
|
<pageTemplate>
|
|
<frame id="first" x1="1cm" y1="2.5cm" width="19.0cm" height="23.0cm"/>
|
|
<pageGraphics>
|
|
<!--logo-->
|
|
<setFont name="Helvetica" size="30"/>
|
|
<fill color="black"/>
|
|
<stroke color="black"/>
|
|
<image x="13cm" y="25.4cm" height="2.4cm" name="logo"/>
|
|
<drawString x="1cm" y="27.8cm">[[ company.partner_id.name ]]</drawString>
|
|
<lines>1cm 27.7cm 20cm 27.7cm</lines>
|
|
<setFont name="Helvetica" size="10"/>
|
|
<drawRightString x="20cm" y="27.8cm">[[ company.rml_header1 ]]</drawRightString>
|
|
<drawString x="1cm" y="27.2cm">[[ company.partner_id.address and company.partner_id.address[0].street ]]</drawString>
|
|
<drawString x="1cm" y="26.8cm">[[ company.partner_id.address and company.partner_id.address[0].zip ]] [[ company.partner_id.address and company.partner_id.address[0].city ]] - [[ company.partner_id.address and company.partner_id.address[0].country_id and company.partner_id.address[0].country_id.name ]]</drawString>
|
|
<drawString x="1cm" y="26.4cm">Phone:</drawString>
|
|
<drawRightString x="7cm" y="26.4cm">[[ company.partner_id.address and company.partner_id.address[0].phone ]]</drawRightString>
|
|
<drawString x="1cm" y="26.0cm">Mail:</drawString>
|
|
<drawRightString x="7cm" y="26.0cm">[[ company.partner_id.address and company.partner_id.address[0].email ]]</drawRightString>
|
|
<lines>1cm 25.9cm 7cm 25.9cm</lines>
|
|
|
|
<!--page bottom-->
|
|
|
|
<lines>1.5cm 2.15cm 19.9cm 2.15cm</lines>
|
|
|
|
<drawCentredString x="10.5cm" y="1.7cm">[[ company.rml_footer1 ]]</drawCentredString>
|
|
<drawCentredString x="10.5cm" y="1.25cm">[[ company.rml_footer2 ]]</drawCentredString>
|
|
<drawCentredString x="10.5cm" y="0.8cm">Contact : [[ user.name ]]</drawCentredString>
|
|
</pageGraphics>
|
|
</pageTemplate>
|
|
</header>"""
|
|
_defaults = {
|
|
'currency_id': _get_euro,
|
|
'rml_header':_get_header
|
|
}
|
|
|
|
_constraints = [
|
|
(_check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
|
|
]
|
|
|
|
res_company()
|
|
|