#!/usr/bin/env python # ERPNext - web based ERP (http://erpnext.com) # Copyright (C) 2012 Web Notes Technologies Pvt Ltd # # 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 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . import os, sys def replace_code(start, txt1, txt2, extn): """replace all txt1 by txt2 in files with extension (extn)""" import os, re for wt in os.walk(start, followlinks=1): for fn in wt[2]: if fn.split('.')[-1]==extn: fpath = os.path.join(wt[0], fn) with open(fpath, 'r') as f: content = f.read() if re.search(txt1, content): a = raw_input('Change in %s [y/n]?' % fpath) if a=='y': with open(fpath, 'w') as f: f.write(re.sub(txt1, txt2, content)) print 'updated in %s' % fpath def setup_options(): from optparse import OptionParser parser = OptionParser() parser.add_option("-d", "--db", dest="db_name", help="Apply the patches on given db") # build parser.add_option("-b", "--build", default=False, action="store_true", help="minify + concat js files") parser.add_option("-c", "--clear", default=False, action="store_true", help="increment version") # git parser.add_option("--status", default=False, action="store_true", help="git status") parser.add_option("--pull", nargs=2, default=False, metavar = "remote branch", help="git pull (both repos)") parser.add_option("--push", nargs=3, default=False, metavar = "remote branch comment", help="git commit + push (both repos) [remote] [branch] [comment]") parser.add_option("-l", "--latest", action="store_true", dest="run_latest", default=False, help="Apply the latest patches") # patch parser.add_option("-p", "--patch", nargs=1, dest="patch_list", metavar='patch_module', action="append", help="Apply patch") parser.add_option("-f", "--force", action="store_true", dest="force", default=False, help="Force Apply all patches specified using option -p or --patch") parser.add_option('--reload_doc', nargs=3, metavar = "module doctype docname", help="reload doc") parser.add_option('--export_doc', nargs=2, metavar = "doctype docname", help="export doc") # install parser.add_option('--install', nargs=3, metavar = "rootpassword dbname source", help="install fresh db") parser.add_option('--sync_with_gateway', nargs=1, metavar = "1/0", \ help="Set or Unset Sync with Gateway") # diff parser.add_option('--diff_ref_file', nargs=0, \ help="Get missing database records and mismatch properties, with file as reference") parser.add_option('--diff_ref_db', nargs=0, \ help="Get missing .txt files and mismatch properties, with database as reference") # scheduler parser.add_option('--run_scheduler', default=False, action="store_true", help="Trigger scheduler") parser.add_option('--run_scheduler_event', nargs=1, metavar="[all|daily|weekly|monthly]", help="Run scheduler event") # misc parser.add_option("--replace", nargs=3, default=False, metavar = "search replace_by extension", help="file search-replace") return parser.parse_args() def run(): sys.path.append('lib') sys.path.append('lib/py') import webnotes import webnotes.defs sys.path.append(webnotes.defs.modules_path) (options, args) = setup_options() from webnotes.db import Database import webnotes.modules.patch_handler # connect if options.db_name is not None: webnotes.connect(options.db_name) # build if options.build: import build.project build.project.build() elif options.clear: from build.project import increment_version print "Version:" + str(increment_version()) # code replace elif options.replace: replace_code('.', options.replace[0], options.replace[1], options.replace[2]) # git elif options.status: os.system('git status') os.chdir('lib') os.system('git status') elif options.pull: os.system('git pull %s %s' % (options.pull[0], options.pull[1])) os.chdir('lib') os.system('git pull %s %s' % (options.pull[0], options.pull[1])) elif options.push: os.system('git commit -a -m "%s"' % options.push[2]) os.system('git push %s %s' % (options.push[0], options.push[1])) os.chdir('lib') os.system('git commit -a -m "%s"' % options.push[2]) os.system('git push %s %s' % (options.push[0], options.push[1])) # patch elif options.patch_list: # clear log webnotes.modules.patch_handler.log_list = [] # run individual patches for patch in options.patch_list: webnotes.modules.patch_handler.run_single(\ patchmodule = patch, force = options.force) print '\n'.join(webnotes.modules.patch_handler.log_list) # reload elif options.reload_doc: webnotes.modules.patch_handler.reload_doc(\ {"module":options.reload_doc[0], "dt":options.reload_doc[1], "dn":options.reload_doc[2]}) print '\n'.join(webnotes.modules.patch_handler.log_list) elif options.export_doc: from webnotes.modules import export_doc export_doc(options.export_doc[0], options.export_doc[1]) # run all pending elif options.run_latest: webnotes.modules.patch_handler.run_all() print '\n'.join(webnotes.modules.patch_handler.log_list) elif options.install: from webnotes.install_lib.install import Installer inst = Installer('root', options.install[0]) inst.import_from_db(options.install[1], source_path=options.install[2], \ password='admin', verbose = 1) elif options.sync_with_gateway: if int(options.sync_with_gateway[0]) in [0, 1]: webnotes.conn.begin() webnotes.conn.sql("""\ UPDATE `tabSingles` SET value=%s WHERE field='sync_with_gateway' AND doctype='Control Panel'""", int(options.sync_with_gateway[0])) webnotes.conn.commit() webnotes.message_log.append("sync_with_gateway set to %s" % options.sync_with_gateway[0]) else: webnotes.message_log.append("ERROR: sync_with_gateway can be either 0 or 1") elif options.diff_ref_file is not None: import webnotes.modules.diff webnotes.modules.diff.diff_ref_file() elif options.diff_ref_db is not None: import webnotes.modules.diff webnotes.modules.diff.diff_ref_db() elif options.run_scheduler: import webnotes.utils.scheduler print webnotes.utils.scheduler.execute() elif options.run_scheduler_event is not None: import webnotes.utils.scheduler print webnotes.utils.scheduler.trigger('execute_' + options.run_scheduler_event) # print messages if webnotes.message_log: print '\n'.join(webnotes.message_log) if __name__=='__main__': run()