-
Notifications
You must be signed in to change notification settings - Fork 1
/
tasks.py
50 lines (43 loc) · 1.12 KB
/
tasks.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from pathlib import Path
from adit_radis_shared import invoke_tasks
from adit_radis_shared.invoke_tasks import ( # noqa: F401
backup_db,
format,
init_workspace,
lint,
reset_dev,
restore_db,
show_outdated,
stack_deploy,
stack_rm,
test,
try_github_actions,
upgrade_adit_radis_shared,
upgrade_postgresql,
web_shell,
)
from invoke.context import Context
from invoke.tasks import task
invoke_tasks.PROJECT_NAME = "radis"
invoke_tasks.PROJECT_DIR = Path(__file__).resolve().parent
@task
def compose_up(
ctx: Context,
env: invoke_tasks.Environments = "dev",
no_build: bool = False,
gpu: bool = False,
):
"""Start containers in specified environment"""
if gpu:
profiles = ["gpu"]
else:
profiles = ["cpu"]
invoke_tasks.compose_up(ctx, env=env, no_build=no_build, profile=profiles)
@task
def compose_down(
ctx: Context,
env: invoke_tasks.Environments = "dev",
cleanup: bool = False,
):
"""Stop containers in specified environment"""
invoke_tasks.compose_down(ctx, env=env, cleanup=cleanup, profile=["cpu", "gpu", "mock"])