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
|
import os
from datetime import datetime
from invoke import task
from .utils import ctx_run
@task
def install(ctx, upgrade=False):
"""
Install dependencies and update lock file.
"""
if upgrade:
ctx_run(ctx, "poetry update")
else:
ctx_run(ctx, "poetry lock")
ctx_run(ctx, "poetry install")
@task
def clean(ctx):
"""
Remove build files e.g. package, distributable, compiled etc.
"""
ctx_run(ctx, "rm -rf *.egg-info dist build __pycache__ .pytest_cache artifacts/*")
def version_scheme(v):
if v.exact:
return v.format_with("{tag}")
return datetime.now().strftime("%Y.%m.%d.%H%M%S%f")
@task(pre=[clean])
def dist(ctx):
"""
Generate version from scm and build package distributable
"""
from setuptools_scm import get_version
version = get_version(version_scheme=version_scheme, local_scheme=lambda _: "")
ctx_run(ctx, f"poetry version {version}")
ctx_run(ctx, "poetry build")
@task
def publish(ctx, dry_run=True):
"""
Upload built package to pypi
"""
repo_url = "--repository-url https://test.pypi.org/legacy/" if dry_run else ""
ctx_run(ctx, f"twine upload --skip-existing {repo_url} dist/*")
@task(pre=[dist])
def release(ctx, dry_run=True, version=None):
"""
Build and publish package to pypi index based on scm version
"""
from semver.version import Version
if not dry_run and not os.environ.get("CI"):
print("This is a CI only command")
exit(1)
if dry_run and not version:
version = ctx_run(ctx, "poetry version --short").stdout.strip()
if not version:
print("Missing version.")
exit(1)
try:
Version.parse(version)
should_publish_to_pypi = not dry_run
except ValueError:
should_publish_to_pypi = False
# publish to test to verify builds
if dry_run:
publish(ctx, dry_run=True)
# publish to pypi if test succeeds
if should_publish_to_pypi:
publish(ctx, dry_run=False)
|
