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
|
# pyinfra
# File: pyinfra/example/config.py
# Desc: entirely optional config file for the CLI deploy
# see: pyinfra/api/config.py for defaults
from pyinfra import local, hook
# These can be here or in deploy.py
TIMEOUT = 1
FAIL_PERCENT = 81
# Deploy hook, can also be in deploy.py
@hook.before_connect
def ensure_branch(data, state):
# Check something local is correct, etc
branch = local.shell('git rev-parse --abbrev-ref HEAD')
app_branch = data.app_branch
if branch != app_branch:
# Raise hook.Error for pyinfra to handle
raise hook.Error('We\'re on the wrong branch (want {0}, got {1})!'.format(
app_branch, branch
))
@hook.after_deploy
def notify_people(data, state):
print('After deploy hook!')
|
