bin/addons/base/res/res_company.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
   6 #
   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.
  11 #
  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.
  16 #
  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/>.
  19 #
  20 ##############################################################################
  21
  22 from osv import osv
  23 from osv import fields
  24 import os
  25 import tools
  26 from tools.translate import _
  27 from tools.safe_eval import safe_eval as eval
  28
  29 class multi_company_default(osv.osv):
  30     """
  31     Manage multi company default value
  32     """
  33     _name = 'multi_company.default'
  34     _description = 'Default multi company'
  35     _order = 'company_id,sequence,id'
  36
  37     _columns = {
  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 affected by this rule'),
  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'),
  49     }
  50
  51     _defaults = {
  52         'expression': lambda *a: 'True',
  53         'sequence': lambda *a: 100,
  54     }
  55
  56     def copy(self, cr, uid, id, default=None, context=None):
  57         """
  58         Add (copy) in the name when duplicate record
  59         """
  60         if not context:
  61             context = {}
  62         if not default:
  63             default = {}
  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)
  68
  69 multi_company_default()
  70
  71
  72 class res_company(osv.osv):
  73     _name = "res.company"
  74     _description = 'Companies'
  75     _columns = {
  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         'rml_header3' : fields.text('RML Internal Header'),
  86         'logo' : fields.binary('Logo'),
  87         'currency_id': fields.many2one('res.currency', 'Currency', required=True),
  88         'currency_ids': fields.one2many('res.currency', 'company_id', 'Currency'),
  89         'user_ids': fields.many2many('res.users', 'res_company_users_rel', 'cid', 'user_id', 'Accepted Users'),
  90         'account_no':fields.char('Account No.', size=64),
  91     }
  92
  93     def search(self, cr, uid, args, offset=0, limit=None, order=None,
  94             context=None, count=False):
  95
  96         if context is None:
  97             context = {}
  98         user_preference = context.get('user_preference', False)
  99         if user_preference:
 100             # TODO: improve this as soon as the client sends the proper
 101             # combination of active_id and active_model we'll be able to
 102             # use active_id here to restrict to the user being modified instead
 103             # of current user.
 104             user_id = context.get('user_id', uid)
 105
 106             user = self.pool.get('res.users').browse(cr, uid, user_id, context=context)
 107             cmp_ids = list(set([user.company_id.id] + [cmp.id for cmp in user.company_ids]))
 108             return cmp_ids
 109         return super(res_company, self).search(cr, uid, args, offset=offset, limit=limit, order=order,
 110             context=context, count=count)
 111
 112     def _company_default_get(self, cr, uid, object=False, field=False, context=None):
 113         """
 114         Check if the object for this company have a default value
 115         """
 116         if not context:
 117             context = {}
 118         proxy = self.pool.get('multi_company.default')
 119         args = [
 120             ('object_id.model', '=', object),
 121             ('field_id', '=', field),
 122         ]
 123
 124         ids = proxy.search(cr, uid, args, context=context)
 125         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
 126         for rule in proxy.browse(cr, uid, ids, context):
 127             if eval(rule.expression, {'context': context, 'user': user}):
 128                 return rule.company_dest_id.id
 129         return user.company_id.id
 130
 131     def _get_child_ids(self, cr, uid, uid2, context={}):
 132         company = self.pool.get('res.users').company_get(cr, uid, uid2)
 133         ids = self._get_company_children(cr, uid, company)
 134         return ids
 135
 136     @tools.cache()
 137     def _get_company_children(self, cr, uid=None, company=None):
 138         if not company:
 139             return []
 140         ids =  self.search(cr, uid, [('parent_id','child_of',[company])])
 141         return ids
 142
 143     def _get_partner_hierarchy(self, cr, uid, company_id, context={}):
 144         if company_id:
 145             parent_id = self.browse(cr, uid, company_id)['parent_id']
 146             if parent_id:
 147                 return self._get_partner_hierarchy(cr, uid, parent_id.id, context)
 148             else:
 149                 return self._get_partner_descendance(cr, uid, company_id, [], context)
 150         return []
 151
 152     def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}):
 153         descendance.append(self.browse(cr, uid, company_id).partner_id.id)
 154         for child_id in self._get_company_children(cr, uid, company_id):
 155             if child_id != company_id:
 156                 descendance = self._get_partner_descendance(cr, uid, child_id, descendance)
 157         return descendance
 158
 159     #
 160     # This function restart the cache on the _get_company_children method
 161     #
 162     def cache_restart(self, cr):
 163         self._get_company_children.clear_cache(cr.dbname)
 164
 165     def create(self, cr, uid, vals, context=None):
 166         if not vals.get('name', False) or vals.get('partner_id', False):
 167             self.cache_restart(cr)
 168             return super(res_company, self).create(cr, uid, vals, context=context)
 169         obj_partner = self.pool.get('res.partner')
 170         partner_id = obj_partner.create(cr, uid, {'name': vals['name']}, context=context)
 171         vals.update({'partner_id': partner_id})
 172         self.cache_restart(cr)
 173         company_id = super(res_company, self).create(cr, uid, vals, context=context)
 174         obj_partner.write(cr, uid, partner_id, {'company_id': company_id}, context=context)
 175         return company_id
 176
 177     def write(self, cr, *args, **argv):
 178         self.cache_restart(cr)
 179         return super(res_company, self).write(cr, *args, **argv)
 180
 181     def _get_euro(self, cr, uid, context={}):
 182         try:
 183             return self.pool.get('res.currency').search(cr, uid, [])[0]
 184         except:
 185             return False
 186
 187     def _check_recursion(self, cr, uid, ids):
 188         level = 100
 189         while len(ids):
 190             cr.execute('select distinct parent_id from res_company where id IN %s',(tuple(ids),))
 191             ids = filter(None, map(lambda x:x[0], cr.fetchall()))
 192             if not level:
 193                 return False
 194             level -= 1
 195         return True
 196
 197     def _get_logo(self, cr, uid, ids):
 198         return open(os.path.join(
 199             tools.config['root_path'], '..', 'pixmaps', 'openerp-header.png'),
 200                     'rb') .read().encode('base64')
 201
 202     def _get_header3(self,cr,uid,ids):
 203         return """
 204 <header>
 205 <pageTemplate>
 206     <frame id="first" x1="22.0" y1="22.0" width="1080" height="700"/>
 207     <pageGraphics>
 208         <fill color="black"/>
 209         <stroke color="black"/>
 210         <setFont name="DejaVu Sans" size="8"/>
 211         <drawString x="25" y="725"> [[ formatLang(time.strftime("%Y-%m-%d"), date=True) ]]  [[ time.strftime("%H:%M") ]]</drawString>
 212         <setFont name="DejaVu Sans Bold" size="10"/>
 213         <drawString x="490" y="725">[[ company.partner_id.name ]]</drawString>
 214         <stroke color="#000000"/>
 215         <lines>25 720 1085 720</lines>
 216     </pageGraphics>
 217     </pageTemplate>
 218 </header>"""
 219     def _get_header2(self,cr,uid,ids):
 220         return """
 221         <header>
 222         <pageTemplate>
 223         <frame id="first" x1="1.3cm" y1="1.5cm" width="18.4cm" height="26.5cm"/>
 224         <pageGraphics>
 225         <fill color="black"/>
 226         <stroke color="black"/>
 227         <setFont name="DejaVu Sans" size="8"/>
 228         <drawString x="1.3cm" y="28.3cm"> [[ formatLang(time.strftime("%Y-%m-%d"), date=True) ]]  [[ time.strftime("%H:%M") ]]</drawString>
 229         <setFont name="DejaVu Sans Bold" size="10"/>
 230         <drawString x="9.8cm" y="28.3cm">[[ company.partner_id.name ]]</drawString>
 231         <stroke color="#000000"/>
 232         <lines>1.3cm 28.1cm 20cm 28.1cm</lines>
 233         </pageGraphics>
 234         </pageTemplate>
 235 </header>"""
 236     def _get_header(self,cr,uid,ids):
 237         try :
 238             return tools.file_open(os.path.join('base', 'report', 'corporate_rml_header.rml')).read()
 239         except:
 240             return """
 241     <header>
 242     <pageTemplate>
 243         <frame id="first" x1="1.3cm" y1="2.5cm" height="23.0cm" width="19cm"/>
 244         <pageGraphics>
 245             <!-- You Logo - Change X,Y,Width and Height -->
 246             <image x="1.3cm" y="27.6cm" height="40.0" >[[ company.logo or removeParentNode('image') ]]</image>
 247             <setFont name="DejaVu Sans" size="8"/>
 248             <fill color="black"/>
 249             <stroke color="black"/>
 250             <lines>1.3cm 27.7cm 20cm 27.7cm</lines>
 251
 252             <drawRightString x="20cm" y="27.8cm">[[ company.rml_header1 ]]</drawRightString>
 253
 254
 255             <drawString x="1.3cm" y="27.2cm">[[ company.partner_id.name ]]</drawString>
 256             <drawString x="1.3cm" y="26.8cm">[[ company.partner_id.address and company.partner_id.address[0].street or  '' ]]</drawString>
 257             <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>
 258             <drawString x="1.3cm" y="26.0cm">Phone:</drawString>
 259             <drawRightString x="7cm" y="26.0cm">[[ company.partner_id.address and company.partner_id.address[0].phone or '' ]]</drawRightString>
 260             <drawString x="1.3cm" y="25.6cm">Mail:</drawString>
 261             <drawRightString x="7cm" y="25.6cm">[[ company.partner_id.address and company.partner_id.address[0].email or '' ]]</drawRightString>
 262             <lines>1.3cm 25.5cm 7cm 25.5cm</lines>
 263
 264             <!--page bottom-->
 265
 266             <lines>1.2cm 2.15cm 19.9cm 2.15cm</lines>
 267
 268             <drawCentredString x="10.5cm" y="1.7cm">[[ company.rml_footer1 ]]</drawCentredString>
 269             <drawCentredString x="10.5cm" y="1.25cm">[[ company.rml_footer2 ]]</drawCentredString>
 270             <drawCentredString x="10.5cm" y="0.8cm">Contact : [[ user.name ]] - Page: <pageNumber/></drawCentredString>
 271         </pageGraphics>
 272     </pageTemplate>
 273 </header>"""
 274     _defaults = {
 275         'currency_id': _get_euro,
 276         'rml_header':_get_header,
 277         'rml_header2': _get_header2,
 278         'rml_header3': _get_header3,
 279         #'logo':_get_logo
 280     }
 281
 282     _constraints = [
 283         (_check_recursion, 'Error! You can not create recursive companies.', ['parent_id'])
 284     ]
 285
 286 res_company()
 287
 288 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 289