mirror of
https://github.com/frappe/bench.git
synced 2025-01-25 07:58:24 +00:00
114 lines
3.5 KiB
Python
Executable File
114 lines
3.5 KiB
Python
Executable File
import click
|
|
import os, sys, logging, json, pwd, subprocess
|
|
from bench.utils import is_root, PatchError, drop_privileges, get_env_cmd, get_cmd_output, get_frappe
|
|
from bench.app import get_apps
|
|
from bench.config.common_site_config import get_config
|
|
from bench.commands import bench_command
|
|
|
|
logger = logging.getLogger('bench')
|
|
from_command_line = False
|
|
|
|
def cli():
|
|
global from_command_line
|
|
from_command_line = True
|
|
|
|
check_uid()
|
|
change_dir()
|
|
change_uid()
|
|
|
|
if len(sys.argv) > 2 and sys.argv[1] == "frappe":
|
|
return old_frappe_cli()
|
|
|
|
elif len(sys.argv) > 1 and sys.argv[1] in get_frappe_commands():
|
|
return frappe_cmd()
|
|
|
|
elif len(sys.argv) > 1 and sys.argv[1] in ("--site", "--verbose", "--force", "--profile"):
|
|
return frappe_cmd()
|
|
|
|
elif len(sys.argv) > 1 and sys.argv[1]=="--help":
|
|
print(click.Context(bench_command).get_help())
|
|
print()
|
|
print(get_frappe_help())
|
|
return
|
|
|
|
elif len(sys.argv) > 1 and sys.argv[1] in get_apps():
|
|
return app_cmd()
|
|
|
|
else:
|
|
try:
|
|
# NOTE: this is the main bench command
|
|
bench_command()
|
|
except PatchError:
|
|
sys.exit(1)
|
|
|
|
def check_uid():
|
|
if cmd_requires_root() and not is_root():
|
|
print('superuser privileges required for this command')
|
|
sys.exit(1)
|
|
|
|
def cmd_requires_root():
|
|
if len(sys.argv) > 2 and sys.argv[2] in ('production', 'sudoers', 'lets-encrypt', 'fonts',
|
|
'print', 'firewall', 'ssh-port', 'role', 'fail2ban'):
|
|
return True
|
|
if len(sys.argv) >= 2 and sys.argv[1] in ('patch', 'renew-lets-encrypt', 'disable-production',
|
|
'install'):
|
|
return True
|
|
|
|
def change_dir():
|
|
if os.path.exists('config.json') or "init" in sys.argv:
|
|
return
|
|
dir_path_file = '/etc/frappe_bench_dir'
|
|
if os.path.exists(dir_path_file):
|
|
with open(dir_path_file) as f:
|
|
dir_path = f.read().strip()
|
|
if os.path.exists(dir_path):
|
|
os.chdir(dir_path)
|
|
|
|
def change_uid():
|
|
if is_root() and not cmd_requires_root():
|
|
frappe_user = get_config(".").get('frappe_user')
|
|
if frappe_user:
|
|
drop_privileges(uid_name=frappe_user, gid_name=frappe_user)
|
|
os.environ['HOME'] = pwd.getpwnam(frappe_user).pw_dir
|
|
else:
|
|
print('You should not run this command as root')
|
|
sys.exit(1)
|
|
|
|
def old_frappe_cli(bench_path='.'):
|
|
f = get_frappe(bench_path=bench_path)
|
|
os.chdir(os.path.join(bench_path, 'sites'))
|
|
os.execv(f, [f] + sys.argv[2:])
|
|
|
|
def app_cmd(bench_path='.'):
|
|
f = get_env_cmd('python', bench_path=bench_path)
|
|
os.chdir(os.path.join(bench_path, 'sites'))
|
|
os.execv(f, [f] + ['-m', 'frappe.utils.bench_helper'] + sys.argv[1:])
|
|
|
|
def frappe_cmd(bench_path='.'):
|
|
f = get_env_cmd('python', bench_path=bench_path)
|
|
os.chdir(os.path.join(bench_path, 'sites'))
|
|
os.execv(f, [f] + ['-m', 'frappe.utils.bench_helper', 'frappe'] + sys.argv[1:])
|
|
|
|
def get_frappe_commands(bench_path='.'):
|
|
python = get_env_cmd('python', bench_path=bench_path)
|
|
sites_path = os.path.join(bench_path, 'sites')
|
|
if not os.path.exists(sites_path):
|
|
return []
|
|
try:
|
|
output = get_cmd_output("{python} -m frappe.utils.bench_helper get-frappe-commands".format(python=python), cwd=sites_path)
|
|
# output = output.decode('utf-8')
|
|
return json.loads(output)
|
|
except subprocess.CalledProcessError:
|
|
return []
|
|
|
|
def get_frappe_help(bench_path='.'):
|
|
python = get_env_cmd('python', bench_path=bench_path)
|
|
sites_path = os.path.join(bench_path, 'sites')
|
|
if not os.path.exists(sites_path):
|
|
return []
|
|
try:
|
|
out = get_cmd_output("{python} -m frappe.utils.bench_helper get-frappe-help".format(python=python), cwd=sites_path)
|
|
return "Framework commands:\n" + out.split('Commands:')[1]
|
|
except subprocess.CalledProcessError:
|
|
return ""
|