git-sync-deps: handle recursion and deps_os at the same time

tools/git-sync-deps

 #!/usr/bin/python
 # Copyright 2014 Google Inc.
 #
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 
 """Parse a DEPS file and git checkout all of the dependencies.
 
 Args:
@@ skipping 136 matching lines @@
   execfile(path, dictionary)
   return dictionary
 
 
 class DepsError(Exception):
   """Raised if deps_os is a bad key.
   """
   pass
 
 
-def git_sync_deps(deps_file_path, deps_os_list, verbose):
+def git_sync_deps(deps_file_path, command_line_os_requests, verbose):
   """Grab dependencies, with optional platform support.
 
   Args:
     deps_file_path (string) Path to the DEPS file.
 
     deps_os_list (list of strings) Can be empty list.  List of

[mtklein, 2014/08/26 18:58:36]

update?

                  strings that should each be a key in the deps_os
                  dictionary in the DEPS file.
 
   Raises DepsError exception and git Exceptions.
   """
   git = git_executable()
   assert git
 
   deps_file_directory = os.path.dirname(deps_file_path)
-  deps = parse_file_to_dict(deps_file_path)
-  dependencies = deps['deps'].copy()
-  for deps_os in deps_os_list:
+  deps_file = parse_file_to_dict(deps_file_path)
+  dependencies = deps_file['deps'].copy()
+  os_specific_dependencies = deps_file.get('deps_os', [])
+  for os_name in command_line_os_requests:
     # Add OS-specific dependencies
-    if deps_os not in deps['deps_os']:
-      raise DepsError(
-        'Argument "%s" not found within deps_os keys %r' %
-        (deps_os, deps['deps_os'].keys()))
-    for dep in deps['deps_os'][deps_os]:
-      dependencies[dep] = deps['deps_os'][deps_os][dep]
+    if os_name in os_specific_dependencies:
+      dependencies.update(os_specific_dependencies[os_name])
   list_of_arg_lists = []
   for directory in dependencies:
     if '@' in dependencies[directory]:
       repo, checkoutable = dependencies[directory].split('@', 1)
     else:
       repo, checkoutable = dependencies[directory], 'origin/master'
 
     relative_directory = os.path.join(deps_file_directory, directory)
 
     list_of_arg_lists.append(
       (git, repo, checkoutable, relative_directory, verbose))
 
   multithread(git_checkout_to_directory, list_of_arg_lists)
 
-  for directory in deps.get('recursedeps', []):
+  for directory in deps_file.get('recursedeps', []):
     recursive_path = os.path.join(deps_file_directory, directory, 'DEPS')
-    git_sync_deps(recursive_path, deps_os_list, verbose)
+    git_sync_deps(recursive_path, command_line_os_requests, verbose)
 
 
 def multithread(function, list_of_arg_lists):
   # for args in list_of_arg_lists:
   #   function(*args)
   # return
   threads = []
   for args in list_of_arg_lists:
     thread = threading.Thread(None, function, None, args)
     thread.start()
     threads.append(thread)
   for thread in threads:
     thread.join()
 
 
 def main(argv):
   deps_file_path = os.environ.get('GIT_SYNC_DEPS_PATH', DEFAULT_DEPS_PATH)
   verbose = not bool(os.environ.get('GIT_SYNC_DEPS_QUIET', False))
   try:
     git_sync_deps(deps_file_path, argv, verbose)
     return 0
   except DepsError:
     usage(deps_file_path)
     return 1
 
 
 if __name__ == '__main__':
   exit(main(sys.argv[1:]))