git.sur5r.net Git - u-boot/blob - tools/patman/settings.py

   1 # Copyright (c) 2011 The Chromium OS Authors.
   2 #
   3 # See file CREDITS for list of people who contributed to this
   4 # project.
   5 #
   6 # This program is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU General Public License as
   8 # published by the Free Software Foundation; either version 2 of
   9 # the License, or (at your option) any later version.
  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,
  19 # MA 02111-1307 USA
  20 #
  21
  22 import ConfigParser
  23 import os
  24 import re
  25
  26 import command
  27 import gitutil
  28
  29 def ReadGitAliases(fname):
  30     """Read a git alias file. This is in the form used by git:
  31
  32     alias uboot  u-boot@lists.denx.de
  33     alias wd     Wolfgang Denk <wd@denx.de>
  34
  35     Args:
  36         fname: Filename to read
  37     """
  38     try:
  39         fd = open(fname, 'r')
  40     except IOError:
  41         print "Warning: Cannot find alias file '%s'" % fname
  42         return
  43
  44     re_line = re.compile('alias\s+(\S+)\s+(.*)')
  45     for line in fd.readlines():
  46         line = line.strip()
  47         if not line or line[0] == '#':
  48             continue
  49
  50         m = re_line.match(line)
  51         if not m:
  52             print "Warning: Alias file line '%s' not understood" % line
  53             continue
  54
  55         list = alias.get(m.group(1), [])
  56         for item in m.group(2).split(','):
  57             item = item.strip()
  58             if item:
  59                 list.append(item)
  60         alias[m.group(1)] = list
  61
  62     fd.close()
  63
  64 def CreatePatmanConfigFile(config_fname):
  65     """Creates a config file under $(HOME)/.patman if it can't find one.
  66
  67     Args:
  68         config_fname: Default config filename i.e., $(HOME)/.patman
  69
  70     Returns:
  71         None
  72     """
  73     name = gitutil.GetDefaultUserName()
  74     if name == None:
  75         name = raw_input("Enter name: ")
  76
  77     email = gitutil.GetDefaultUserEmail()
  78
  79     if email == None:
  80         email = raw_input("Enter email: ")
  81
  82     try:
  83         f = open(config_fname, 'w')
  84     except IOError:
  85         print "Couldn't create patman config file\n"
  86         raise
  87
  88     print >>f, "[alias]\nme: %s <%s>" % (name, email)
  89     f.close();
  90
  91 def Setup(config_fname=''):
  92     """Set up the settings module by reading config files.
  93
  94     Args:
  95         config_fname:   Config filename to read ('' for default)
  96     """
  97     settings = ConfigParser.SafeConfigParser()
  98     if config_fname == '':
  99         config_fname = '%s/.patman' % os.getenv('HOME')
 100
 101     if not os.path.exists(config_fname):
 102         print "No config file found ~/.patman\nCreating one...\n"
 103         CreatePatmanConfigFile(config_fname)
 104
 105     settings.read(config_fname)
 106
 107     for name, value in settings.items('alias'):
 108         alias[name] = value.split(',')
 109
 110
 111 # These are the aliases we understand, indexed by alias. Each member is a list.
 112 alias = {}