stgit/templates.py

   1 """Template files look-up
   2 """
   3
   4 __copyright__ = """
   5 Copyright (C) 2006, Catalin Marinas <catalin.marinas@gmail.com>
   6
   7 This program is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License version 2 as
   9 published by the Free Software Foundation.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with this program; if not, write to the Free Software
  18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  19 """
  20
  21 import sys, os
  22
  23 from stgit import basedir
  24
  25
  26 def get_template(tfile):
  27     """Return the string in the template file passed as argument or
  28     None if the file wasn't found.
  29     """
  30     tmpl_list = [ os.path.join(basedir.get(), tfile),
  31                   os.path.join(os.path.expanduser('~'), '.stgit', 'templates',
  32                                tfile),
  33                   os.path.join(sys.prefix, 'share', 'stgit', 'templates',
  34                                tfile) ]
  35
  36     tmpl = None
  37     for t in tmpl_list:
  38         if os.path.isfile(t):
  39             tmpl = file(t).read()
  40             break
  41
  42     return tmpl