chiark / gitweb /
Infrastructure for current directory handling
[stgit] / stgit / commands / status.py
1
2 __copyright__ = """
3 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License version 2 as
7 published by the Free Software Foundation.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 """
18
19 import sys, os
20 from optparse import OptionParser, make_option
21
22 from stgit.commands.common import *
23 from stgit.utils import *
24 from stgit import stack, git
25
26
27 help = 'show the tree status'
28 usage = """%prog [options] [<files...>]
29
30 Show the status of the whole working copy or the given files. The
31 command also shows the files in the current directory which are not
32 under revision control. The files are prefixed as follows:
33
34   M - locally modified
35   N - newly added to the repository
36   D - deleted from the repository
37   C - conflict
38   ? - unknown
39
40 A 'refresh' command clears the status of the modified, new and deleted
41 files."""
42
43 directory = DirectoryHasRepository()
44 options = [make_option('-m', '--modified',
45                        help = 'show modified files only',
46                        action = 'store_true'),
47            make_option('-n', '--new',
48                        help = 'show new files only',
49                        action = 'store_true'),
50            make_option('-d', '--deleted',
51                        help = 'show deleted files only',
52                        action = 'store_true'),
53            make_option('-c', '--conflict',
54                        help = 'show conflict files only',
55                        action = 'store_true'),
56            make_option('-u', '--unknown',
57                        help = 'show unknown files only',
58                        action = 'store_true'),
59            make_option('-x', '--noexclude',
60                        help = 'do not exclude any files from listing',
61                        action = 'store_true'),
62            make_option('-O', '--diff-opts',
63                        help = 'options to pass to git-diff'),
64            make_option('--reset',
65                        help = 'reset the current tree changes',
66                        action = 'store_true')]
67
68
69 def status(files = None, modified = False, new = False, deleted = False,
70            conflict = False, unknown = False, noexclude = False,
71            diff_flags = []):
72     """Show the tree status
73     """
74     cache_files = git.tree_status(files,
75                                   unknown = (files == None),
76                                   noexclude = noexclude,
77                                   diff_flags = diff_flags)
78     filtered = (modified or new or deleted or conflict or unknown)
79
80     if filtered:
81         filestat = []
82         if modified:
83             filestat.append('M')
84         if new:
85             filestat.append('A')
86             filestat.append('N')
87         if deleted:
88             filestat.append('D')
89         if conflict:
90             filestat.append('C')
91         if unknown:
92             filestat.append('?')
93         cache_files = [x for x in cache_files if x[0] in filestat]
94
95     output = []
96     for st, fn in cache_files:
97         assert files == None or fn in files
98         if filtered:
99             output.append(fn)
100         else:
101             output.append('%s %s' % (st, fn))
102     for o in sorted(output):
103         out.stdout(o)
104
105 def func(parser, options, args):
106     """Show the tree status
107     """
108     if options.reset:
109         if args:
110             for f in args:
111                 resolved(f)
112             git.reset(args)
113         else:
114             resolved_all()
115             git.reset()
116     else:
117         if options.diff_opts:
118             diff_flags = options.diff_opts.split()
119         else:
120             diff_flags = []
121
122         # No args means all files
123         if not args:
124             args = None
125         status(args, options.modified, options.new, options.deleted,
126                options.conflict, options.unknown, options.noexclude,
127                diff_flags = diff_flags)