8bf6dd385a
Rational: when calling oow psql, sometimes, we don't want to put a database. Example, if we call "oow -d postgres -c 'psql'". In that case, this improvment makes that default postgres database will be proposed in the 'prompt', if no database is provided
111 lines
3.0 KiB
Python
111 lines
3.0 KiB
Python
import click
|
|
|
|
|
|
def versions_options(function):
|
|
function = click.option(
|
|
"-v",
|
|
"--versions",
|
|
type=str,
|
|
help="Coma-separated values of odoo versions for which"
|
|
" you want to perform the operation."
|
|
" Let empty to perform the operation on all the versions"
|
|
" of the project",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def step_option(function):
|
|
function = click.option(
|
|
"-s",
|
|
"--step",
|
|
required=True,
|
|
prompt=True,
|
|
type=int,
|
|
help="Migration step for which you want to perform the operation.",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def first_step_option(function):
|
|
function = click.option(
|
|
"--first-step",
|
|
type=int,
|
|
help="First step for which to perform the operation",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def last_step_option(function):
|
|
function = click.option(
|
|
"--last-step",
|
|
type=int,
|
|
help="Last step for which to perform the operation",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def demo_option(function):
|
|
function = click.option(
|
|
"--with-demo/--without-demo",
|
|
default=False,
|
|
help="Create database with or without demo data.",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def database_option_required(function):
|
|
function = click.option(
|
|
"-d",
|
|
"--database",
|
|
required=True,
|
|
prompt=True,
|
|
default="postgres",
|
|
type=str,
|
|
help="Odoo Database for which you want to perform the operation.",
|
|
)(function)
|
|
return function
|
|
|
|
|
|
def get_odoo_versions_from_options(ctx, versions_arg):
|
|
if not versions_arg:
|
|
return ctx.obj["config"]["odoo_versions"]
|
|
else:
|
|
odoo_versions = []
|
|
versions = [float(x) for x in versions_arg.split(",")]
|
|
for odoo_version in ctx.obj["config"]["odoo_versions"]:
|
|
if odoo_version in versions:
|
|
odoo_versions.append(odoo_version)
|
|
return odoo_versions
|
|
|
|
|
|
def get_migration_step_from_options(ctx, step_arg):
|
|
step = float(step_arg)
|
|
for migration_step in ctx.obj["config"]["migration_steps"]:
|
|
if migration_step["name"] == step:
|
|
return migration_step
|
|
raise ValueError(
|
|
f"No migration step found in configuration for step {step_arg}"
|
|
)
|
|
|
|
|
|
def get_migration_steps_from_options(ctx, first_step_arg, last_step_arg):
|
|
result = []
|
|
if first_step_arg:
|
|
first_step = int(first_step_arg)
|
|
else:
|
|
first_step = ctx.obj["config"]["migration_steps"][0]["name"]
|
|
if last_step_arg:
|
|
last_step = int(last_step_arg)
|
|
else:
|
|
last_step = ctx.obj["config"]["migration_steps"][-1]["name"]
|
|
for migration_step in ctx.obj["config"]["migration_steps"]:
|
|
if migration_step["name"] in list(range(first_step, last_step + 1)):
|
|
result.append(migration_step.copy())
|
|
if result:
|
|
return result
|
|
|
|
raise ValueError(
|
|
"Unable to define steps in configuration from options."
|
|
f" (first step {first_step_arg} ; last step {last_step_arg})"
|
|
)
|