xref: /openbmc/u-boot/tools/patman/settings.py (revision 4e3349b6) 
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
22import ConfigParser
23import os
24import re
25
26import command
27import gitutil
28
29def 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
64def 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
91def 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.
112alias = {}
113