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
|
# Copyright Jean-Paul Calderone. See LICENSE file for details.
"""
Bazaar plugin for automatically writing out an updated iCalendar schedule.
"""
import sys, os, tempfile
from bzrlib import branch
from twisted.python.filepath import FilePath
def post_change_branch_tip(params):
conf = params.branch.get_config()
sys.stderr.write(str(conf) + '\n')
if not conf.get_user_option('farm-schedule:enabled'):
return
destination = conf.get_user_option('farm-schedule:destination')
destination = os.path.expanduser(destination)
try:
checkout = tempfile.mktemp()
sys.stderr.write('Updating code...\n')
make_importable(params.branch, checkout)
sys.stderr.write('Writing schedule...\n')
write_schedule(destination)
sys.stderr.write('Done\n')
finally:
FilePath(checkout).remove()
def make_importable(branch, checkout):
branch.create_checkout(to_location=checkout, lightweight=True)
sys.path.append(checkout)
return checkout
def write_schedule(destination):
from cropplan import (
__file__, load_crops, load_seeds, create_tasks, schedule_tasks,
schedule_ical)
HERE = FilePath(__file__).realpath().parent()
CROP_PLAN = HERE.child('2012 Crop Plan.csv')
CROP_VARIETIES = HERE.child('2012 Crop Plan - Varieties.csv')
crops = load_crops(CROP_PLAN)
sys.stderr.write('Loaded %d crops...\n' % (len(crops),))
seeds = load_seeds(CROP_VARIETIES, crops)
sys.stderr.write('Loaded %d seeds...\n' % (len(seeds),))
tasks = create_tasks(crops, seeds)
schedule = schedule_tasks(tasks)
output = file(destination, 'w')
stdout = sys.stdout
try:
sys.stdout = output
schedule_ical(schedule)
finally:
sys.stdout = stdout
branch.Branch.hooks.install_named_hook(
'post_change_branch_tip', post_change_branch_tip, 'deployment restart')
|