from osv import fields
import os
import tools
+import openerp
+from openerp import SUPERUSER_ID
from tools.translate import _
from tools.safe_eval import safe_eval as eval
multi_company_default()
-
class res_company(osv.osv):
_name = "res.company"
_description = 'Companies'
_order = 'name'
+
+ def _get_address_data(self, cr, uid, ids, field_names, arg, context=None):
+ """ Read the 'address' functional fields. """
+ result = {}
+ part_obj = self.pool.get('res.partner')
+ for company in self.browse(cr, uid, ids, context=context):
+ result[company.id] = {}.fromkeys(field_names, False)
+ if company.partner_id:
+ address_data = part_obj.address_get(cr, openerp.SUPERUSER_ID, [company.partner_id.id], adr_pref=['default'])
+ if address_data['default']:
+ address = part_obj.read(cr, openerp.SUPERUSER_ID, address_data['default'], field_names, context=context)
+ for field in field_names:
+ result[company.id][field] = address[field] or False
+ return result
+
+ def _set_address_data(self, cr, uid, company_id, name, value, arg, context=None):
+ """ Write the 'address' functional fields. """
+ company = self.browse(cr, uid, company_id, context=context)
+ if company.partner_id:
+ part_obj = self.pool.get('res.partner')
+ address_data = part_obj.address_get(cr, uid, [company.partner_id.id], adr_pref=['default'])
+ address = address_data['default']
+ if address:
+ part_obj.write(cr, uid, [address], {name: value or False})
+ else:
+ part_obj.create(cr, uid, {name: value or False, 'parent_id': company.partner_id.id}, context=context)
+ return True
+
_columns = {
- 'name': fields.char('Company Name', size=64, required=True),
+ 'name': fields.related('partner_id', 'name', string='Company Name', size=128, required=True, store=True, type='char'),
'parent_id': fields.many2one('res.company', 'Parent Company', select=True),
'child_ids': fields.one2many('res.company', 'parent_id', 'Child Companies'),
'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', required=True),
- 'rml_header2' : fields.text('RML Internal Header', required=True),
- 'rml_header3' : fields.text('RML Internal Header', required=True),
- 'logo' : fields.binary('Logo'),
+ 'rml_header': fields.text('RML Header', required=True),
+ 'rml_header1': fields.char('Company Slogan', size=200, help="Appears by default on the top right corner of your printed documents (report header)."),
+ 'rml_header2': fields.text('RML Internal Header', required=True),
+ 'rml_header3': fields.text('RML Internal Header for Landscape Reports', required=True),
+ 'rml_footer': fields.text('Report Footer', help="Footer text displayed at the bottom of all reports. Automatically set based on company details, "\
+ "but may also be customized by directly editing it."),
+ 'custom_footer': fields.boolean('Custom Footer', help="Check this to define the report footer manually. Otherwise it will be filled in automatically."),
+ 'logo': fields.related('partner_id', 'image', string="Logo", type="binary"),
'currency_id': fields.many2one('res.currency', 'Currency', required=True),
'currency_ids': fields.one2many('res.currency', 'company_id', 'Currency'),
'user_ids': fields.many2many('res.users', 'res_company_users_rel', 'cid', 'user_id', 'Accepted Users'),
'account_no':fields.char('Account No.', size=64),
+ 'street': fields.function(_get_address_data, fnct_inv=_set_address_data, size=128, type='char', string="Street", multi='address'),
+ 'street2': fields.function(_get_address_data, fnct_inv=_set_address_data, size=128, type='char', string="Street2", multi='address'),
+ 'zip': fields.function(_get_address_data, fnct_inv=_set_address_data, size=24, type='char', string="Zip", multi='address'),
+ 'city': fields.function(_get_address_data, fnct_inv=_set_address_data, size=24, type='char', string="City", multi='address'),
+ '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'),
+ 'bank_ids': fields.one2many('res.partner.bank','company_id', 'Bank Accounts', help='Bank accounts related to this company'),
+ 'country_id': fields.function(_get_address_data, fnct_inv=_set_address_data, type='many2one', relation='res.country', string="Country", multi='address'),
+ 'email': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Email", multi='address'),
+ 'phone': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Phone", multi='address'),
+ 'fax': fields.function(_get_address_data, fnct_inv=_set_address_data, size=64, type='char', string="Fax", multi='address'),
+ 'website': fields.related('partner_id', 'website', string="Website", type="char", size=64),
+ 'vat': fields.related('partner_id', 'vat', string="Tax ID", type="char", size=32),
+ 'company_registry': fields.char('Company Registry', size=64),
+ 'paper_format': fields.selection([('a4', 'A4'), ('us_letter', 'US Letter')], "Paper Format", required=True),
}
+ _sql_constraints = [
+ ('name_uniq', 'unique (name)', 'The company name must be unique !')
+ ]
+
+ def onchange_footer(self, cr, uid, ids, context=None):
+ # when touched, the footer becomes custom
+ return {'value': {'custom_footer': True}}
+
+ def set_auto_footer(self, cr, uid, ids, context=None):
+ # unset the flag 'custom_footer'; this will automatically compute the footer
+ return self.write(cr, uid, ids, {'custom_footer': False}, context=context)
+
+ def compute_footer(self, cr, uid, ids, context=None):
+ res_partner_bank = self.pool.get('res.partner.bank')
+ for company in self.browse(cr, uid, ids, context):
+ if not company.custom_footer:
+ # first line (notice that missing elements are filtered out before the join)
+ res = ' | '.join(filter(bool, [
+ company.phone and '%s: %s' % (_('Phone'), company.phone),
+ company.fax and '%s: %s' % (_('Fax'), company.fax),
+ company.email and '%s: %s' % (_('Email'), company.email),
+ company.website and '%s: %s' % (_('Website'), company.website),
+ company.vat and '%s: %s' % (_('TIN'), company.vat),
+ company.company_registry and '%s: %s' % (_('Reg'), company.company_registry),
+ ]))
+ # second line: bank accounts
+ account_ids = [acc.id for acc in company.bank_ids if acc.footer]
+ account_names = res_partner_bank.name_get(cr, uid, account_ids, context=context)
+ if account_names:
+ title = _('Bank Accounts') if len(account_names) > 1 else _('Bank Account')
+ res += '\n%s: %s' % (title, ', '.join(name for id, name in account_names))
+ # update footer
+ self.write(cr, uid, [company.id], {'rml_footer': res}, context=context)
+ return True
+
+ def on_change_country(self, cr, uid, ids, country_id, context=None):
+ currency_id = self._get_euro(cr, uid, context=context)
+ if country_id:
+ currency_id = self.pool.get('res.country').browse(cr, uid, country_id, context=context).currency_id.id
+ return {'value': {'currency_id': currency_id}}
def _search(self, cr, uid, args, offset=0, limit=None, order=None,
context=None, count=False, access_rights_uid=None):
-
if context is None:
context = {}
- user_preference = context.get('user_preference', False)
- if user_preference:
+ if context.get('user_preference'):
# We browse as superuser. Otherwise, the user would be able to
# select only the currently visible companies (according to rules,
# which are probably to allow to see the child companies) even if
# she belongs to some other companies.
- user = self.pool.get('res.users').browse(cr, 1, uid, context=context)
+ user = self.pool.get('res.users').browse(cr, SUPERUSER_ID, uid, context=context)
cmp_ids = list(set([user.company_id.id] + [cmp.id for cmp in user.company_ids]))
return cmp_ids
return super(res_company, self)._search(cr, uid, args, offset=offset, limit=limit, order=order,
return rule.company_dest_id.id
return user.company_id.id
- 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
-
@tools.ormcache()
def _get_company_children(self, cr, uid=None, company=None):
if not company:
ids = self.search(cr, uid, [('parent_id','child_of',[company])])
return ids
- def _get_partner_hierarchy(self, cr, uid, company_id, context={}):
+ def _get_partner_hierarchy(self, cr, uid, company_id, context=None):
if company_id:
parent_id = self.browse(cr, uid, company_id)['parent_id']
if parent_id:
return self._get_partner_descendance(cr, uid, company_id, [], context)
return []
- def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}):
+ def _get_partner_descendance(self, cr, uid, company_id, descendance, context=None):
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:
self.cache_restart(cr)
return super(res_company, self).create(cr, uid, vals, context=context)
obj_partner = self.pool.get('res.partner')
- partner_id = obj_partner.create(cr, uid, {'name': vals['name']}, context=context)
+ partner_id = obj_partner.create(cr, uid, {'name': vals['name'], 'is_company':True}, context=context)
vals.update({'partner_id': partner_id})
self.cache_restart(cr)
company_id = super(res_company, self).create(cr, uid, vals, context=context)
obj_partner.write(cr, uid, partner_id, {'company_id': company_id}, context=context)
+ self.compute_footer(cr, uid, [company_id], context=context)
return company_id
- def write(self, cr, *args, **argv):
+ def write(self, cr, uid, ids, values, context=None):
self.cache_restart(cr)
- return super(res_company, self).write(cr, *args, **argv)
-
- def _get_euro(self, cr, uid, context={}):
- try:
- return self.pool.get('res.currency').search(cr, uid, [])[0]
- except:
- return False
+ if isinstance(ids, (int, long)):
+ ids = [ids]
+ super(res_company, self).write(cr, uid, ids, values, context=context)
+ if 'rml_footer' not in values:
+ self.compute_footer(cr, uid, ids, context=context)
+ return True
+
+ def _get_euro(self, cr, uid, context=None):
+ rate_obj = self.pool.get('res.currency.rate')
+ rate_id = rate_obj.search(cr, uid, [('rate', '=', 1)], context=context)
+ return rate_id and rate_obj.browse(cr, uid, rate_id[0], context=context).currency_id.id or False
def _get_logo(self, cr, uid, ids):
- return open(os.path.join(
- tools.config['root_path'], '..', 'pixmaps', 'openerp-header.png'),
- 'rb') .read().encode('base64')
+ return open(os.path.join( tools.config['root_path'], 'addons', 'base', 'res', 'res_company_logo.png'), 'rb') .read().encode('base64')
_header = """
<header>
finally:
header_file.close()
except:
- return """
- <header>
+ return self._header_a4
+
+ _header_main = """
+<header>
<pageTemplate>
- <frame id="first" x1="1.3cm" y1="2.5cm" height="23.0cm" width="19cm"/>
+ <frame id="first" x1="1.3cm" y1="3.0cm" height="%s" width="19.0cm"/>
+ <stylesheet>
+ <paraStyle name="main_footer" fontName="DejaVu Sans" fontSize="8.0" alignment="CENTER"/>
+ </stylesheet>
<pageGraphics>
<!-- You Logo - Change X,Y,Width and Height -->
- <image x="1.3cm" y="27.6cm" height="40.0" >[[ company.logo or removeParentNode('image') ]]</image>
+ <image x="1.3cm" y="%s" height="40.0" >[[ company.logo or removeParentNode('image') ]]</image>
<setFont name="DejaVu Sans" size="8"/>
<fill color="black"/>
<stroke color="black"/>
- <lines>1.3cm 27.7cm 20cm 27.7cm</lines>
-
- <drawRightString x="20cm" y="27.8cm">[[ company.rml_header1 ]]</drawRightString>
-
- <drawString x="1.3cm" y="27.2cm">[[ company.partner_id.name ]]</drawString>
- <drawString x="1.3cm" y="26.8cm">[[ company.partner_id.address and company.partner_id.address[0].street or '' ]]</drawString>
- <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>
- <drawString x="1.3cm" y="26.0cm">Phone:</drawString>
- <drawRightString x="7cm" y="26.0cm">[[ company.partner_id.address and company.partner_id.address[0].phone or '' ]]</drawRightString>
- <drawString x="1.3cm" y="25.6cm">Mail:</drawString>
- <drawRightString x="7cm" y="25.6cm">[[ company.partner_id.address and company.partner_id.address[0].email or '' ]]</drawRightString>
- <lines>1.3cm 25.5cm 7cm 25.5cm</lines>
+ <!-- page header -->
+ <lines>1.3cm %s 20cm %s</lines>
+ <drawRightString x="20cm" y="%s">[[ company.rml_header1 ]]</drawRightString>
+ <drawString x="1.3cm" y="%s">[[ company.partner_id.name ]]</drawString>
+ <drawString x="1.3cm" y="%s">[[ company.partner_id.street or '' ]]</drawString>
+ <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>
+ <drawString x="1.3cm" y="%s">Phone:</drawString>
+ <drawRightString x="7cm" y="%s">[[ company.partner_id.phone or '' ]]</drawRightString>
+ <drawString x="1.3cm" y="%s">Mail:</drawString>
+ <drawRightString x="7cm" y="%s">[[ company.partner_id.email or '' ]]</drawRightString>
+ <lines>1.3cm %s 7cm %s</lines>
+
+ <!-- left margin -->
+ <rotate degrees="90"/>
+ <fill color="grey"/>
+ <drawString x="2.65cm" y="-0.4cm">produced by OpenERP.com</drawString>
+ <fill color="black"/>
+ <rotate degrees="-90"/>
<!--page bottom-->
-
- <lines>1.2cm 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 ]] - Page: <pageNumber/></drawCentredString>
+ <lines>1.2cm 2.65cm 19.9cm 2.65cm</lines>
+ <place x="1.3cm" y="0cm" height="2.55cm" width="19.0cm">
+ <para style="main_footer">[[ company.rml_footer ]]</para>
+ <para style="main_footer">Contact : [[ user.name ]] - Page: <pageNumber/></para>
+ </place>
</pageGraphics>
</pageTemplate>
</header>"""
+
+ _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')
+ _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')
+
+ def onchange_paper_format(self, cr, uid, ids, paper_format, context=None):
+ if paper_format == 'us_letter':
+ return {'value': {'rml_header': self._header_letter}}
+ return {'value': {'rml_header': self._header_a4}}
+
_defaults = {
'currency_id': _get_euro,
+ 'paper_format': 'a4',
'rml_header':_get_header,
'rml_header2': _header2,
'rml_header3': _header3,
- #'logo':_get_logo
+ 'logo':_get_logo
}
_constraints = [
(osv.osv._check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
]
+
res_company()
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: