doc/tools/cvsinfo.py

   1 """Utility class and function to get information about the CVS repository
   2 based on checked-out files.
   3 """
   4
   5 import os
   6
   7
   8 def get_repository_list(paths):
   9     d = {}
  10     for name in paths:
  11         if os.path.isfile(name):
  12             dir = os.path.dirname(name)
  13         else:
  14             dir = name
  15         rootfile = os.path.join(name, "CVS", "Root")
  16         root = open(rootfile).readline().strip()
  17         if not d.has_key(root):
  18             d[root] = RepositoryInfo(dir), [name]
  19         else:
  20             d[root][1].append(name)
  21     return d.values()
  22
  23
  24 class RepositoryInfo:
  25     """Record holding information about the repository we want to talk to."""
  26     cvsroot_path = None
  27     branch = None
  28
  29     # type is '', ':ext', or ':pserver:'
  30     type = ""
  31
  32     def __init__(self, dir=None):
  33         if dir is None:
  34             dir = os.getcwd()
  35         dir = os.path.join(dir, "CVS")
  36         root = open(os.path.join(dir, "Root")).readline().strip()
  37         if root.startswith(":pserver:"):
  38             self.type = ":pserver:"
  39             root = root[len(":pserver:"):]
  40         elif ":" in root:
  41             if root.startswith(":ext:"):
  42                 root = root[len(":ext:"):]
  43             self.type = ":ext:"
  44         self.repository = root
  45         if ":" in root:
  46             host, path = root.split(":", 1)
  47             self.cvsroot_path = path
  48         else:
  49             self.cvsroot_path = root
  50         fn = os.path.join(dir, "Tag")
  51         if os.path.isfile(fn):
  52             self.branch = open(fn).readline().strip()[1:]
  53
  54     def get_cvsroot(self):
  55         return self.type + self.repository
  56
  57     _repository_dir_cache = {}
  58
  59     def get_repository_file(self, path):
  60         filename = os.path.abspath(path)
  61         if os.path.isdir(path):
  62             dir = path
  63             join = 0
  64         else:
  65             dir = os.path.dirname(path)
  66             join = 1
  67         try:
  68             repodir = self._repository_dir_cache[dir]
  69         except KeyError:
  70             repofn = os.path.join(dir, "CVS", "Repository")
  71             repodir = open(repofn).readline().strip()
  72             repodir = os.path.join(self.cvsroot_path, repodir)
  73             self._repository_dir_cache[dir] = repodir
  74         if join:
  75             fn = os.path.join(repodir, os.path.basename(path))
  76         else:
  77             fn = repodir
  78         return fn[len(self.cvsroot_path)+1:]
  79
  80     def __repr__(self):
  81         return "<RepositoryInfo for %s>" % `self.get_cvsroot()`