1 # -*- encoding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
8 # This program is free software: you can redistribute it and/or modify
9 # it under the terms of the GNU General Public License as published by
10 # the Free Software Foundation, either version 3 of the License, or
11 # (at your option) any later version.
13 # This program is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU General Public License for more details.
18 # You should have received a copy of the GNU General Public License
19 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 ##############################################################################
23 from osv import fields,osv
26 class res_company(osv.osv):
30 'name': fields.char('Company Name', size=64, required=True),
31 'parent_id': fields.many2one('res.company', 'Parent Company', select=True),
32 'child_ids': fields.one2many('res.company', 'parent_id', 'Child Companies'),
33 'partner_id': fields.many2one('res.partner', 'Partner', required=True),
34 'rml_header1': fields.char('Report Header', size=200),
35 'rml_footer1': fields.char('Report Footer 1', size=200),
36 'rml_footer2': fields.char('Report Footer 2', size=200),
37 'rml_header' : fields.text('RML Header'),
38 'rml_header2' : fields.text('RML Internal Header'),
39 'logo' : fields.binary('Logo'),
40 'currency_id': fields.many2one('res.currency', 'Currency', required=True),
43 def _get_child_ids(self, cr, uid, uid2, context={}):
44 company = self.pool.get('res.users').company_get(cr, uid, uid2)
45 ids = self._get_company_children(cr, uid, company)
48 def _get_company_children(self, cr, uid=None, company=None):
51 ids = self.search(cr, uid, [('parent_id','child_of',[company])])
53 _get_company_children = tools.cache()(_get_company_children)
55 def _get_partner_hierarchy(self, cr, uid, company_id, context={}):
57 parent_id = self.browse(cr, uid, company_id)['parent_id']
59 return self._get_partner_hierarchy(cr, uid, parent_id.id, context)
61 return self._get_partner_descendance(cr, uid, company_id, [], context)
64 def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}):
65 descendance.append(self.browse(cr, uid, company_id).partner_id.id)
66 for child_id in self._get_company_children(cr, uid, company_id):
67 if child_id != company_id:
68 descendance = self._get_partner_descendance(cr, uid, child_id, descendance)
72 # This function restart the cache on the _get_company_children method
74 def cache_restart(self, cr):
75 self._get_company_children.clear_cache(cr.dbname)
77 def create(self, cr, *args, **argv):
78 self.cache_restart(cr)
79 return super(res_company, self).create(cr, *args, **argv)
81 def write(self, cr, *args, **argv):
82 self.cache_restart(cr)
83 # Restart the cache on the company_get method
84 self.pool.get('ir.rule').domain_get.clear_cache(cr.dbname)
85 return super(res_company, self).write(cr, *args, **argv)
87 def _get_euro(self, cr, uid, context={}):
89 return self.pool.get('res.currency').search(cr, uid, [])[0]
93 def _check_recursion(self, cr, uid, ids):
96 cr.execute('select distinct parent_id from res_company where id in ('+','.join(map(str,ids))+')')
97 ids = filter(None, map(lambda x:x[0], cr.fetchall()))
103 def _get_header2(self,cr,uid,ids):
107 <frame id="first" x1="1.3cm" y1="1.5cm" width="18.4cm" height="26.5cm"/>
109 <fill color="black"/>
110 <stroke color="black"/>
111 <setFont name="Helvetica" size="8"/>
112 <drawString x="1.3cm" y="28.3cm"> [[ formatLang(time.strftime("%Y-%m-%d"), date=True) ]] [[ time.strftime("%H:%M") ]]</drawString>
113 <setFont name="Helvetica-Bold" size="10"/>
114 <drawString x="9.8cm" y="28.3cm">[[ company.partner_id.name ]]</drawString>
115 <setFont name="Helvetica" size="8"/>
116 <drawRightString x="19.7cm" y="28.3cm"><pageNumber/> / </drawRightString>
117 <drawString x="19.8cm" y="28.3cm"><pageCount/></drawString>
118 <stroke color="#000000"/>
119 <lines>1.3cm 28.1cm 20cm 28.1cm</lines>
123 def _get_header(self,cr,uid,ids):
125 return tools.file_open('custom/corporate_rml_header.rml').read()
130 <frame id="first" x1="1.3cm" y1="2.5cm" height="23.0cm" width="19cm"/>
132 <!-- You Logo - Change X,Y,Width and Height -->
133 <image x="1.3cm" y="27.6cm" height="40.0" >[[company.logo]]</image>
134 <setFont name="Helvetica" size="8"/>
135 <fill color="black"/>
136 <stroke color="black"/>
137 <lines>1.3cm 27.7cm 20cm 27.7cm</lines>
139 <drawRightString x="20cm" y="27.8cm">[[ company.rml_header1 ]]</drawRightString>
142 <drawString x="1.3cm" y="27.2cm">[[ company.partner_id.name ]]</drawString>
143 <drawString x="1.3cm" y="26.8cm">[[ company.partner_id.address and company.partner_id.address[0].street or '' ]]</drawString>
144 <drawString x="1.3cm" y="26.4cm">[[ company.partner_id.address and company.partner_id.address[0].zip or '' ]] [[ company.partner_id.address and company.partner_id.address[0].city or '' ]] - [[ company.partner_id.address and company.partner_id.address[0].country_id and company.partner_id.address[0].country_id.name or '']]</drawString>
145 <drawString x="1.3cm" y="26.0cm">Phone:</drawString>
146 <drawRightString x="7cm" y="26.0cm">[[ company.partner_id.address and company.partner_id.address[0].phone or '' ]]</drawRightString>
147 <drawString x="1.3cm" y="25.6cm">Mail:</drawString>
148 <drawRightString x="7cm" y="25.6cm">[[ company.partner_id.address and company.partner_id.address[0].email or '' ]]</drawRightString>
149 <lines>1.3cm 25.5cm 7cm 25.5cm</lines>
153 <lines>1.2cm 2.15cm 19.9cm 2.15cm</lines>
155 <drawCentredString x="10.5cm" y="1.7cm">[[ company.rml_footer1 ]]</drawCentredString>
156 <drawCentredString x="10.5cm" y="1.25cm">[[ company.rml_footer2 ]]</drawCentredString>
157 <drawCentredString x="10.5cm" y="0.8cm">Contact : [[ user.name ]] - Page: <pageNumber/></drawCentredString>
162 'currency_id': _get_euro,
163 'rml_header':_get_header,
164 'rml_header2': _get_header2
168 (_check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
174 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: