cvs2svn_lib/main.py

   1 #!/usr/bin/env python
   2 # (Be in -*- python -*- mode.)
   3 #
   4 # ====================================================================
   5 # Copyright (c) 2000-2009 CollabNet.  All rights reserved.
   6 #
   7 # This software is licensed as described in the file COPYING, which
   8 # you should have received as part of this distribution.  The terms
   9 # are also available at http://subversion.tigris.org/license-1.html.
  10 # If newer versions of this license are posted there, you may use a
  11 # newer version instead, at your option.
  12 #
  13 # This software consists of voluntary contributions made by many
  14 # individuals.  For exact contribution history, see the revision
  15 # history and logs, available at http://cvs2svn.tigris.org/.
  16 # ====================================================================
  17
  18 import os
  19 import errno
  20 import gc
  21
  22 try:
  23   # Try to get access to a bunch of encodings for use with --encoding.
  24   # See http://cjkpython.i18n.org/ for details.
  25   import iconv_codec
  26 except ImportError:
  27   pass
  28
  29 from cvs2svn_lib.common import FatalError
  30 from cvs2svn_lib.svn_run_options import SVNRunOptions
  31 from cvs2svn_lib.git_run_options import GitRunOptions
  32 from cvs2svn_lib.bzr_run_options import BzrRunOptions
  33 from cvs2svn_lib.context import Ctx
  34 from cvs2svn_lib.pass_manager import PassManager
  35 from cvs2svn_lib.passes import passes
  36
  37
  38 def main(progname, run_options, pass_manager):
  39   # Disable garbage collection, as we try not to create any circular
  40   # data structures:
  41   gc.disable()
  42
  43   # Convenience var, so we don't have to keep instantiating this Borg.
  44   ctx = Ctx()
  45
  46   # Make sure the tmp directory exists.  Note that we don't check if
  47   # it's empty -- we want to be able to use, for example, "." to hold
  48   # tempfiles.
  49   if not os.path.exists(ctx.tmpdir):
  50     erase_tmpdir = True
  51     os.mkdir(ctx.tmpdir)
  52   elif not os.path.isdir(ctx.tmpdir):
  53     raise FatalError(
  54         "cvs2svn tried to use '%s' for temporary files, but that path\n"
  55         "  exists and is not a directory.  Please make it be a directory,\n"
  56         "  or specify some other directory for temporary files."
  57         % (ctx.tmpdir,))
  58   else:
  59     erase_tmpdir = False
  60
  61   # But do lock the tmpdir, to avoid process clash.
  62   try:
  63     os.mkdir(os.path.join(ctx.tmpdir, 'cvs2svn.lock'))
  64   except OSError, e:
  65     if e.errno == errno.EACCES:
  66       raise FatalError("Permission denied:"
  67                        + " No write access to directory '%s'." % ctx.tmpdir)
  68     if e.errno == errno.EEXIST:
  69       raise FatalError(
  70           "cvs2svn is using directory '%s' for temporary files, but\n"
  71           "  subdirectory '%s/cvs2svn.lock' exists, indicating that another\n"
  72           "  cvs2svn process is currently using '%s' as its temporary\n"
  73           "  workspace.  If you are certain that is not the case,\n"
  74           "  then remove the '%s/cvs2svn.lock' subdirectory."
  75           % (ctx.tmpdir, ctx.tmpdir, ctx.tmpdir, ctx.tmpdir,))
  76     raise
  77
  78   try:
  79     if run_options.profiling:
  80       try:
  81         import cProfile
  82       except ImportError:
  83         # Old version of Python without cProfile.  Use hotshot instead.
  84         import hotshot
  85         prof = hotshot.Profile('cvs2svn.hotshot')
  86         prof.runcall(pass_manager.run, run_options)
  87         prof.close()
  88       else:
  89         # Recent version of Python (2.5+) with cProfile.
  90         def run_with_profiling():
  91           pass_manager.run(run_options)
  92         cProfile.runctx(
  93             'run_with_profiling()', globals(), locals(), 'cvs2svn.cProfile'
  94             )
  95     else:
  96       pass_manager.run(run_options)
  97   finally:
  98     try:
  99       os.rmdir(os.path.join(ctx.tmpdir, 'cvs2svn.lock'))
 100     except:
 101       pass
 102
 103     if erase_tmpdir:
 104       try:
 105         os.rmdir(ctx.tmpdir)
 106       except:
 107         pass
 108
 109
 110 def svn_main(progname, cmd_args):
 111   pass_manager = PassManager(passes)
 112   run_options = SVNRunOptions(progname, cmd_args, pass_manager)
 113   main(progname, run_options, pass_manager)
 114
 115
 116 def git_main(progname, cmd_args):
 117   pass_manager = PassManager(passes)
 118   run_options = GitRunOptions(progname, cmd_args, pass_manager)
 119   main(progname, run_options, pass_manager)
 120
 121
 122 def bzr_main(progname, cmd_args):
 123   pass_manager = PassManager(passes)
 124   run_options = BzrRunOptions(progname, cmd_args, pass_manager)
 125   main(progname, run_options, pass_manager)
 126
 127
 128 def hg_main(progname, cmd_args):
 129   # Import late so cvs2{svn,git} do not depend on being able to import
 130   # the Mercurial API.
 131   from cvs2svn_lib.hg_run_options import HgRunOptions
 132
 133   pass_manager = PassManager(passes)
 134   run_options = HgRunOptions(progname, cmd_args, pass_manager)
 135   main(progname, run_options, pass_manager)