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
27 from openerp import SUPERUSER_ID
28 from tools.translate import _
29 from tools.safe_eval import safe_eval as eval
31 class multi_company_default(osv.osv):
33 Manage multi company default value
35 _name = 'multi_company.default'
36 _description = 'Default multi company'
37 _order = 'company_id,sequence,id'
40 'sequence': fields.integer('Sequence'),
41 'name': fields.char('Name', size=256, required=True, help='Name it to easily find a record'),
42 'company_id': fields.many2one('res.company', 'Main Company', required=True,
43 help='Company where the user is connected'),
44 'company_dest_id': fields.many2one('res.company', 'Default Company', required=True,
45 help='Company to store the current record'),
46 'object_id': fields.many2one('ir.model', 'Object', required=True,
47 help='Object affected by this rule'),
48 'expression': fields.char('Expression', size=256, required=True,
49 help='Expression, must be True to match\nuse context.get or user (browse)'),
50 'field_id': fields.many2one('ir.model.fields', 'Field', help='Select field property'),
54 'expression': lambda *a: 'True',
55 'sequence': lambda *a: 100,
58 def copy(self, cr, uid, id, default=None, context=None):
60 Add (copy) in the name when duplicate record
66 company = self.browse(cr, uid, id, context=context)
67 default = default.copy()
68 default['name'] = company.name + _(' (copy)')
69 return super(multi_company_default, self).copy(cr, uid, id, default, context=context)
71 multi_company_default()
73 class res_company(osv.osv):
75 _description = 'Companies'
78 def _get_address_data(self, cr, uid, ids, field_names, arg, context=None):
79 """ Read the 'address' functional fields. """
81 part_obj = self.pool.get('res.partner')
82 for company in self.browse(cr, uid, ids, context=context):
83 result[company.id] = {}.fromkeys(field_names, False)
84 if company.partner_id:
85 address_data = part_obj.address_get(cr, openerp.SUPERUSER_ID, [company.partner_id.id], adr_pref=['default'])
86 if address_data['default']:
87 address = part_obj.read(cr, openerp.SUPERUSER_ID, address_data['default'], field_names, context=context)
88 for field in field_names:
89 result[company.id][field] = address[field] or False
92 def _set_address_data(self, cr, uid, company_id, name, value, arg, context=None):
93 """ Write the 'address' functional fields. """
94 company = self.browse(cr, uid, company_id, context=context)
95 if company.partner_id:
96 part_obj = self.pool.get('res.partner')
97 address_data = part_obj.address_get(cr, uid, [company.partner_id.id], adr_pref=['default'])
98 address = address_data['default']
100 part_obj.write(cr, uid, [address], {name: value or False})
102 part_obj.create(cr, uid, {name: value or False, 'parent_id': company.partner_id.id}, context=context)
105 def _get_rml_footer(self, cr, uid, ids, field_name, arg, context=None):
106 """ simply return the field 'rml_footer'; useful for the client """
107 companies = self.browse(cr, uid, ids, context=context)
108 return dict((company.id, company.rml_footer) for company in companies)
111 'name': fields.related('partner_id', 'name', string='Company Name', size=128, required=True, store=True, type='char'),
112 'parent_id': fields.many2one('res.company', 'Parent Company', select=True),
113 'child_ids': fields.one2many('res.company', 'parent_id', 'Child Companies'),
114 'partner_id': fields.many2one('res.partner', 'Partner', required=True),
115 'rml_header': fields.text('RML Header', required=True),
116 'rml_header1': fields.char('Company Slogan', size=200, help="Appears by default on the top right corner of your printed documents (report header)."),
117 'rml_header2': fields.text('RML Internal Header', required=True),
118 'rml_header3': fields.text('RML Internal Header for Landscape Reports', required=True),
119 'rml_footer': fields.text('Report Footer'),
120 'rml_footer_readonly': fields.function(_get_rml_footer, type='text', string='Report Footer'),
121 'custom_footer': fields.boolean('Custom Footer', help="Check this to define the report footer manually. Otherwise it will be filled in automatically."),
122 'logo': fields.related('partner_id', 'image', string="Logo", type="binary"),
123 'currency_id': fields.many2one('res.currency', 'Currency', required=True),
124 'currency_ids': fields.one2many('res.currency', 'company_id', 'Currency'),
125 'user_ids': fields.many2many('res.users', 'res_company_users_rel', 'cid', 'user_id', 'Accepted Users'),
126 'account_no':fields.char('Account No.', size=64),
127 'street': fields.function(_get_address_data, fnct_inv=_set_address_data, size=128, type='char', string="Street", multi='address'),
128 'street2': fields.function(_get_address_data, fnct_inv=_set_address_data, size=128, type='char', string="Street2", multi='address'),
129 'zip': fields.function(_get_address_data, fnct_inv=_set_address_data, size=24, type='char', string="Zip", multi='address'),
130 'city': fields.function(_get_address_data, fnct_inv=_set_address_data, size=24, type='char', string="City", multi='address'),
131 'state_id': fields.function(_get_address_data, fnct_inv=_set_address_data, type='many2one', domain="[('country_id', '=', country_id)]", relation='res.country.state', string="Fed. State", multi='address'),
132 'bank_ids': fields.one2many('res.partner.bank','company_id', 'Bank Accounts', help='Bank accounts related to this company'),
133 'country_id': fields.function(_get_address_data, fnct_inv=_set_address_data, type='many2one', relation='res.country', string="Country", multi='address'),
134 'email': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Email", multi='address'),
135 'phone': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Phone", multi='address'),
136 'fax': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Fax", multi='address'),
137 'website': fields.related('partner_id', 'website', string="Website", type="char", size=64),
138 'vat': fields.related('partner_id', 'vat', string="Tax ID", type="char", size=32),
139 'company_registry': fields.char('Company Registry', size=64),
140 'paper_format': fields.selection([('a4', 'A4'), ('us_letter', 'US Letter')], "Paper Format", required=True),
143 ('name_uniq', 'unique (name)', 'The company name must be unique !')
146 def on_change_footer(self, cr, uid, ids, custom_footer, phone, email, fax,
147 website, vat, company_registry, bank_ids, context=None):
151 # first line (notice that missing elements are filtered out before the join)
152 res = ' | '.join(filter(bool, [
153 phone and '%s: %s' % (_('Phone'), phone),
154 fax and '%s: %s' % (_('Fax'), fax),
155 email and '%s: %s' % (_('Email'), email),
156 website and '%s: %s' % (_('Website'), website),
157 vat and '%s: %s' % (_('TIN'), vat),
158 company_registry and '%s: %s' % (_('Reg'), company_registry),
161 # second line: bank accounts
162 accounts = self.resolve_2many_commands(cr, uid, 'bank_ids', bank_ids, context=context)
163 accounts_names = [('%(bank_name)s %(acc_number)s' % acc) for acc in accounts if acc['footer']]
165 title = _('Bank Accounts') if len(accounts_names) > 1 else _('Bank Account')
166 res += '\n%s: %s' % (title, ', '.join(accounts_names))
168 return {'value': {'rml_footer': res, 'rml_footer_readonly': res}}
170 def _get_rml_footer_by_line(self, cr, uid, ids, rml_footer, line, context=None):
171 rml_footer_lines = rml_footer.split('\n')
172 return rml_footer_lines[line] if line < len(rml_footer_lines) else ""
174 def _search(self, cr, uid, args, offset=0, limit=None, order=None,
175 context=None, count=False, access_rights_uid=None):
178 if context.get('user_preference'):
179 # We browse as superuser. Otherwise, the user would be able to
180 # select only the currently visible companies (according to rules,
181 # which are probably to allow to see the child companies) even if
182 # she belongs to some other companies.
183 user = self.pool.get('res.users').browse(cr, SUPERUSER_ID, uid, context=context)
184 cmp_ids = list(set([user.company_id.id] + [cmp.id for cmp in user.company_ids]))
186 return super(res_company, self)._search(cr, uid, args, offset=offset, limit=limit, order=order,
187 context=context, count=count, access_rights_uid=access_rights_uid)
189 def _company_default_get(self, cr, uid, object=False, field=False, context=None):
191 Check if the object for this company have a default value
195 proxy = self.pool.get('multi_company.default')
197 ('object_id.model', '=', object),
198 ('field_id', '=', field),
201 ids = proxy.search(cr, uid, args, context=context)
202 user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
203 for rule in proxy.browse(cr, uid, ids, context):
204 if eval(rule.expression, {'context': context, 'user': user}):
205 return rule.company_dest_id.id
206 return user.company_id.id
209 def _get_company_children(self, cr, uid=None, company=None):
212 ids = self.search(cr, uid, [('parent_id','child_of',[company])])
215 def _get_partner_hierarchy(self, cr, uid, company_id, context=None):
217 parent_id = self.browse(cr, uid, company_id)['parent_id']
219 return self._get_partner_hierarchy(cr, uid, parent_id.id, context)
221 return self._get_partner_descendance(cr, uid, company_id, [], context)
224 def _get_partner_descendance(self, cr, uid, company_id, descendance, context=None):
225 descendance.append(self.browse(cr, uid, company_id).partner_id.id)
226 for child_id in self._get_company_children(cr, uid, company_id):
227 if child_id != company_id:
228 descendance = self._get_partner_descendance(cr, uid, child_id, descendance)
232 # This function restart the cache on the _get_company_children method
234 def cache_restart(self, cr):
235 self._get_company_children.clear_cache(self)
237 def create(self, cr, uid, vals, context=None):
238 if not vals.get('name', False) or vals.get('partner_id', False):
239 self.cache_restart(cr)
240 return super(res_company, self).create(cr, uid, vals, context=context)
241 obj_partner = self.pool.get('res.partner')
242 partner_id = obj_partner.create(cr, uid, {'name': vals['name'], 'is_company':True}, context=context)
243 vals.update({'partner_id': partner_id})
244 self.cache_restart(cr)
245 company_id = super(res_company, self).create(cr, uid, vals, context=context)
246 obj_partner.write(cr, uid, partner_id, {'company_id': company_id}, context=context)
249 def write(self, cr, *args, **argv):
250 self.cache_restart(cr)
251 return super(res_company, self).write(cr, *args, **argv)
253 def _get_euro(self, cr, uid, context=None):
255 return self.pool.get('res.currency').search(cr, uid, [])[0]
259 def _get_logo(self, cr, uid, ids):
260 return open(os.path.join( tools.config['root_path'], 'addons', 'base', 'res', 'res_company_logo.png'), 'rb') .read().encode('base64')
265 <frame id="first" x1="28.0" y1="28.0" width="%s" height="%s"/>
267 <fill color="black"/>
268 <stroke color="black"/>
269 <setFont name="DejaVu Sans" size="8"/>
270 <drawString x="%s" y="%s"> [[ formatLang(time.strftime("%%Y-%%m-%%d"), date=True) ]] [[ time.strftime("%%H:%%M") ]]</drawString>
271 <setFont name="DejaVu Sans Bold" size="10"/>
272 <drawCentredString x="%s" y="%s">[[ company.partner_id.name ]]</drawCentredString>
273 <stroke color="#000000"/>
279 _header2 = _header % (539, 772, "1.0cm", "28.3cm", "11.1cm", "28.3cm", "1.0cm 28.1cm 20.1cm 28.1cm")
281 _header3 = _header % (786, 525, 25, 555, 440, 555, "25 550 818 550")
283 def _get_header(self,cr,uid,ids):
285 header_file = tools.file_open(os.path.join('base', 'report', 'corporate_rml_header.rml'))
287 return header_file.read()
291 return self._header_a4
296 <frame id="first" x1="1.3cm" y1="3.0cm" height="%s" width="19.0cm"/>
298 <!-- You Logo - Change X,Y,Width and Height -->
299 <image x="1.3cm" y="%s" height="40.0" >[[ company.logo or removeParentNode('image') ]]</image>
300 <setFont name="DejaVu Sans" size="8"/>
301 <fill color="black"/>
302 <stroke color="black"/>
303 <lines>1.3cm %s 20cm %s</lines>
305 <drawRightString x="20cm" y="%s">[[ company.rml_header1 ]]</drawRightString>
308 <drawString x="1.3cm" y="%s">[[ company.partner_id.name ]]</drawString>
309 <drawString x="1.3cm" y="%s">[[ company.partner_id.street or '' ]]</drawString>
310 <drawString x="1.3cm" y="%s">[[ company.partner_id.city or '' ]] - [[ company.partner_id.country_id and company.partner_id.country_id.name or '']]</drawString>
311 <drawString x="1.3cm" y="%s">Phone:</drawString>
312 <drawRightString x="7cm" y="%s">[[ company.partner_id.phone or '' ]]</drawRightString>
313 <drawString x="1.3cm" y="%s">Mail:</drawString>
314 <drawRightString x="7cm" y="%s">[[ company.partner_id.email or '' ]]</drawRightString>
315 <lines>1.3cm %s 7cm %s</lines>
319 <lines>1.2cm 2.65cm 19.9cm 2.65cm</lines>
320 <drawCentredString x="10.5cm" y="2.3cm">[[ company._get_rml_footer_by_line(company.rml_footer,line=0) ]]</drawCentredString>
321 <drawCentredString x="10.5cm" y="1.8cm">[[ company._get_rml_footer_by_line(company.rml_footer,line=1) ]]</drawCentredString>
322 <drawCentredString x="10.5cm" y="1.3cm">[[ company._get_rml_footer_by_line(company.rml_footer,line=2) ]]</drawCentredString>
323 <drawCentredString x="10.5cm" y="0.8cm">Contact : [[ user.name ]] - Page: <pageNumber/></drawCentredString>
328 _header_a4 = _header_main % ('23.0cm', '27.6cm', '27.7cm', '27.7cm', '27.8cm', '27.2cm', '26.8cm', '26.4cm', '26.0cm', '26.0cm', '25.6cm', '25.6cm', '25.5cm', '25.5cm')
329 _header_letter = _header_main % ('21.3cm', '25.9cm', '26.0cm', '26.0cm', '26.1cm', '25.5cm', '25.1cm', '24.7cm', '24.3cm', '24.3cm', '23.9cm', '23.9cm', '23.8cm', '23.8cm')
331 def onchange_paper_format(self, cr, uid, ids, paper_format, context=None):
332 if paper_format == 'us_letter':
333 return {'value': {'rml_header': self._header_letter}}
334 return {'value': {'rml_header': self._header_a4}}
337 'currency_id': _get_euro,
338 'paper_format': 'a4',
339 'rml_header':_get_header,
340 'rml_header2': _header2,
341 'rml_header3': _header3,
346 (osv.osv._check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
352 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: