Import unmodified loman into chromite.
Doing this in preparation for rewrite/sanitizing by fire that's
coming in the next rev. Imported from crosutils as of
d7a8aafa54f41ca2c9621699dcf9a10440d2ccbf .
Right now it's not enabled/exposed; this is just the initial
transferance across unmodified.
Note that the unittests were skipped in importation; they're
fairly incomplete and are about to be made irrelevant.
BUG=chromium-os:32247,chromium-os:31867,chromium-os:9914
TEST=n/a
Change-Id: Icc1c5bd2b6efcbdc0136843d74bf1c2fce0e1a5b
Reviewed-on: https://gerrit.chromium.org/gerrit/29249
Tested-by: Brian Harring <ferringb@chromium.org>
Reviewed-by: David James <davidjames@chromium.org>
Commit-Ready: Brian Harring <ferringb@chromium.org>
diff --git a/scripts/loman.py b/scripts/loman.py
new file mode 100644
index 0000000..3764a40
--- /dev/null
+++ b/scripts/loman.py
@@ -0,0 +1,150 @@
+#!/usr/bin/python
+
+# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""This module allows adding and deleting of projects to the local manifest."""
+
+import sys
+import optparse
+import os
+import xml.etree.ElementTree as ElementTree
+
+from cros_build_lib import Die, FindRepoDir
+
+
+def _ReadManifest(manifest, err_not_found=False):
+ if os.path.isfile(manifest):
+ ptree = LocalManifest(open(manifest).read())
+ elif err_not_found:
+ Die('Manifest file, %s, not found' % manifest)
+ else:
+ ptree = LocalManifest()
+ ptree.Parse()
+ return ptree
+
+
+class LocalManifest:
+ """Class which provides an abstraction for manipulating the local manifest."""
+
+ def __init__(self, text=None):
+ self._text = text or '<manifest>\n</manifest>'
+
+ def Parse(self):
+ """Parse the manifest."""
+ self._root = ElementTree.fromstring(self._text)
+
+ def AddProjectElement(self, element, workon='False', remote=None):
+ """Add a new project element to the manifest tree.
+
+ Returns:
+ True on success.
+ """
+ name = element.attrib['name']
+ path = element.attrib['path']
+ for project in self._root.findall('project'):
+ if project.attrib['path'] == path or project.attrib['name'] == name:
+ if project.attrib['path'] == path and project.attrib['name'] == name:
+ return True
+ else:
+ return False
+ element.attrib['workon'] = workon
+ if remote is not None:
+ element.attrib['remote'] = remote
+ element.tail = '\n'
+ self._root.append(element)
+ return True
+
+ def AddProject(self, name, path, workon='False', remote=None):
+ """Add a workon project if it is not already in the manifest.
+
+ Returns:
+ True on success.
+ """
+ element = ElementTree.Element('project', name=name, path=path)
+ return self.AddProjectElement(element, workon=workon, remote=remote)
+
+ def AddWorkonProjectElement(self, element):
+ return self.AddProjectElement(element, workon='True')
+
+ def AddWorkonProject(self, name, path):
+ return self.AddProject(name, path, workon='True')
+
+ def AddNonWorkonProjectElement(self, element, remote):
+ return self.AddProjectElement(element, workon='False', remote=remote)
+
+ def AddNonWorkonProject(self, name, path, remote):
+ return self.AddProject(name, path, workon='False', remote=remote)
+
+ def GetProject(self, name):
+ """Accessor method for getting a project node from the manifest tree.
+
+ Returns:
+ project element node from ElementTree, otherwise, None
+ """
+
+ for project in self._root.findall('project'):
+ if project.attrib['name'] == name:
+ return project
+ return None
+
+ def ToString(self):
+ return ElementTree.tostring(self._root, encoding='UTF-8')
+
+
+def main(argv):
+ repo_dir = FindRepoDir()
+ if not repo_dir:
+ Die("Unable to find repo dir.")
+
+ usage = 'usage: %prog add [options] <name> <path>'
+ parser = optparse.OptionParser(usage=usage)
+ parser.add_option('-w', '--workon', action='store_true', dest='workon',
+ default=False, help='Is this a workon package?')
+ parser.add_option('-f', '--file', dest='local_manifest',
+ default='%s/local_manifest.xml' % repo_dir,
+ help='Non-default manifest file to read.')
+ parser.add_option('-m', '--main', dest='main_manifest',
+ default='%s/manifest.xml' % repo_dir,
+ help='Main manifest file to read.')
+ parser.add_option('-d', '--default', dest='full_manifest',
+ default='%s/manifests/full.xml' % repo_dir,
+ help='Default manifest file to read.')
+ parser.add_option('-r', '--remote', dest='remote',
+ default=None)
+ (options, args) = parser.parse_args(argv[2:])
+ if len(args) < 1:
+ parser.error('Not enough arguments')
+ if argv[1] not in ['add']:
+ parser.error('Unsupported command: %s.' % argv[1])
+ if not options.workon and options.remote is None:
+ parser.error('Adding non-workon projects requires a remote.')
+ name = args[0]
+
+ local_tree = _ReadManifest(options.local_manifest)
+ main_tree = _ReadManifest(options.main_manifest)
+ full_tree = _ReadManifest(options.full_manifest)
+
+ # Only add this project to local_manifest.xml if not in manifest.xml
+ if options.workon:
+ if main_tree.GetProject(name) is None:
+ project_element = full_tree.GetProject(name)
+ if project_element is None:
+ Die('No project named %s, in the default manifest.' % name)
+ success = local_tree.AddWorkonProjectElement(project_element)
+ if not success:
+ Die('Name "%s" already exists with a different path.' % name)
+ else:
+ success = local_tree.AddNonWorkonProject(name, args[1], options.remote)
+ if not success:
+ Die('Name "%s" already exists with a different path.' % name)
+
+ try:
+ print >> open(options.local_manifest, 'w'), local_tree.ToString()
+ except Exception, e:
+ Die('Error writing to manifest: %s' % e)
+
+
+if __name__ == '__main__':
+ main(sys.argv)