Source code for openerp.addons.magentoerpconnect.partner

# -*- coding: utf-8 -*-
##############################################################################
#
#    Author: Guewen Baconnier
#    Copyright 2013 Camptocamp SA
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

import logging
import xmlrpclib
from collections import namedtuple
from openerp import models, fields, api
from openerp.addons.connector.queue.job import job
from openerp.addons.connector.connector import ConnectorUnit
from openerp.addons.connector.exception import MappingError
from openerp.addons.connector.unit.backend_adapter import BackendAdapter
from openerp.addons.connector.unit.mapper import (mapping,
                                                  only_create,
                                                  ImportMapper
                                                  )
from openerp.addons.connector.exception import IDMissingInBackend
from .unit.backend_adapter import (GenericAdapter,
                                   MAGENTO_DATETIME_FORMAT,
                                   )
from .unit.import_synchronizer import (DelayedBatchImporter,
                                       MagentoImporter,
                                       )
from .unit.mapper import normalize_datetime
from .backend import magento
from .connector import get_environment

_logger = logging.getLogger(__name__)


[docs]class ResPartner(models.Model): _inherit = 'res.partner' magento_bind_ids = fields.One2many( comodel_name='magento.res.partner', inverse_name='openerp_id', string="Magento Bindings", ) magento_address_bind_ids = fields.One2many( comodel_name='magento.address', inverse_name='openerp_id', string="Magento Address Bindings", ) birthday = fields.Date(string='Birthday') company = fields.Char(string='Company') @api.model def _address_fields(self): """ Returns the list of address fields that are synced from the parent when the `use_parent_address` flag is set. """ fields = super(ResPartner, self)._address_fields() fields.append('company') return fields
[docs]class MagentoResPartner(models.Model): _name = 'magento.res.partner' _inherit = 'magento.binding' _inherits = {'res.partner': 'openerp_id'} _description = 'Magento Partner' _rec_name = 'name' openerp_id = fields.Many2one(comodel_name='res.partner', string='Partner', required=True, ondelete='cascade') backend_id = fields.Many2one( related='website_id.backend_id', comodel_name='magento.backend', string='Magento Backend', store=True, readonly=True, # override 'magento.binding', can't be INSERTed if True: required=False, ) website_id = fields.Many2one(comodel_name='magento.website', string='Magento Website', required=True, ondelete='restrict') group_id = fields.Many2one(comodel_name='magento.res.partner.category', string='Magento Group (Category)') created_at = fields.Datetime(string='Created At (on Magento)', readonly=True) updated_at = fields.Datetime(string='Updated At (on Magento)', readonly=True) emailid = fields.Char(string='E-mail address') taxvat = fields.Char(string='Magento VAT') newsletter = fields.Boolean(string='Newsletter') guest_customer = fields.Boolean(string='Guest Customer') consider_as_company = fields.Boolean( string='Considered as company', help="An account imported with a 'company' in " "the billing address is considered as a company.\n " "The partner takes the name of the company and " "is not merged with the billing address.", )
[docs]class MagentoAddress(models.Model): _name = 'magento.address' _inherit = 'magento.binding' _inherits = {'res.partner': 'openerp_id'} _description = 'Magento Address' _rec_name = 'backend_id' openerp_id = fields.Many2one(comodel_name='res.partner', string='Partner', required=True, ondelete='cascade') created_at = fields.Datetime(string='Created At (on Magento)', readonly=True) updated_at = fields.Datetime(string='Updated At (on Magento)', readonly=True) is_default_billing = fields.Boolean(string='Default Invoice') is_default_shipping = fields.Boolean(string='Default Shipping') magento_partner_id = fields.Many2one(comodel_name='magento.res.partner', string='Magento Partner', required=True, ondelete='cascade') backend_id = fields.Many2one( related='magento_partner_id.backend_id', comodel_name='magento.backend', string='Magento Backend', store=True, readonly=True, # override 'magento.binding', can't be INSERTed if True: required=False, ) website_id = fields.Many2one( related='magento_partner_id.website_id', comodel_name='magento.website', string='Magento Website', store=True, readonly=True, ) is_magento_order_address = fields.Boolean( string='Address from a Magento Order', ) _sql_constraints = [ ('openerp_uniq', 'unique(backend_id, openerp_id)', 'A partner address can only have one binding by backend.'), ]
[docs]@magento class PartnerAdapter(GenericAdapter): _model_name = 'magento.res.partner' _magento_model = 'customer' _admin_path = '/{model}/edit/id/{id}' def _call(self, method, arguments): try: return super(PartnerAdapter, self)._call(method, arguments) except xmlrpclib.Fault as err: # this is the error in the Magento API # when the customer does not exist if err.faultCode == 102: raise IDMissingInBackend else: raise
[docs] def search(self, filters=None, from_date=None, to_date=None, magento_website_ids=None): """ Search records according to some criteria and return a list of ids :rtype: list """ if filters is None: filters = {} dt_fmt = MAGENTO_DATETIME_FORMAT if from_date is not None: # updated_at include the created records filters.setdefault('updated_at', {}) filters['updated_at']['from'] = from_date.strftime(dt_fmt) if to_date is not None: filters.setdefault('updated_at', {}) filters['updated_at']['to'] = to_date.strftime(dt_fmt) if magento_website_ids is not None: filters['website_id'] = {'in': magento_website_ids} # the search method is on ol_customer instead of customer return self._call('ol_customer.search', [filters] if filters else [{}])
[docs]@magento class PartnerBatchImporter(DelayedBatchImporter): """ Import the Magento Partners. For every partner in the list, a delayed job is created. """ _model_name = ['magento.res.partner']
[docs] def run(self, filters=None): """ Run the synchronization """ from_date = filters.pop('from_date', None) to_date = filters.pop('to_date', None) magento_website_ids = [filters.pop('magento_website_id')] record_ids = self.backend_adapter.search( filters, from_date=from_date, to_date=to_date, magento_website_ids=magento_website_ids) _logger.info('search for magento partners %s returned %s', filters, record_ids) for record_id in record_ids: self._import_record(record_id)
PartnerBatchImport = PartnerBatchImporter # deprecated
[docs]@magento class PartnerImportMapper(ImportMapper): _model_name = 'magento.res.partner' direct = [ ('email', 'email'), ('dob', 'birthday'), (normalize_datetime('created_at'), 'created_at'), (normalize_datetime('updated_at'), 'updated_at'), ('email', 'emailid'), ('taxvat', 'taxvat'), ('group_id', 'group_id'), ]
[docs] @only_create @mapping def is_company(self, record): # partners are companies so we can bind # addresses on them return {'is_company': True}
[docs] @mapping def names(self, record): # TODO create a glue module for base_surname parts = [part for part in (record['firstname'], record['middlename'], record['lastname']) if part] return {'name': ' '.join(parts)}
[docs] @mapping def customer_group_id(self, record): # import customer groups binder = self.binder_for(model='magento.res.partner.category') category_id = binder.to_openerp(record['group_id'], unwrap=True) if category_id is None: raise MappingError("The partner category with " "magento id %s does not exist" % record['group_id']) # FIXME: should remove the previous tag (all the other tags from # the same backend) return {'category_id': [(4, category_id)]}
[docs] @mapping def website_id(self, record): binder = self.binder_for(model='magento.website') website_id = binder.to_openerp(record['website_id']) return {'website_id': website_id}
[docs] @only_create @mapping def company_id(self, record): binder = self.binder_for(model='magento.storeview') storeview = binder.to_openerp(record['store_id'], browse=True) if storeview: company = storeview.backend_id.company_id if company: return {'company_id': company.id} return {'company_id': False}
[docs] @mapping def lang(self, record): binder = self.binder_for(model='magento.storeview') storeview = binder.to_openerp(record['store_id'], browse=True) if storeview: if storeview.lang_id: return {'lang': storeview.lang_id.code}
[docs] @only_create @mapping def customer(self, record): return {'customer': True}
[docs] @mapping def type(self, record): return {'type': 'default'}
[docs] @only_create @mapping def openerp_id(self, record): """ Will bind the customer on a existing partner with the same email """ partner = self.env['res.partner'].search( [('email', '=', record['email']), ('customer', '=', True), '|', ('is_company', '=', True), ('parent_id', '=', False)], limit=1, ) if partner: return {'openerp_id': partner.id}
[docs]@magento class PartnerImporter(MagentoImporter): _model_name = ['magento.res.partner'] _base_mapper = PartnerImportMapper def _import_dependencies(self): """ Import the dependencies for the record""" record = self.magento_record self._import_dependency(record['group_id'], 'magento.res.partner.category') def _after_import(self, partner_binding): """ Import the addresses """ book = self.unit_for(PartnerAddressBook, model='magento.address') book.import_addresses(self.magento_id, partner_binding.id)
PartnerImport = PartnerImporter # deprecated AddressInfos = namedtuple('AddressInfos', ['magento_record', 'partner_binding_id', 'merge'])
[docs]@magento class PartnerAddressBook(ConnectorUnit): """ Import all addresses from the address book of a customer. This class is responsible to define which addresses should be imported and how (merge with the partner or not...). Then, it delegate the import to the appropriate importer. This is really intricate. The datamodel are different between Magento and OpenERP and we have many uses cases to cover. The first thing is that: - we do not import companies and individuals the same manner - we do not know if an account is a company -> we assume that if we found something in the company field of the billing address, the whole account is a company. Differences: - Individuals: we merge the billing address with the partner, so we'll end with 1 entity if the customer has 1 address - Companies: we never merge the addresses with the partner, but we use the company name of the billing address as name of the partner. We also copy the address informations from the billing address as default values. More information on: https://bugs.launchpad.net/openerp-connector/+bug/1193281 """ _model_name = 'magento.address'
[docs] def import_addresses(self, magento_partner_id, partner_binding_id): addresses = self._get_address_infos(magento_partner_id, partner_binding_id) for address_id, infos in addresses: importer = self.unit_for(MagentoImporter) importer.run(address_id, address_infos=infos)
def _get_address_infos(self, magento_partner_id, partner_binding_id): adapter = self.unit_for(BackendAdapter) mag_address_ids = adapter.search({'customer_id': {'eq': magento_partner_id}}) if not mag_address_ids: return for address_id in mag_address_ids: magento_record = adapter.read(address_id) # defines if the billing address is merged with the partner # or imported as a standalone contact merge = False if magento_record.get('is_default_billing'): binding_model = self.env['magento.res.partner'] partner_binding = binding_model.browse(partner_binding_id) if magento_record.get('company'): # when a company is there, we never merge the contact # with the partner. # Copy the billing address on the company # and use the name of the company for the name company_mapper = self.unit_for(CompanyImportMapper, model='magento.res.partner') map_record = company_mapper.map_record(magento_record) parent = partner_binding.openerp_id.parent_id values = map_record.values(parent_partner=parent) partner_binding.write(values) else: # for B2C individual customers, merge with the main # partner merge = True # in the case if the billing address no longer # has a company, reset the flag partner_binding.write({'consider_as_company': False}) address_infos = AddressInfos(magento_record=magento_record, partner_binding_id=partner_binding_id, merge=merge) yield address_id, address_infos
[docs]class BaseAddressImportMapper(ImportMapper): """ Defines the base mappings for the imports in ``res.partner`` (state, country, ...) """ direct = [('postcode', 'zip'), ('city', 'city'), ('telephone', 'phone'), ('fax', 'fax'), ('company', 'company'), ]
[docs] @mapping def state(self, record): if not record.get('region'): return state = self.env['res.country.state'].search( [('name', '=ilike', record['region'])], limit=1, ) if state: return {'state_id': state.id}
[docs] @mapping def country(self, record): if not record.get('country_id'): return country = self.env['res.country'].search( [('code', '=', record['country_id'])], limit=1, ) if country: return {'country_id': country.id}
[docs] @mapping def street(self, record): value = record['street'] if not value: return {} lines = [line.strip() for line in value.split('\n') if line.strip()] if len(lines) == 1: result = {'street': lines[0], 'street2': False} elif len(lines) >= 2: result = {'street': lines[0], 'street2': u' - '.join(lines[1:])} else: result = {} return result
[docs] @mapping def title(self, record): prefix = record['prefix'] if not prefix: return title = self.env['res.partner.title'].search( [('domain', '=', 'contact'), ('shortcut', '=ilike', prefix)], limit=1 ) if not title: title = self.env['res.partner.title'].create( {'domain': 'contact', 'shortcut': prefix, 'name': prefix, } ) return {'title': title.id}
[docs] @only_create @mapping def company_id(self, record): parent = self.options.parent_partner if parent: if parent.company_id: return {'company_id': parent.company_id.id} else: return {'company_id': False} # Don't return anything, we are merging into an existing partner return
[docs]@magento class CompanyImportMapper(BaseAddressImportMapper): """ Special mapping used when we import a company. A company is considered as such when the billing address of an account has something in the 'company' field. This is a very special mapping not used in the same way than the other. The billing address will exist as a contact, but we want to *copy* the data on the company. The input record is the billing address. The mapper returns data which will be written on the main partner, in other words, the company. The ``@only_create`` decorator would not have any effect here because the mapper is always called for updates. """ _model_name = 'magento.res.partner' direct = BaseAddressImportMapper.direct + [ ('company', 'name'), ]
[docs] @mapping def consider_as_company(self, record): return {'consider_as_company': True}
[docs]@magento class AddressAdapter(GenericAdapter): _model_name = 'magento.address' _magento_model = 'customer_address'
[docs] def search(self, filters=None): """ Search records according to some criterias and returns a list of ids :rtype: list """ return [int(row['customer_address_id']) for row in self._call('%s.list' % self._magento_model, [filters] if filters else [{}])]
[docs] def create(self, customer_id, data): """ Create a record on the external system """ return self._call('%s.create' % self._magento_model, [customer_id, data])
[docs]@magento class AddressImporter(MagentoImporter): _model_name = ['magento.address']
[docs] def run(self, magento_id, address_infos=None, force=False): """ Run the synchronization """ if address_infos is None: # only possible for updates self.address_infos = AddressInfos(None, None, None) else: self.address_infos = address_infos return super(AddressImporter, self).run(magento_id, force=force)
def _get_magento_data(self): """ Return the raw Magento data for ``self.magento_id`` """ # we already read the data from the Partner Importer if self.address_infos.magento_record: return self.address_infos.magento_record else: return super(AddressImporter, self)._get_magento_data() def _define_partner_relationship(self, data): """ Link address with partner or parent company. """ partner_binding_id = self.address_infos.partner_binding_id assert partner_binding_id, ("AdressInfos are required for creation of " "a new address.") binder = self.binder_for('magento.res.partner') partner = binder.unwrap_binding(partner_binding_id, browse=True) if self.address_infos.merge: # it won't be imported as an independent address, # but will be linked with the main res.partner data['openerp_id'] = partner.id data['type'] = 'default' else: data['parent_id'] = partner.id data['lang'] = partner.lang data['magento_partner_id'] = self.address_infos.partner_binding_id return data def _create(self, data): data = self._define_partner_relationship(data) return super(AddressImporter, self)._create(data)
AddressImport = AddressImporter # deprecated
[docs]@magento class AddressImportMapper(BaseAddressImportMapper): _model_name = 'magento.address' # TODO fields not mapped: # "suffix"=>"a", # "vat_id"=>"12334", direct = BaseAddressImportMapper.direct + [ ('created_at', 'created_at'), ('updated_at', 'updated_at'), ('is_default_billing', 'is_default_billing'), ('is_default_shipping', 'is_default_shipping'), ('company', 'company'), ]
[docs] @mapping def names(self, record): # TODO create a glue module for base_surname parts = [part for part in (record['firstname'], record.get('middlename'), record['lastname']) if part] return {'name': ' '.join(parts)}
[docs] @mapping def use_parent_address(self, record): return {'use_parent_address': False}
[docs] @mapping def type(self, record): if record.get('is_default_billing'): address_type = 'invoice' elif record.get('is_default_shipping'): address_type = 'delivery' else: address_type = 'contact' return {'type': address_type}
[docs]@job(default_channel='root.magento') def partner_import_batch(session, model_name, backend_id, filters=None): """ Prepare the import of partners modified on Magento """ if filters is None: filters = {} assert 'magento_website_id' in filters, ( 'Missing information about Magento Website') env = get_environment(session, model_name, backend_id) importer = env.get_connector_unit(PartnerBatchImporter) importer.run(filters=filters)