""" Export files to modules """ from webnotes.modules import scrub, get_module_path def export_to_files(record_list=[], record_module=None, verbose=0): """ Export record_list to files. record_list is a list of lists ([doctype],[docname] ) , """ import webnotes.model.doc module_doclist =[] if record_list: for record in record_list: doclist = [d.fields for d in webnotes.model.doc.get(record[0], record[1])] write_document_file(doclist, record_module) def create_init_py(modules_path, dt, dn): """ Creates __init__.py in the module directory structure """ import os def create_if_not_exists(path): initpy = os.path.join(path, '__init__.py') if not os.path.exists(initpy): open(initpy, 'w').close() create_if_not_exists(os.path.join(modules_path)) create_if_not_exists(os.path.join(modules_path, dt)) create_if_not_exists(os.path.join(modules_path, dt, dn)) def create_folder(module, dt, dn): """ Creates directories for module and their __init__.py """ import webnotes, os # get module path by importing the module modules_path = get_module_path(module) code_type = dt in ['DocType', 'Page', 'Search Criteria'] # create folder folder = os.path.join(modules_path, code_type and scrub(dt) or dt, code_type and scrub(dn) or dn) webnotes.create_folder(folder) # create init_py_files if code_type: create_init_py(modules_path, scrub(dt), scrub(dn)) return folder def get_module_name(doclist, record_module=None): """ Returns the module-name of a doclist """ # module name if doclist[0]['doctype'] == 'Module Def': module = doclist[0]['name'] elif doclist[0]['doctype']=='Control Panel': module = 'Core' elif record_module: module = record_module else: module = doclist[0]['module'] return module def write_document_file(doclist, record_module=None): """ Write a doclist to file, can optionally specify module name """ import os from webnotes.model.utils import pprint_doclist module = get_module_name(doclist, record_module) # create the folder code_type = doclist[0]['doctype'] in ['DocType','Page','Search Criteria'] # create folder folder = create_folder(module, doclist[0]['doctype'], doclist[0]['name']) # separate code files clear_code_fields(doclist, folder, code_type) # write the data file fname = (code_type and scrub(doclist[0]['name'])) or doclist[0]['name'] txtfile = open(os.path.join(folder, fname +'.txt'),'w+') txtfile.write(pprint_doclist(doclist)) #dict_list = [pprint_dict(d) for d in doclist] #txtfile.write('[\n' + ',\n'.join(dict_list) + '\n]') txtfile.close() def clear_code_fields(doclist, folder, code_type): """ Removes code from the doc """ import os import webnotes # code will be in the parent only code_fields = webnotes.code_fields_dict.get(doclist[0]['doctype'], []) for code_field in code_fields: if doclist[0].get(code_field[0]): doclist[0][code_field[0]] = None