111 lines
4.1 KiB
Python
111 lines
4.1 KiB
Python
# Copyright (c) 2012 Web Notes Technologies Pvt Ltd (http://erpnext.com)
|
|
#
|
|
# MIT License (MIT)
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the "Software"),
|
|
# to deal in the Software without restriction, including without limitation
|
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
|
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
|
|
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
|
# CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
|
|
# OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
#
|
|
|
|
from __future__ import unicode_literals
|
|
import webnotes, json
|
|
from webnotes import _
|
|
from webnotes.utils import cstr
|
|
from webnotes.model import default_fields
|
|
|
|
def get_mapped_doclist(from_doctype, from_docname, table_maps, target_doclist=[], postprocess=None):
|
|
if isinstance(target_doclist, basestring):
|
|
target_doclist = json.loads(target_doclist)
|
|
|
|
source = webnotes.bean(from_doctype, from_docname)
|
|
|
|
if not webnotes.has_permission(from_doctype, "read", source.doc):
|
|
webnotes.msgprint("No Permission", raise_exception=webnotes.PermissionError)
|
|
|
|
source_meta = webnotes.get_doctype(from_doctype)
|
|
target_meta = webnotes.get_doctype(table_maps[from_doctype]["doctype"])
|
|
|
|
# main
|
|
if target_doclist:
|
|
target_doc = webnotes.doc(target_doclist[0])
|
|
else:
|
|
target_doc = webnotes.new_doc(table_maps[from_doctype]["doctype"])
|
|
|
|
map_doc(source.doc, target_doc, table_maps[source.doc.doctype], source_meta, target_meta)
|
|
doclist = [target_doc]
|
|
|
|
# children
|
|
for source_d in source.doclist[1:]:
|
|
table_map = table_maps.get(source_d.doctype)
|
|
if table_map:
|
|
if "condition" in table_map:
|
|
if not table_map["condition"](source_d):
|
|
continue
|
|
target_doctype = table_map["doctype"]
|
|
parentfield = target_meta.get({
|
|
"parent": target_doc.doctype,
|
|
"doctype": "DocField",
|
|
"fieldtype": "Table",
|
|
"options": target_doctype
|
|
})[0].fieldname
|
|
|
|
target_d = webnotes.new_doc(target_doctype, target_doc, parentfield)
|
|
map_doc(source_d, target_d, table_map, source_meta, target_meta, source.doclist[0])
|
|
doclist.append(target_d)
|
|
|
|
doclist = webnotes.doclist(doclist)
|
|
|
|
if postprocess:
|
|
postprocess(source, doclist)
|
|
|
|
return doclist
|
|
|
|
def map_doc(source_doc, target_doc, table_map, source_meta, target_meta, source_parent=None):
|
|
no_copy_fields = set(\
|
|
[d.fieldname for d in source_meta.get({"no_copy": 1,
|
|
"parent": source_doc.doctype})] \
|
|
+ [d.fieldname for d in target_meta.get({"no_copy": 1,
|
|
"parent": target_doc.doctype})] \
|
|
+ default_fields)
|
|
|
|
if table_map.get("validation"):
|
|
for key, condition in table_map["validation"].items():
|
|
if condition[0]=="=":
|
|
if source_doc.fields.get(key) != condition[1]:
|
|
webnotes.msgprint(_("Cannot map because following condition fails: ")
|
|
+ key + "=" + cstr(condition[1]), raise_exception=webnotes.ValidationError)
|
|
|
|
# map same fields
|
|
target_fields = target_meta.get({"doctype": "DocField", "parent": target_doc.doctype})
|
|
for key in [d.fieldname for d in target_fields]:
|
|
if key not in no_copy_fields:
|
|
val = source_doc.fields.get(key)
|
|
if val not in (None, ""):
|
|
target_doc.fields[key] = val
|
|
|
|
|
|
# map other fields
|
|
for source_key, target_key in table_map.get("field_map", {}).items():
|
|
val = source_doc.fields.get(source_key)
|
|
if val not in (None, ""):
|
|
target_doc.fields[target_key] = val
|
|
|
|
# map idx
|
|
if source_doc.idx:
|
|
target_doc.idx = source_doc.idx
|
|
|
|
if "postprocess" in table_map:
|
|
table_map["postprocess"](source_doc, target_doc, source_parent)
|