Die build.py, Die: Part 2 (LocalFetcher, Handlebar support, build script)

chrome/common/extensions/docs/server2/branch_utility.py

 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import logging
 import re
 import json
 
 class BranchUtility(object):
   """Utility class for dealing with different doc branches.
   """
   def __init__(self, urlfetch):
     self.omaha_proxy_url = 'http://omahaproxy.appspot.com/json'
     self.urlfetch = urlfetch
 
   def SetURL(self, url):
     self.omaha_proxy_url = url
 
   def GetChannelNameFromPath(self, path):
     first_part = path.split('/')[0]
-    if first_part in ['trunk', 'dev', 'beta', 'stable']:
+    if first_part in ['trunk', 'dev', 'beta', 'stable', 'local']:
       return first_part
     else:
       return 'stable'
 
   def GetBranchNumberForChannelName(self, channel_name):
     """Returns an empty string if the branch number cannot be found.
     Throws exception on network errors.
     """
-    if channel_name == 'trunk':
-      return 'trunk'
+    if channel_name == 'trunk' or channel_name == 'local':
+      return channel_name
 
     fetch_data = self.urlfetch.fetch(self.omaha_proxy_url)
     if fetch_data.content == '':
       raise Exception('Fetch returned zero results.')
 
     version_json = json.loads(fetch_data.content)
     branch_numbers = {}
     for entry in version_json:
       if entry['os'] not in ['win', 'linux', 'mac', 'cros']:
         continue
       for version in entry['versions']:
         if version['channel'] != channel_name:
           continue
         if version['true_branch'] not in branch_numbers:
           branch_numbers[version['true_branch']] = 0
         else:
           branch_numbers[version['true_branch']] += 1
 
     sorted_list = [x for x in branch_numbers.iteritems()]
     sorted_list.sort(key = lambda x: x[1])
     sorted_list.reverse()
 
     branch_number, _ = sorted_list[0]
 
     return branch_number