[IMP] develop a edi purchase order import and export

bzr revid: aar@tinyerp.com-20110829104421-6562nvcyulzl4o84
This commit is contained in:
Amit Parmar (OpenERP)
2011-08-29 16:14:21 +05:30
parent 7d33fcfa66
commit d6b7e8abb7
3 changed files with 272 additions and 0 deletions
+190
View File
@@ -0,0 +1,190 @@
# -*- coding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
#
# 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/>.
#
##############################################################################
from osv import fields, osv, orm
from base.ir import ir_edi
from tools.translate import _
from datetime import date
class purchase_order(osv.osv, ir_edi.edi):
_inherit = 'purchase.order'
def edi_export(self, cr, uid, records, edi_struct=None, context=None):
"""Exports a supplier or customer invoice"""
edi_struct = {
'name': True,
'origin': True,
'date_order': True,
'date_approve': True,
'partner_id': True,
'partner_address_id': True,
'dest_address_id': True,
'warehouse_id': True,
'location_id': True,
'pricelist_id': True,
'validator' : True,
'order_line': {
'name': True,
'product_qty': True,
'date_planned': True,
'taxes_id': True,
'product_uom': True,
'product_id': True,
'move_dest_id': True,
'price_unit': True,
'order_id': True,
'invoiced': True,
'price_subtotal': True,
},
'invoice_ids': True,
'shipped': True,
'company_id': True,
}
partner_pool = self.pool.get('res.partner')
partner_address_pool = self.pool.get('res.partner.address')
company_address_dict = {
'street': True,
'street2': True,
'zip': True,
'city': True,
'state_id': True,
'country_id': True,
'email': True,
'phone': True,
}
edi_doc_list = []
for order in records:
# Get EDI doc based on struct. The result will also contain all metadata fields and attachments.
edi_doc = super(purchase_order,self).edi_export(cr, uid, [order], edi_struct, context)
if not edi_doc:
continue
edi_doc = edi_doc[0]
# Add company info and address
res = partner_pool.address_get(cr, uid, [order.company_id.partner_id.id], ['contact', 'order'])
contact_addr_id = res['contact']
invoice_addr_id = res['order']
address = partner_address_pool.browse(cr, uid, invoice_addr_id, context=context)
edi_company_address_dict = {}
for key, value in company_address_dict.items():
if not value:
continue
address_rec = getattr(address, key)
if not address_rec:
continue
if key.endswith('_id'):
address_rec = self.edi_m2o(cr, uid, address_rec, context=context)
edi_company_address_dict[key] = address_rec
edi_doc.update({
'company_address': edi_company_address_dict,
#'company_logo': inv_comp.logo,#TODO
#'paid': inv_comp.paid, #TODO
})
edi_doc_list.append(edi_doc)
return edi_doc_list
def edi_import(self, cr, uid, edi_document, context=None):
partner_pool = self.pool.get('res.partner')
partner_address_pool = self.pool.get('res.partner.address')
model_data_pool = self.pool.get('ir.model.data')
product_pool = self.pool.get('product.product')
product_categ_pool = self.pool.get('product.category')
company_pool = self.pool.get('res.company')
country_pool = self.pool.get('res.country')
state_pool = self.pool.get('res.country.state')
account_journal_pool = self.pool.get('account.journal')
invoice_line_pool = self.pool.get('account.invoice.line')
account_pool = self.pool.get('account.account')
stock = self.pool.get('stock.location')
tax_id = []
account_id = []
partner_id = None
company_id = None
if context is None:
context = {}
# import company as a new partner, if type==in then supplier=1, else customer=1
# partner_id field is modified to point to the new partner
# company_address data used to add address to new partner
edi_company_address = edi_document['company_address']
edi_partner_id = edi_document['partner_id']
company_name = edi_document['shop_id'][1]
state_id = edi_company_address.get('state_id', False)
state_name = state_id and state_id[1]
country_id = edi_company_address.get('country_id', False)
country_name = country_id and country_id[1]
country_id = country_name and self.edi_import_relation(cr, uid, 'res.country', country_name, context=context) or False
state_id = state_name and self.edi_import_relation(cr, uid, 'res.country.state', state_name,
values={'country_id': country_id, 'code': state_name}, context=context) or False
address_value = {
'street': edi_company_address.get('street', False),
'street2': edi_company_address.get('street2', False),
'zip': edi_company_address.get('zip', False),
'city': edi_company_address.get('city', False),
'state_id': state_id,
'country_id': country_id,
'email': edi_company_address.get('email', False),
'phone': edi_company_address.get('phone', False),
}
partner_value = {'name': company_name}
partner_value.update({'customer': False, 'supplier': True})
partner_id = partner_pool.create(cr, uid, partner_value, context=context)
address_value.update({'partner_id': partner_id})
address_id = partner_address_pool.create(cr, uid, address_value, context=context)
partner_address = partner_address_pool.browse(cr, uid, address_id, context=context)
edi_document.update({'partner_address_id':self.edi_m2o(cr, uid, partner_address, context=context)})
partner = partner_pool.browse(cr, uid, partner_id, context=context)
edi_document['partner_id'] = self.edi_m2o(cr, uid, partner, context=context)
location_id = stock.search(cr, uid,[('name','=','Stock')])
location = stock.browse(cr, uid, location_id[0])
edi_document.update({'dest_address_id': edi_document['partner_shipping_id'],'location_id': self.edi_m2o(cr, uid, location, context=context)})
for line in range(len(edi_document['order_line'])):
product_qty = edi_document['order_line'][line]['product_uom_qty']
edi_document['order_line'][line].update({'product_qty': product_qty})
# all fields are converted for purchase order import so unnecessary fields are deleted
delete_key = ['sequence','procurement_id','product_uom_qty','company_address','shop_id','create_date','picking_policy','order_policy','partner_order_id','partner_shipping_id','invoice_quantity','partner_invoice_id','price_subtotal','date_confirm']
for key in delete_key:
if edi_document.has_key(key):
del edi_document[key]
else:
for document in edi_document['order_line']:
if document.has_key(key):
del document[key]
print edi_document
return super(purchase_order,self).edi_import(cr, uid, edi_document, context=context)
purchase_order()
class purchase_order_line(osv.osv, ir_edi.edi):
_inherit='purchase.order.line'
purchase_order_line()
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
@@ -0,0 +1,51 @@
<?xml version="1.0" ?>
<openerp>
<data>
<!--Export edi document -->
<record id="ir_actions_server_edi_purchase" model="ir.actions.server">
<field name="code">context.update({'edi_web_url_view': '%s/edi/view_edi?db=%s&amp;token=%s' %(self.pool.get('ir.config_parameter').get_param(cr, uid, 'web.base.url'),cr.dbname, self.pool.get('ir.edi.document').export_edi(cr, uid, [object], context = context)[0])})
self.pool.get('email.template').generate_mail(cr,
uid,
self.pool.get('ir.model.data').get_object_reference(cr, uid, 'purchase', 'email_template_edi_purchase')[1],
[object.id],
context=context)</field>
<field eval="8" name="sequence"/>
<field name="state">code</field>
<field name="type">ir.actions.server</field>
<field name="model_id" ref="purchase.model_purchase_order"/>
<field name="condition">True</field>
<field name="name">EDI Document - Purchase Order</field>
</record>
<!--calling multi action in workflow.activity -->
<record id="purchase.act_confirmed" model="workflow.activity">
<field name="action_id" ref="ir_actions_server_edi_purchase"/>
</record>
<!--Email template -->
<record id="email_template_edi_purchase" model="email.template">
<field name="def_subject">${object.company_id.name} - Purchase Order ${object.name}</field>
<field name="def_to">${object.partner_address_id.email}</field>
<field name="object_name" ref="purchase.model_purchase_order"/>
<field name="use_sign">True</field>
<field name="def_body_text">
Hello ${object.partner_address_id.name},
We just registered the purchase order ${object.name} for ${object.amount_total} for your company ${object.partner_id.name}.
You can click on the following link to preview.
${object._context.get('edi_web_url_view')}
Confirmed on ${object.date_approve}
Regards,
</field>
<field name="template_language">mako</field>
<field name="name">Mail Template of Purchase Order For EDI Document</field>
<field name="model_int_name">purchase.order</field>
<!--field name="reply_to">${object.user_id.email or ''}</field-->
</record>
</data>
</openerp>
@@ -0,0 +1,31 @@
-
I Tesing of EDI functionality. First I export customer invoice from my company than import that invoice into customer company
-
!python {model: ir.edi.document}: |
import json
order_pool = self.pool.get('purchase.order')
purchase_id = order_pool.search(cr, uid, [])
orders = order_pool.browse(cr, uid, purchase_id)
if type(orders).__name__ == 'list':
tokens = self.export_edi(cr, uid, orders)
for token in tokens:
document = self.get_document(cr, uid, token)
document = json.loads(document)
edi_doc = []
for doc in document:
doc.update({'__model': 'sale.order'})
edi_doc.append(doc)
a = self.import_edi(cr, uid, edi_document = json.dumps(edi_doc))
else:
tokens = self.export_edi(cr, uid, orders)
for token in tokens:
document = self.get_document(cr, uid, token)
document = json.loads(document)
edi_doc = []
for doc in document:
doc.update({'__model': 'sale.order'})
edi_doc.append(doc)
a = self.import_edi(cr, uid, edi_document = json.dumps(edi_doc))