werkit/dev.py at main · metabolize/werkit · GitHub

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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/env -S poetry run python

import os
import click
from dotenv import load_dotenv
import sh

load_dotenv()


def python_source_files():
    import glob

    return (
        glob.glob("*.py")
        + glob.glob("werkit/*.py")
        + glob.glob("werkit/**/*.py")
        + glob.glob("werkit/**/**/*.py")
    )


@click.group()
def cli():
    pass


@cli.command()
def install():
    sh.poetry(
        "install",
        "--sync",
        "--extras",
        "aws_lambda_build",
        "--extras",
        "client",
        "--extras",
        "lambda_common",
        "--extras",
        "compute_graph",
        _fg=True,
    )


@cli.command()
@click.option("--slow/--not-slow", default=True)
def test(slow):
    args = ["-m", "pytest"]
    if not slow:
        args += ["-m", "not slow"]
    sh.python3(*args, _fg=True)


@cli.command()
@click.option("--slow/--not-slow", default=True)
def coverage(slow):
    args = ["-m", "pytest", "--cov=werkit"]
    if not slow:
        args += ["-m", "not slow"]
    sh.python3(*args, _fg=True)


@cli.command()
def coverage_report():
    sh.coverage("html", _fg=True)
    sh.open("htmlcov/index.html", _fg=True)


@cli.command()
def check_types():
    sh.mypy(
        "--package",
        "werkit",
        "--show-error-codes",
        "--enable-incomplete-feature=Unpack",
        _fg=True,
    )


@cli.command()
def lint():
    sh.flake8(*python_source_files(), _fg=True)


@cli.command()
def black():
    sh.black(*python_source_files(), _fg=True)


@cli.command()
def black_check():
    sh.black("--check", *python_source_files(), _fg=True)


@cli.command()
def publish():
    sh.rm("-rf", "dist/", _fg=True)
    sh.poetry("build", _fg=True)
    sh.twine("upload", "dist/*", _fg=True)


if __name__ == "__main__":
    os.chdir(os.path.abspath(os.path.dirname(__file__)))
    cli()