############################################################################## # # Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved. # Fabien Pinckaers # # WARNING: This program as such is intended to be used by professional # programmers who take the whole responsability of assessing all potential # consequences resulting from its eventual inadequacies and bugs # End users who are looking for a ready-to-use solution with commercial # garantees and support are strongly adviced to contract a Free Software # Service Company # # This program is Free Software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # ############################################################################## from osv import fields,osv import tools class res_company(osv.osv): _name = "res.company" _columns = { 'name': fields.char('Company Name', size=64, required=True), 'parent_id': fields.many2one('res.company', 'Parent Company', select=True), 'child_ids': fields.one2many('res.company', 'parent_id', 'Childs Company'), '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), 'currency_id': fields.many2one('res.currency', 'Currency', required=True), } 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 def _get_company_children(self, cr, uid=None, company=None): if not company: return [] ids = self.search(cr, uid, [('parent_id','child_of',[company])]) return ids _get_company_children = tools.cache()(_get_company_children) def _get_partner_hierarchy(self, cr, uid, company_id, context={}): print "Company ID is %s: Looking for parent..." % company_id if company_id: parent_id = self.browse(cr, uid, company_id)['parent_id'] if parent_id: return self._get_partner_hierarchy(cr, uid, parent_id.id, context) else: print "No parent: starting descendance search!" return self._get_partner_descendance(cr, uid, company_id, [], context) return [] def _get_partner_descendance(self, cr, uid, company_id, descendance, context={}): 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: print "Hello, I'm %s, child of %s" % (child_id, company_id) descendance = self._get_partner_descendance(cr, uid, child_id, descendance) return descendance def __init__(self, *args, **argv): return super(res_company, self).__init__(*args, **argv) # # This function restart the cache on the _get_company_children method # def cache_restart(self, uid=None): self._get_company_children() def create(self, *args, **argv): self.cache_restart() return super(res_company, self).create(*args, **argv) def write(self, *args, **argv): self.cache_restart() return super(res_company, self).write(*args, **argv) def _get_euro(self, cr, uid, context={}): try: return self.pool.get('res.currency').search(cr, uid, [('rate', '=', 1.0),])[0] except: return 1 _defaults = { 'currency_id': _get_euro, } res_company()