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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#!/usr/bin/python
#
# Copyright 2010 Google Inc. All Rights Reserved.
"""Script to build the ChromeOS toolchain.
This script sets up the toolchain if you give it the gcctools directory.
"""
__author__ = "asharif@google.com (Ahmad Sharif)"
import datetime
import os
import optparse
import sys
from utils import email_sender
class SheriffHandler(object):
SHERIFF_FILE = "/home/mobiletc-prebuild/sheriffs.txt"
SUBJECT = "You (%s) are the sheriff for the week: %s - %s"
BODY = "Please see instructions here: https://sites.google.com/a/google.com/compiler-chromeos-workflows/sheriff-corner/sheriff-duties."
def GetWeekInfo(self, day=datetime.datetime.today()):
"""Return week_start, week_end."""
epoch = datetime.datetime.utcfromtimestamp(0)
delta_since_epoch = day - epoch
abs_days = abs(delta_since_epoch.days) - 2 # To get it to start from Sat.
weeks_since_epoch = abs_days / 7
day_of_week = abs_days % 7
week_begin = day - datetime.timedelta(days=day_of_week)
week_end = day + datetime.timedelta(days=(6 - day_of_week))
strftime_format = "%A, %B %d %Y"
return (week_begin.strftime(strftime_format),
week_end.strftime(strftime_format))
def GetCurrentSheriff(self):
"""Return the current sheriff."""
return self.ReadSheriffsAsList()[0]
def ReadSheriffsAsList(self):
"""Return the sheriff file contents."""
contents = ""
with open(self.SHERIFF_FILE, "r") as f:
contents = f.read()
return contents.splitlines()
def WriteSheriffsAsList(self, to_write):
with open(self.SHERIFF_FILE, "w") as f:
f.write("\n".join(to_write))
def GetRotatedSheriffs(self, num_rotations=1):
"""Return the sheriff file contents."""
sheriff_list = self.ReadSheriffsAsList()
new_sheriff_list = []
num_rotations = num_rotations % len(sheriff_list)
new_sheriff_list = (sheriff_list[num_rotations:] +
sheriff_list[:num_rotations])
return new_sheriff_list
def Email(self):
es = email_sender.EmailSender()
current_sheriff = self.GetCurrentSheriff()
week_start, week_end = self.GetWeekInfo()
subject = self.SUBJECT % (current_sheriff, week_start, week_end)
es.SendEmail([current_sheriff],
subject,
self.BODY,
email_from=os.path.basename(__file__),
email_cc=["c-compiler-chrome"])
def Main(argv):
parser = optparse.OptionParser()
parser.add_option("-e",
"--email",
dest="email",
action="store_true",
help="Email the sheriff.")
parser.add_option("-r",
"--rotate",
dest="rotate",
help="Print sheriffs after n rotations.")
parser.add_option("-w",
"--write",
dest="write",
action="store_true",
default=False,
help="Wrote rotated contents to the sheriff file.")
options, _ = parser.parse_args(argv)
sheriff_handler = SheriffHandler()
current_sheriff = sheriff_handler.GetCurrentSheriff()
week_start, week_end = sheriff_handler.GetWeekInfo()
print "Current sheriff: %s (%s - %s)" % (current_sheriff, week_start, week_end)
if options.email:
sheriff_handler.Email()
if options.rotate:
rotated_sheriffs = sheriff_handler.GetRotatedSheriffs(int(options.rotate))
print "Rotated sheriffs (after %s rotations)" % options.rotate
print "\n".join(rotated_sheriffs)
if options.write:
sheriff_handler.WriteSheriffsAsList(rotated_sheriffs)
print "Rotated sheriffs written to file."
return 0
if __name__ == "__main__":
retval = Main(sys.argv)
sys.exit(retval)
|
