1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as
9 # published by the Free Software Foundation, either version 3 of the
10 # License, or (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
23 from osv import fields
26 from tools.translate import _
29 class multi_company_default(osv.osv):
31 Manage multi company default value
33 _name = 'multi_company.default'
34 _description = 'Default multi company'
35 _order = 'company_id,sequence,id'
38 'sequence': fields.integer('Sequence'),
39 'name': fields.char('Name', size=256, required=True, help='Name it to easily find a record'),
40 'company_id': fields.many2one('res.company', 'Main Company', required=True,
41 help='Company where the user is connected'),
42 'company_dest_id': fields.many2one('res.company', 'Default Company', required=True,
43 help='Company to store the current record'),
44 'object_id': fields.many2one('ir.model', 'Object', required=True,
45 help='Object affect by this rules'),
46 'expression': fields.char('Expression', size=256, required=True,
47 help='Expression, must be True to match\nuse context.get or user (browse)'),
48 'field_id': fields.many2one('ir.model.fields', 'Field', help='Select field property'),
52 'expression': lambda *a: 'True',
53 'sequence': lambda *a: 100,
56 def copy(self, cr, uid, id, default=None, context=None):
58 Add (copy) in the name when duplicate record
64 company = self.browse(cr, uid, id, context=context)
65 default = default.copy()
66 default['name'] = company.name + _(' (copy)')
67 return super(multi_company_default, self).copy(cr, uid, id, default, context=context)
69 multi_company_default()
72 class res_company(osv.osv):
74 _description = 'Companies'
76 'name': fields.char('Company Name', size=64, required=True),
77 'parent_id': fields.many2one('res.company', 'Parent Company', select=True),
78 'child_ids': fields.one2many('res.company', 'parent_id', 'Child Companies'),
79 'partner_id': fields.many2one('res.partner', 'Partner', required=True),
80 'rml_header1': fields.char('Report Header', size=200),
81 'rml_footer1': fields.char('Report Footer 1', size=200),
82 'rml_footer2': fields.char('Report Footer 2', size=200),
83 'rml_header' : fields.text('RML Header'),
84 'rml_header2' : fields.text('RML Internal Header'),
85 'logo' : fields.binary('Logo'),
86 'currency_id': fields.many2one('res.currency', 'Currency', required=True),
87 'currency_ids': fields.one2many('res.currency', 'company_id', 'Currency'),
88 'user_ids': fields.many2many('res.users', 'res_company_users_rel', 'cid', 'user_id', 'Accepted Users')
91 def search(self, cr, user, args, offset=0, limit=None, order=None,
92 context=None, count=False):
93 if context and context.has_key('user_prefence') and context['user_prefence']:
95 data_user = self.pool.get('res.users').browse(cr, user, [user], context=context)
96 map(lambda x: cmp_ids.append(x.id), data_user[0].company_ids)
97 return [data_user[0].company_id.id] + cmp_ids
98 return super(res_company, self).search(cr, user, args, offset=offset, limit=limit, order=order,
99 context=context, count=count)
101 def _company_default_get(self, cr, uid, object=False, field=False, context=None):
103 Check if the object for this company have a default value
107 proxy = self.pool.get('multi_company.default')
109 ('object_id.model', '=', object),
112 args.append(('field_id.name','=',field))
114 args.append(('field_id','=',False))
115 ids = proxy.search(cr, uid, args, context=context)
116 for rule in proxy.browse(cr, uid, ids, context):
117 user = self.pool.get('res.users').browse(cr, uid, uid)
118 if eval(rule.expression, {'context': context, 'user': user}):
119 return rule.company_dest_id.id
120 user_company_id = self.pool.get('res.users').browse(cr, uid, uid).company_id.id
121 return user_company_id
123 def _get_child_ids(self, cr, uid, uid2, context={}):
124 company = self.pool.get('res.users').company_get(cr, uid, uid2)
125 ids = self._get_company_children(cr, uid, company)
128 def _get_company_children(self, cr, uid=None, company=None):
131 ids = self.search(cr, uid, [('parent_id','child_of',[company])])
133 _get_company_children = tools.cache()(_get_company_children)
135 def _get_partner_hierarchy(self, cr, uid, company_id, context={}):
137 parent_id = self.browse(cr, uid, company_id)['parent_id']
139 return self._get_partner_hierarchy(cr, uid, parent_id.id, context)
141 return self._get_partner_descendance(cr, uid, company_id, [], context)
144 def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}):
145 descendance.append(self.browse(cr, uid, company_id).partner_id.id)
146 for child_id in self._get_company_children(cr, uid, company_id):
147 if child_id != company_id:
148 descendance = self._get_partner_descendance(cr, uid, child_id, descendance)
152 # This function restart the cache on the _get_company_children method
154 def cache_restart(self, cr):
155 self._get_company_children.clear_cache(cr.dbname)
157 def create(self, cr, *args, **argv):
158 self.cache_restart(cr)
159 return super(res_company, self).create(cr, *args, **argv)
161 def write(self, cr, *args, **argv):
162 self.cache_restart(cr)
163 # Restart the cache on the company_get method
164 return super(res_company, self).write(cr, *args, **argv)
166 def _get_euro(self, cr, uid, context={}):
168 return self.pool.get('res.currency').search(cr, uid, [])[0]
172 def _check_recursion(self, cr, uid, ids):
175 cr.execute('select distinct parent_id from res_company where id in ('+','.join(map(str, ids))+')')
176 ids = filter(None, map(lambda x:x[0], cr.fetchall()))
182 def _get_header2(self,cr,uid,ids):
186 <frame id="first" x1="1.3cm" y1="1.5cm" width="18.4cm" height="26.5cm"/>
188 <fill color="black"/>
189 <stroke color="black"/>
190 <setFont name="DejaVu Sans" size="8"/>
191 <drawString x="1.3cm" y="28.3cm"> [[ formatLang(time.strftime("%Y-%m-%d"), date=True) ]] [[ time.strftime("%H:%M") ]]</drawString>
192 <setFont name="DejaVu Sans Bold" size="10"/>
193 <drawString x="9.8cm" y="28.3cm">[[ company.partner_id.name ]]</drawString>
194 <setFont name="DejaVu Sans" size="8"/>
195 <drawRightString x="19.7cm" y="28.3cm"><pageNumber/> / </drawRightString>
196 <drawString x="19.8cm" y="28.3cm"><pageCount/></drawString>
197 <stroke color="#000000"/>
198 <lines>1.3cm 28.1cm 20cm 28.1cm</lines>
202 def _get_header(self,cr,uid,ids):
204 return tools.file_open(os.path.join('base', 'report', 'corporate_rml_header.rml')).read()
209 <frame id="first" x1="1.3cm" y1="2.5cm" height="23.0cm" width="19cm"/>
211 <!-- You Logo - Change X,Y,Width and Height -->
212 <image x="1.3cm" y="27.6cm" height="40.0" >[[ company.logo or removeParentNode('image') ]]</image>
213 <setFont name="DejaVu Sans" size="8"/>
214 <fill color="black"/>
215 <stroke color="black"/>
216 <lines>1.3cm 27.7cm 20cm 27.7cm</lines>
218 <drawRightString x="20cm" y="27.8cm">[[ company.rml_header1 ]]</drawRightString>
221 <drawString x="1.3cm" y="27.2cm">[[ company.partner_id.name ]]</drawString>
222 <drawString x="1.3cm" y="26.8cm">[[ company.partner_id.address and company.partner_id.address[0].street or '' ]]</drawString>
223 <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>
224 <drawString x="1.3cm" y="26.0cm">Phone:</drawString>
225 <drawRightString x="7cm" y="26.0cm">[[ company.partner_id.address and company.partner_id.address[0].phone or '' ]]</drawRightString>
226 <drawString x="1.3cm" y="25.6cm">Mail:</drawString>
227 <drawRightString x="7cm" y="25.6cm">[[ company.partner_id.address and company.partner_id.address[0].email or '' ]]</drawRightString>
228 <lines>1.3cm 25.5cm 7cm 25.5cm</lines>
232 <lines>1.2cm 2.15cm 19.9cm 2.15cm</lines>
234 <drawCentredString x="10.5cm" y="1.7cm">[[ company.rml_footer1 ]]</drawCentredString>
235 <drawCentredString x="10.5cm" y="1.25cm">[[ company.rml_footer2 ]]</drawCentredString>
236 <drawCentredString x="10.5cm" y="0.8cm">Contact : [[ user.name ]] - Page: <pageNumber/></drawCentredString>
241 'currency_id': _get_euro,
242 'rml_header':_get_header,
243 'rml_header2': _get_header2
247 (_check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
252 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: