• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/python2
2#
3# Copyright 2010 Google Inc. All Rights Reserved.
4"""Script to rotate the weekly team sheriff.
5
6This script determines who the next sheriff is, updates the file
7appropriately and sends out email notifying the team.
8"""
9
10from __future__ import print_function
11
12__author__ = 'asharif@google.com (Ahmad Sharif)'
13
14import argparse
15import datetime
16import os
17import sys
18
19from cros_utils import constants
20from cros_utils import email_sender
21
22
23class SheriffHandler(object):
24  """Main class for handling sheriff rotations."""
25
26  SHERIFF_FILE = os.path.join(constants.CROSTC_WORKSPACE, 'sheriffs.txt')
27  SUBJECT = 'You (%s) are the sheriff for the week: %s - %s'
28  BODY = ('Please see instructions here: '
29          'https://sites.google.com/a/google.com/chromeos-toolchain-team-home2'
30          '/home/sheriff-s-corner/sheriff-duties')
31
32  def GetWeekInfo(self, day=datetime.datetime.today()):
33    """Return week_start, week_end."""
34
35    epoch = datetime.datetime.utcfromtimestamp(0)
36    delta_since_epoch = day - epoch
37
38    abs_days = abs(delta_since_epoch.days) - 2  # To get it to start from Sat.
39    day_of_week = abs_days % 7
40
41    week_begin = day - datetime.timedelta(days=day_of_week)
42    week_end = day + datetime.timedelta(days=(6 - day_of_week))
43
44    strftime_format = '%A, %B %d %Y'
45
46    return (week_begin.strftime(strftime_format),
47            week_end.strftime(strftime_format))
48
49  def GetCurrentSheriff(self):
50    """Return the current sheriff."""
51    return self.ReadSheriffsAsList()[0]
52
53  def ReadSheriffsAsList(self):
54    """Return the sheriff file contents."""
55    contents = ''
56    with open(self.SHERIFF_FILE, 'r') as f:
57      contents = f.read()
58    return contents.splitlines()
59
60  def WriteSheriffsAsList(self, to_write):
61    with open(self.SHERIFF_FILE, 'w') as f:
62      f.write('\n'.join(to_write))
63
64  def GetRotatedSheriffs(self, num_rotations=1):
65    """Return the sheriff file contents."""
66    sheriff_list = self.ReadSheriffsAsList()
67
68    new_sheriff_list = []
69    num_rotations = num_rotations % len(sheriff_list)
70    new_sheriff_list = (
71        sheriff_list[num_rotations:] + sheriff_list[:num_rotations])
72    return new_sheriff_list
73
74  def Email(self):
75    es = email_sender.EmailSender()
76    current_sheriff = self.GetCurrentSheriff()
77    week_start, week_end = self.GetWeekInfo()
78    subject = self.SUBJECT % (current_sheriff, week_start, week_end)
79    es.SendEmail([current_sheriff],
80                 subject,
81                 self.BODY,
82                 email_from=os.path.basename(__file__),
83                 email_cc=['c-compiler-chrome'])
84
85
86def Main(argv):
87  parser = argparse.ArgumentParser()
88  parser.add_argument('-e',
89                      '--email',
90                      dest='email',
91                      action='store_true',
92                      help='Email the sheriff.')
93  parser.add_argument('-r',
94                      '--rotate',
95                      dest='rotate',
96                      help='Print sheriffs after n rotations.')
97  parser.add_argument('-w',
98                      '--write',
99                      dest='write',
100                      action='store_true',
101                      default=False,
102                      help='Wrote rotated contents to the sheriff file.')
103
104  options = parser.parse_args(argv)
105
106  sheriff_handler = SheriffHandler()
107
108  current_sheriff = sheriff_handler.GetCurrentSheriff()
109  week_start, week_end = sheriff_handler.GetWeekInfo()
110
111  print('Current sheriff: %s (%s - %s)' % (current_sheriff, week_start,
112                                           week_end))
113
114  if options.email:
115    sheriff_handler.Email()
116
117  if options.rotate:
118    rotated_sheriffs = sheriff_handler.GetRotatedSheriffs(int(options.rotate))
119    print('Rotated sheriffs (after %s rotations)' % options.rotate)
120    print('\n'.join(rotated_sheriffs))
121    if options.write:
122      sheriff_handler.WriteSheriffsAsList(rotated_sheriffs)
123      print('Rotated sheriffs written to file.')
124
125  return 0
126
127
128if __name__ == '__main__':
129  retval = Main(sys.argv[1:])
130  sys.exit(retval)
131