diff --git a/bench/commands/setup.py b/bench/commands/setup.py index 4aa4a7fc..999f8dbd 100755 --- a/bench/commands/setup.py +++ b/bench/commands/setup.py @@ -7,6 +7,7 @@ import click # imports - module imports from bench.utils import exec_cmd, run_playbook, which +from bench.utils.cli import SugaredOption @click.group(help="Setup command group for enabling setting up a Frappe environment") @@ -28,13 +29,20 @@ def setup_sudoers(user): @click.option( "--logging", default="combined", type=click.Choice(["none", "site", "combined"]) ) +@click.option( + "--log_format", + help="Specify the log_format for nginx. Use none or '' to not set a value.", + only_if_set=["logging"], + cls=SugaredOption, + default="main", +) @click.option( "--yes", help="Yes to regeneration of nginx config file", default=False, is_flag=True ) -def setup_nginx(yes=False, logging="combined"): +def setup_nginx(yes=False, logging="combined", log_format=None): from bench.config.nginx import make_nginx_conf - make_nginx_conf(bench_path=".", yes=yes, logging=logging) + make_nginx_conf(bench_path=".", yes=yes, logging=logging, log_format=log_format) @click.command("reload-nginx", help="Checks NGINX config file and reloads service") diff --git a/bench/config/nginx.py b/bench/config/nginx.py index 274c4e70..1760ccc1 100644 --- a/bench/config/nginx.py +++ b/bench/config/nginx.py @@ -9,11 +9,12 @@ import click # imports - module imports import bench +import bench.config from bench.bench import Bench from bench.utils import get_bench_name -def make_nginx_conf(bench_path, yes=False, logging=None): +def make_nginx_conf(bench_path, yes=False, logging=None, log_format=None): conf_path = os.path.join(bench_path, "config", "nginx.conf") if not yes and os.path.exists(conf_path): @@ -43,9 +44,14 @@ def make_nginx_conf(bench_path, yes=False, logging=None): "allow_rate_limiting": allow_rate_limiting, # for nginx map variable "random_string": "".join(random.choice(string.ascii_lowercase) for i in range(7)), - "logging": logging, } + if logging and logging != "none": + _log_format = "" + if log_format and log_format != "none": + _log_format = log_format + template_vars["logging"] = {"level": logging, "log_format": _log_format} + if allow_rate_limiting: template_vars.update( { @@ -281,8 +287,6 @@ def use_wildcard_certificate(bench_path, ret): def get_error_pages(): - import bench - bench_app_path = os.path.abspath(bench.__path__[0]) templates = os.path.join(bench_app_path, "config", "templates") diff --git a/bench/config/templates/nginx.conf b/bench/config/templates/nginx.conf index b03f16d2..5cba5782 100644 --- a/bench/config/templates/nginx.conf +++ b/bench/config/templates/nginx.conf @@ -31,7 +31,7 @@ server { {% if allow_rate_limiting %} limit_conn per_host_{{ bench_name_hash }} 8; {% endif %} - + proxy_buffer_size 128k; proxy_buffers 4 256k; proxy_busy_buffers_size 256k; @@ -114,16 +114,18 @@ server { {% endfor -%} - {%- if logging == "site" -%} + {% if logging %} + {%- if logging.level == "site" -%} - access_log /var/log/nginx/{{ site_name }}_access.log main; + access_log /var/log/nginx/{{ site_name }}_access.log {{ logging.log_format }}; error_log /var/log/nginx/{{ site_name }}_error.log; - {%- elif logging == "combined" -%} + {%- elif logging.level == "combined" -%} - access_log /var/log/nginx/access.log main; + access_log /var/log/nginx/access.log {{ logging.log_format }}; error_log /var/log/nginx/error.log; + {%- endif %} {%- endif %} # optimizations diff --git a/bench/utils/cli.py b/bench/utils/cli.py index 7749e00f..6cae95f6 100644 --- a/bench/utils/cli.py +++ b/bench/utils/cli.py @@ -1,3 +1,4 @@ +from typing import List import click from click.core import _check_multicommand @@ -34,6 +35,27 @@ class MultiCommandGroup(click.Group): self.commands[_name] = cmd +class SugaredOption(click.Option): + def __init__(self, *args, **kwargs): + self.only_if_set: List = kwargs.pop("only_if_set") + kwargs["help"] = ( + kwargs.get("help", "") + + f". Option is acceptable only if {', '.join(self.only_if_set)} is used." + ) + super().__init__(*args, **kwargs) + + def handle_parse_result(self, ctx, opts, args): + current_opt = self.name in opts + if current_opt and self.only_if_set: + for opt in self.only_if_set: + if opt not in opts: + deafaults_set = [x.default for x in ctx.command.params if x.name == opt] + if not deafaults_set: + raise click.UsageError(f"Illegal Usage: Set '{opt}' before '{self.name}'.") + + return super().handle_parse_result(ctx, opts, args) + + def use_experimental_feature(ctx, param, value): if not value: return