512 lines
16 KiB
Python
Executable file
512 lines
16 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd.
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
import sys
|
|
|
|
if __name__=="__main__":
|
|
sys.path = [".", "lib", "app"] + sys.path
|
|
|
|
import webnotes
|
|
|
|
def main():
|
|
parsed_args = webnotes._dict(vars(setup_parser()))
|
|
fn = get_function(parsed_args)
|
|
if parsed_args.get("site")=="all":
|
|
for site in get_sites():
|
|
args = parsed_args.copy()
|
|
args["site"] = site
|
|
run(fn, args)
|
|
else:
|
|
run(fn, parsed_args)
|
|
|
|
def cmd(fn):
|
|
def new_fn(*args, **kwargs):
|
|
import inspect
|
|
fnargs, varargs, varkw, defaults = inspect.getargspec(fn)
|
|
new_kwargs = {}
|
|
for a in fnargs:
|
|
if a in kwargs:
|
|
new_kwargs[a] = kwargs.get(a)
|
|
|
|
return fn(*args, **new_kwargs)
|
|
|
|
return new_fn
|
|
|
|
|
|
def run(fn, args):
|
|
if isinstance(args.get(fn), (list, tuple)):
|
|
out = globals().get(fn)(*args.get(fn), **args)
|
|
else:
|
|
out = globals().get(fn)(**args)
|
|
webnotes.destroy()
|
|
return out
|
|
|
|
def get_function(args):
|
|
for fn, val in args.items():
|
|
if (val or isinstance(val, list)) and globals().get(fn):
|
|
return fn
|
|
|
|
def get_sites():
|
|
import os
|
|
import conf
|
|
return [site for site in os.listdir(conf.sites_dir)
|
|
if not os.path.islink(os.path.join(conf.sites_dir, site))]
|
|
|
|
def setup_parser():
|
|
import argparse
|
|
parser = argparse.ArgumentParser(description="Run webnotes utility functions")
|
|
|
|
setup_install(parser)
|
|
setup_utilities(parser)
|
|
setup_translation(parser)
|
|
setup_git(parser)
|
|
|
|
# common
|
|
parser.add_argument("-f", "--force", default=False, action="store_true",
|
|
help="Force execution where applicable (look for [-f] in help)")
|
|
parser.add_argument("--quiet", default=True, action="store_false", dest="verbose",
|
|
help="Show verbose output where applicable")
|
|
parser.add_argument("--site", nargs="?", metavar="SITE-NAME or all",
|
|
help="Run for a particular site")
|
|
|
|
return parser.parse_args()
|
|
|
|
def setup_install(parser):
|
|
parser.add_argument("--install", metavar="DB-NAME", nargs=1,
|
|
help="Install a new app")
|
|
parser.add_argument("--root-password", nargs=1,
|
|
help="Root password for new app")
|
|
parser.add_argument("--reinstall", default=False, action="store_true",
|
|
help="Install a fresh app in db_name specified in conf.py")
|
|
parser.add_argument("--restore", metavar=("DB-NAME", "SQL-FILE"), nargs=2,
|
|
help="Restore from an sql file")
|
|
parser.add_argument("--install_fixtures", default=False, action="store_true",
|
|
help="(Re)Install install-fixtures from app/startup/install_fixtures")
|
|
parser.add_argument("--make_demo", default=False, action="store_true",
|
|
help="Install demo in demo_db_name specified in conf.py")
|
|
parser.add_argument("--make_demo_fresh", default=False, action="store_true",
|
|
help="(Re)Install demo in demo_db_name specified in conf.py")
|
|
|
|
def setup_utilities(parser):
|
|
# update
|
|
parser.add_argument("-u", "--update", nargs="*", metavar=("REMOTE", "BRANCH"),
|
|
help="Perform git pull, run patches, sync schema and rebuild files/translations")
|
|
parser.add_argument("--patch", nargs=1, metavar="PATCH-MODULE",
|
|
help="Run a particular patch [-f]")
|
|
parser.add_argument("-l", "--latest", default=False, action="store_true",
|
|
help="Run patches, sync schema and rebuild files/translations")
|
|
parser.add_argument("--sync_all", default=False, action="store_true",
|
|
help="Reload all doctypes, pages, etc. using txt files [-f]")
|
|
parser.add_argument("--update_all_sites", nargs="*", metavar=("REMOTE", "BRANCH"),
|
|
help="Perform git pull, run patches, sync schema and rebuild files/translations")
|
|
|
|
parser.add_argument("--reload_doc", nargs=3,
|
|
metavar=('"MODULE"', '"DOCTYPE"', '"DOCNAME"'))
|
|
|
|
# build
|
|
parser.add_argument("-b", "--build", default=False, action="store_true",
|
|
help="Minify + concatenate JS and CSS files, build translations")
|
|
parser.add_argument("-w", "--watch", default=False, action="store_true",
|
|
help="Watch and concatenate JS and CSS files as and when they change")
|
|
|
|
# misc
|
|
parser.add_argument("--backup", default=False, action="store_true",
|
|
help="Take backup of database in backup folder [--with_files]")
|
|
parser.add_argument("--with_files", default=False, action="store_true",
|
|
help="Also take backup of files")
|
|
parser.add_argument("--docs", default=False, action="store_true",
|
|
help="Build docs")
|
|
parser.add_argument("--domain", nargs="*",
|
|
help="Get or set domain in Website Settings")
|
|
parser.add_argument("--make_conf", nargs="*", metavar=("DB-NAME", "DB-PASSWORD"),
|
|
help="Create new conf.py file")
|
|
parser.add_argument("--set_admin_password", metavar='ADMIN-PASSWORD', nargs=1,
|
|
help="Set administrator password")
|
|
|
|
# clear
|
|
parser.add_argument("--clear_web", default=False, action="store_true",
|
|
help="Clear website cache")
|
|
parser.add_argument("--clear_cache", default=False, action="store_true",
|
|
help="Clear cache, doctype cache and defaults")
|
|
parser.add_argument("--reset_perms", default=False, action="store_true",
|
|
help="Reset permissions for all doctypes")
|
|
|
|
# scheduler
|
|
parser.add_argument("--run_scheduler", default=False, action="store_true",
|
|
help="Trigger scheduler")
|
|
parser.add_argument("--run_scheduler_event", nargs=1,
|
|
metavar="all | daily | weekly | monthly",
|
|
help="Run a scheduler event")
|
|
|
|
# replace
|
|
parser.add_argument("--replace", nargs=3,
|
|
metavar=("SEARCH-REGEX", "REPLACE-BY", "FILE-EXTN"),
|
|
help="Multi-file search-replace [-f]")
|
|
|
|
# import/export
|
|
parser.add_argument("--export_doc", nargs=2, metavar=('"DOCTYPE"', '"DOCNAME"'))
|
|
parser.add_argument("--export_doclist", nargs=3, metavar=("DOCTYPE", "NAME", "PATH"),
|
|
help="""Export doclist as json to the given path, use '-' as name for Singles.""")
|
|
parser.add_argument("--export_csv", nargs=2, metavar=("DOCTYPE", "PATH"),
|
|
help="""Dump DocType as csv""")
|
|
parser.add_argument("--import_doclist", nargs=1, metavar="PATH",
|
|
help="""Import (insert/update) doclist. If the argument is a directory, all files ending with .json are imported""")
|
|
|
|
def setup_git(parser):
|
|
parser.add_argument("--pull", nargs="*", metavar=("REMOTE", "BRANCH"),
|
|
help="Run git pull for both repositories")
|
|
parser.add_argument("-p", "--push", nargs="*", metavar=("REMOTE", "BRANCH"),
|
|
help="Run git push for both repositories")
|
|
parser.add_argument("--status", default=False, action="store_true",
|
|
help="Run git status for both repositories")
|
|
parser.add_argument("--commit", nargs=1, metavar="COMMIT-MSG",
|
|
help="Run git commit COMMIT-MSG for both repositories")
|
|
parser.add_argument("--checkout", nargs=1, metavar="BRANCH",
|
|
help="Run git checkout BRANCH for both repositories")
|
|
parser.add_argument("--git", nargs="*", metavar="OPTIONS",
|
|
help="Run git command for both repositories")
|
|
|
|
|
|
def setup_translation(parser):
|
|
parser.add_argument("--build_message_files", default=False, action="store_true",
|
|
help="Build message files for translation")
|
|
parser.add_argument("--export_messages", nargs=2, metavar=("LANG-CODE", "FILENAME"),
|
|
help="""Export all messages for a language to translation in a csv file.
|
|
Example, lib/wnf.py --export_messages hi hindi.csv""")
|
|
parser.add_argument("--import_messages", nargs=2, metavar=("LANG-CODE", "FILENAME"),
|
|
help="""Import messages for a language and make language files.
|
|
Example, lib/wnf.py --import_messages hi hindi.csv""")
|
|
parser.add_argument("--google_translate", nargs=3,
|
|
metavar=("LANG-CODE", "INFILE", "OUTFILE"),
|
|
help="Auto translate using Google Translate API")
|
|
parser.add_argument("--translate", nargs=1, metavar="LANG-CODE",
|
|
help="""Rebuild translation for the given langauge and
|
|
use Google Translate to tranlate untranslated messages. use "all" """)
|
|
|
|
# methods
|
|
|
|
# install
|
|
@cmd
|
|
def install(db_name, source_sql=None, site=None, verbose=True, force=False, root_password=None):
|
|
from webnotes.install_lib.install import Installer
|
|
inst = Installer('root', db_name=db_name, site=site, root_password=root_password)
|
|
inst.install(db_name, source_sql=source_sql, verbose=verbose, force=force)
|
|
|
|
@cmd
|
|
def reinstall(site=None, verbose=True):
|
|
webnotes.init(site=site)
|
|
install(webnotes.conf.db_name, site=site, verbose=verbose, force=True)
|
|
|
|
@cmd
|
|
def restore(db_name, source_sql, site=None, verbose=True, force=False):
|
|
install(db_name, source_sql, site=site, verbose=verbose, force=force)
|
|
|
|
@cmd
|
|
def install_fixtures(site=None):
|
|
webnotes.init(site=site)
|
|
from webnotes.install_lib.install import install_fixtures
|
|
install_fixtures()
|
|
|
|
@cmd
|
|
def make_demo(site=None):
|
|
import utilities.demo.make_demo
|
|
webnotes.init(site=site)
|
|
utilities.demo.make_demo.make()
|
|
|
|
@cmd
|
|
def make_demo_fresh(site=None):
|
|
import utilities.demo.make_demo
|
|
webnotes.init(site=site)
|
|
utilities.demo.make_demo.make(reset=True)
|
|
|
|
# utilities
|
|
|
|
@cmd
|
|
def update(remote=None, branch=None, site=None):
|
|
pull(remote=remote, branch=branch, site=site)
|
|
|
|
# maybe there are new framework changes, any consequences?
|
|
reload(webnotes)
|
|
|
|
latest(site=site)
|
|
|
|
@cmd
|
|
def latest(site=None, verbose=True):
|
|
import webnotes.modules.patch_handler
|
|
import webnotes.model.sync
|
|
|
|
webnotes.connect(site=site)
|
|
|
|
try:
|
|
# run patches
|
|
webnotes.modules.patch_handler.log_list = []
|
|
webnotes.modules.patch_handler.run_all()
|
|
if verbose:
|
|
print "\n".join(webnotes.modules.patch_handler.log_list)
|
|
|
|
# sync
|
|
webnotes.model.sync.sync_all()
|
|
except webnotes.modules.patch_handler.PatchError, e:
|
|
print "\n".join(webnotes.modules.patch_handler.log_list)
|
|
raise e
|
|
|
|
@cmd
|
|
def sync_all(site=None, force=False):
|
|
import webnotes.model.sync
|
|
webnotes.connect(site=site)
|
|
webnotes.model.sync.sync_all(force=force)
|
|
|
|
@cmd
|
|
def patch(patch_module, site=None, force=False):
|
|
import webnotes.modules.patch_handler
|
|
webnotes.connect(site=site)
|
|
webnotes.modules.patch_handler.log_list = []
|
|
webnotes.modules.patch_handler.run_single(patch_module, force=force)
|
|
print "\n".join(webnotes.modules.patch_handler.log_list)
|
|
|
|
@cmd
|
|
def update_all_sites(remote=None, branch=None, verbose=True):
|
|
pull(remote, branch)
|
|
build()
|
|
for site in get_sites():
|
|
latest(site=site, verbose=verbose)
|
|
|
|
@cmd
|
|
def reload_doc(module, doctype, docname, site=None, force=False):
|
|
webnotes.connect(site=site)
|
|
webnotes.reload_doc(module, doctype, docname, force=force)
|
|
|
|
@cmd
|
|
def build():
|
|
import webnotes.build
|
|
webnotes.build.bundle(False)
|
|
|
|
@cmd
|
|
def watch():
|
|
import webnotes.build
|
|
webnotes.build.watch(True)
|
|
|
|
@cmd
|
|
def backup(site=None, with_files=False):
|
|
from webnotes.utils.backups import scheduled_backup
|
|
webnotes.connect(site=site)
|
|
scheduled_backup(ignore_files=not with_files)
|
|
|
|
@cmd
|
|
def docs():
|
|
from core.doctype.documentation_tool.documentation_tool import write_static
|
|
write_static()
|
|
|
|
@cmd
|
|
def domain(host_url=None, site=None):
|
|
webnotes.connect(site=site)
|
|
if host_url:
|
|
webnotes.conn.set_value("Website Settings", None, "subdomain", host_url)
|
|
webnotes.conn.commit()
|
|
else:
|
|
print webnotes.conn.get_value("Website Settings", None, "subdomain")
|
|
|
|
@cmd
|
|
def make_conf(db_name=None, db_password=None, site=None, site_config=None):
|
|
from webnotes.install_lib.install import make_conf
|
|
make_conf(db_name=db_name, db_password=db_password, site=site, site_config=site_config)
|
|
|
|
# clear
|
|
@cmd
|
|
def clear_cache(site=None):
|
|
import webnotes.sessions
|
|
webnotes.connect(site=site)
|
|
webnotes.sessions.clear_cache()
|
|
|
|
@cmd
|
|
def clear_web(site=None):
|
|
import webnotes.webutils
|
|
webnotes.connect(site=site)
|
|
webnotes.webutils.clear_cache()
|
|
|
|
@cmd
|
|
def reset_perms(site=None):
|
|
webnotes.connect(site=site)
|
|
for d in webnotes.conn.sql_list("""select name from `tabDocType`
|
|
where ifnull(istable, 0)=0 and ifnull(custom, 0)=0"""):
|
|
webnotes.clear_cache(doctype=d)
|
|
webnotes.reset_perms(d)
|
|
|
|
# scheduler
|
|
@cmd
|
|
def run_scheduler(site=None):
|
|
import webnotes.utils.scheduler
|
|
webnotes.connect(site=site)
|
|
print webnotes.utils.scheduler.execute()
|
|
|
|
@cmd
|
|
def run_scheduler_event(event, site=None):
|
|
import webnotes.utils.scheduler
|
|
webnotes.connect(site=site)
|
|
print webnotes.utils.scheduler.trigger("execute_" + event)
|
|
|
|
# replace
|
|
@cmd
|
|
def replace(search_regex, replacement, extn, force=False):
|
|
print search_regex, replacement, extn
|
|
replace_code('.', search_regex, replacement, extn, force=force)
|
|
|
|
# import/export
|
|
@cmd
|
|
def export_doc(doctype, docname, site=None):
|
|
import webnotes.modules
|
|
webnotes.connect(site=site)
|
|
webnotes.modules.export_doc(doctype, docname)
|
|
|
|
@cmd
|
|
def export_doclist(doctype, name, path, site=None):
|
|
from core.page.data_import_tool import data_import_tool
|
|
webnotes.connect(site=site)
|
|
data_import_tool.export_json(doctype, name, path)
|
|
|
|
@cmd
|
|
def export_csv(doctype, path, site=None):
|
|
from core.page.data_import_tool import data_import_tool
|
|
webnotes.connect(site=site)
|
|
data_import_tool.export_csv(doctype, path)
|
|
|
|
@cmd
|
|
def import_doclist(path, site=None):
|
|
from core.page.data_import_tool import data_import_tool
|
|
webnotes.connect(site=site)
|
|
data_import_tool.import_doclist(path)
|
|
|
|
# translation
|
|
@cmd
|
|
def build_message_files(site=None):
|
|
import webnotes.translate
|
|
webnotes.connect(site=site)
|
|
webnotes.translate.build_message_files()
|
|
|
|
@cmd
|
|
def export_messages(lang, outfile, site=None):
|
|
import webnotes.translate
|
|
webnotes.connect(site=site)
|
|
webnotes.translate.export_messages(lang, outfile)
|
|
|
|
@cmd
|
|
def import_messages(lang, infile, site=None):
|
|
import webnotes.translate
|
|
webnotes.connect(site=site)
|
|
webnotes.translate.import_messages(lang, infile)
|
|
|
|
@cmd
|
|
def google_translate(lang, infile, outfile, site=None):
|
|
import webnotes.translate
|
|
webnotes.connect(site=site)
|
|
webnotes.translate.google_translate(lang, infile, outfile)
|
|
|
|
@cmd
|
|
def translate(lang, site=None):
|
|
import webnotes.translate
|
|
webnotes.connect(site=site)
|
|
webnotes.translate.translate(lang)
|
|
|
|
# git
|
|
@cmd
|
|
def git(param):
|
|
if isinstance(param, (list, tuple)):
|
|
param = " ".join(param)
|
|
import os
|
|
os.system("""cd lib && git %s""" % param)
|
|
os.system("""cd app && git %s""" % param)
|
|
|
|
def get_remote_and_branch(remote=None, branch=None):
|
|
if not (remote and branch):
|
|
webnotes.init()
|
|
if not webnotes.conf.branch:
|
|
raise Exception("Please specify remote and branch")
|
|
|
|
remote = remote or "origin"
|
|
branch = branch or webnotes.conf.branch
|
|
webnotes.destroy()
|
|
|
|
return remote, branch
|
|
|
|
@cmd
|
|
def pull(remote=None, branch=None):
|
|
remote, branch = get_remote_and_branch(remote, branch)
|
|
git(("pull", remote, branch))
|
|
|
|
@cmd
|
|
def push(remote=None, branch=None):
|
|
remote, branch = get_remote_and_branch(remote, branch)
|
|
git(("push", remote, branch))
|
|
|
|
@cmd
|
|
def status():
|
|
git("status")
|
|
|
|
@cmd
|
|
def commit(message):
|
|
git("""commit -a -m "%s" """ % message.replace('"', '\"'))
|
|
|
|
@cmd
|
|
def checkout(branch):
|
|
git(("checkout", branch))
|
|
|
|
@cmd
|
|
def set_admin_password(admin_password, site=None):
|
|
import webnotes
|
|
webnotes.connect(site=site)
|
|
webnotes.conn.sql("""update __Auth set `password`=password(%s)
|
|
where user='Administrator'""", (admin_password,))
|
|
webnotes.conn.commit()
|
|
webnotes.destroy()
|
|
|
|
def replace_code(start, txt1, txt2, extn, search=None, force=False):
|
|
"""replace all txt1 by txt2 in files with extension (extn)"""
|
|
import webnotes.utils
|
|
import os, re
|
|
esc = webnotes.utils.make_esc('[]')
|
|
if not search: search = esc(txt1)
|
|
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(search, content):
|
|
res = search_replace_with_prompt(fpath, txt1, txt2, force)
|
|
if res == 'skip':
|
|
return 'skip'
|
|
|
|
|
|
def search_replace_with_prompt(fpath, txt1, txt2, force=False):
|
|
""" Search and replace all txt1 by txt2 in the file with confirmation"""
|
|
from termcolor import colored
|
|
with open(fpath, 'r') as f:
|
|
content = f.readlines()
|
|
|
|
tmp = []
|
|
for c in content:
|
|
if c.find(txt1) != -1:
|
|
print fpath
|
|
print colored(txt1, 'red').join(c[:-1].split(txt1))
|
|
a = ''
|
|
if force:
|
|
c = c.replace(txt1, txt2)
|
|
else:
|
|
while a.lower() not in ['y', 'n', 'skip']:
|
|
a = raw_input('Do you want to Change [y/n/skip]?')
|
|
if a.lower() == 'y':
|
|
c = c.replace(txt1, txt2)
|
|
elif a.lower() == 'skip':
|
|
return 'skip'
|
|
tmp.append(c)
|
|
|
|
with open(fpath, 'w') as f:
|
|
f.write(''.join(tmp))
|
|
print colored('Updated', 'green')
|
|
|
|
if __name__=="__main__":
|
|
main()
|