104 lines
2.7 KiB
Python
104 lines
2.7 KiB
Python
import click
|
|
import frappe
|
|
import os
|
|
import json
|
|
import importlib
|
|
import frappe.utils
|
|
import traceback
|
|
import warnings
|
|
|
|
click.disable_unicode_literals_warning = True
|
|
|
|
def main():
|
|
commands = get_app_groups()
|
|
commands.update({
|
|
'get-frappe-commands': get_frappe_commands,
|
|
'get-frappe-help': get_frappe_help
|
|
})
|
|
click.Group(commands=commands)(prog_name='bench')
|
|
|
|
def get_app_groups():
|
|
'''Get all app groups, put them in main group "frappe" since bench is
|
|
designed to only handle that'''
|
|
commands = dict()
|
|
for app in get_apps():
|
|
app_commands = get_app_commands(app)
|
|
if app_commands:
|
|
commands.update(app_commands)
|
|
|
|
ret = dict(frappe=click.group(name='frappe', commands=commands)(app_group))
|
|
return ret
|
|
|
|
def get_app_group(app):
|
|
app_commands = get_app_commands(app)
|
|
if app_commands:
|
|
return click.group(name=app, commands=app_commands)(app_group)
|
|
|
|
@click.option('--site')
|
|
@click.option('--profile', is_flag=True, default=False, help='Profile')
|
|
@click.option('--verbose', is_flag=True, default=False, help='Verbose')
|
|
@click.option('--force', is_flag=True, default=False, help='Force')
|
|
@click.pass_context
|
|
def app_group(ctx, site=False, force=False, verbose=False, profile=False):
|
|
ctx.obj = {
|
|
'sites': get_sites(site),
|
|
'force': force,
|
|
'verbose': verbose,
|
|
'profile': profile
|
|
}
|
|
if ctx.info_name == 'frappe':
|
|
ctx.info_name = ''
|
|
|
|
def get_sites(site_arg):
|
|
if site_arg == 'all':
|
|
return frappe.utils.get_sites()
|
|
elif site_arg:
|
|
return [site_arg]
|
|
elif os.environ.get('FRAPPE_SITE'):
|
|
return [os.environ.get('FRAPPE_SITE')]
|
|
elif os.path.exists('currentsite.txt'):
|
|
with open('currentsite.txt') as f:
|
|
site = f.read().strip()
|
|
if site:
|
|
return [site]
|
|
return []
|
|
|
|
def get_app_commands(app):
|
|
if os.path.exists(os.path.join('..', 'apps', app, app, 'commands.py'))\
|
|
or os.path.exists(os.path.join('..', 'apps', app, app, 'commands', '__init__.py')):
|
|
try:
|
|
app_command_module = importlib.import_module(app + '.commands')
|
|
except Exception:
|
|
traceback.print_exc()
|
|
return []
|
|
else:
|
|
return []
|
|
|
|
ret = {}
|
|
for command in getattr(app_command_module, 'commands', []):
|
|
ret[command.name] = command
|
|
return ret
|
|
|
|
@click.command('get-frappe-commands')
|
|
def get_frappe_commands():
|
|
commands = list(get_app_commands('frappe'))
|
|
|
|
for app in get_apps():
|
|
app_commands = get_app_commands(app)
|
|
if app_commands:
|
|
commands.extend(list(app_commands))
|
|
|
|
print(json.dumps(commands))
|
|
|
|
@click.command('get-frappe-help')
|
|
def get_frappe_help():
|
|
print(click.Context(get_app_groups()['frappe']).get_help())
|
|
|
|
def get_apps():
|
|
return frappe.get_all_apps(with_internal_apps=False, sites_path='.')
|
|
|
|
if __name__ == "__main__":
|
|
if not frappe._dev_server:
|
|
warnings.simplefilter('ignore')
|
|
|
|
main()
|