/stdhome

To get this branch, use:
bzr branch http://bzr.ed.am/stdhome

« back to all changes in this revision

Viewing changes to lib/stdhome/program.py

  • Committer: Tim Marston
  • Date: 2014-04-18 14:32:26 UTC
  • Revision ID: tim@ed.am-20140418143226-d3ih1aafs6mg9ndw
made revert require --all to act on all files

Show diffs side-by-side

added added

removed removed

lib/stdhome/program.py
19
19
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
 
21
21
 
22
 
import os, sys, getopt
 
22
import os, sys, getopt, ConfigParser
23
23
import the
24
 
from config import Config
25
24
from vcs.vcs import Vcs
26
25
 
27
26
 
62
61
                print "  init          initialise a local copy of your repositories"
63
62
                print "  update        update files in your home directory"
64
63
                print "  resolve       try to finish an update (that had conflicts)"
65
 
                print "  add           add local files/changes to the repository"
 
64
#               print "  add           add local files/changes to the repository"
66
65
#               print "  remove        remove a local file from the repository"
67
66
                print "  status        list files that have changed locally"
68
67
                print "  diff          shows changes made to local files"
98
97
 
99
98
 
100
99
        def check_command( self, command ):
101
 
                """Check that the given command is valid and return the full name of the
102
 
                command.
103
 
 
104
 
                @param command the given command
105
 
                """
106
 
 
 
100
                """
 
101
                Check that the given command is valid and return the full name of the command.
 
102
        
 
103
                Arguments:
 
104
                - `command`: the given command
 
105
                """
107
106
                # commands
108
107
                if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert', 'status',
109
108
                         'diff', 'stage-add', 'stage-remove', 'stage-revert',
124
123
                return None
125
124
 
126
125
 
127
 
        def get_command_argument( self, args ):
128
 
                """Find the first program argument what isn't an option argument.
129
 
 
130
 
                @param args the program arguments
131
 
                """
132
 
 
133
 
                while args:
134
 
                        if args[ 0 ] == '--':
135
 
                                return args[ 1 ] if len( args ) > 1 else None
136
 
                        if args[ 0 ][ 0 : 1 ] != '-':
137
 
                                return args[ 0 ]
138
 
                        args = args[ 1 : ]
139
 
                return None
140
 
 
141
 
 
142
126
        def run( self ):
143
127
                # make an initial attempt to parse the command line, looking only for
144
 
                # --help, --version and --verbose, so that they have the chance to run
145
 
                # without a command being specified
 
128
                # --help and --version, so that they have the chance to run without a
 
129
                # command being specified
146
130
                try:
147
131
                        opts, args = getopt.gnu_getopt(
148
 
                                sys.argv[ 1: ], "v",
149
 
                                [ "verbose", "help", "version" ] )
 
132
                                sys.argv[ 1: ], "",
 
133
                                [ "help", "version" ] )
150
134
 
151
135
                        for opt, optarg in opts:
152
136
                                # we only show help if there are no non-option arguments (e.g.,
153
137
                                # a command) specified.  If a command has been specified it will
154
138
                                # have to be parsed and --help will be handled by it, instead)
155
 
                                if opt in [ '--verbose', '-v' ]:
156
 
                                        the.verbose += 1
157
 
                                elif opt == "--help" and len( args ) == 0:
 
139
                                if opt == "--help" and len( args ) == 0:
158
140
                                        self.print_help()
159
141
                                elif opt == "--version":
160
142
                                        self.print_version()
164
146
                        pass
165
147
 
166
148
                # read program configuration
167
 
                the.config = Config()
 
149
                self.read_config()
168
150
 
169
 
                # find the command by grabbing the first program argument that doesn't
170
 
                # look like an option (note: the first argument that doesn't look like
171
 
                # an option may actually be an argument to an option, so this is far
172
 
                # from perfect, but, provided the user is mindful, this is preferable to
173
 
                # forcing the user to always specify options after the command)
174
 
                the.command = self.get_command_argument( sys.argv[ 1: ] )
 
151
                # the first argument should be the command
 
152
                the.command = sys.argv[ 1 ] if len( sys.argv ) > 1 else None
175
153
                if the.command == None:
176
154
                        self.print_usage( "missing command" )
177
155
 
210
188
                        instance.run()
211
189
                except Vcs.VcsError as e:
212
190
                        message = e.msg.rstrip()
213
 
                        if the.verbose >= 1:
 
191
                        if the.verbose:
214
192
                                message += '\n\nOUTPUT:\n' + e.output.rstrip()
215
193
                        self.die( message )
216
194
                except self.FatalError as e:
217
195
                        self.die( e.msg )
218
196
 
219
197
 
 
198
        def read_config( self ):
 
199
                config = ConfigParser.SafeConfigParser( allow_no_value = True )
 
200
                config.read( the.config_file )
 
201
                if config.has_option( 'stdhome', 'home-dir' ):
 
202
                        the.home_dir = config.get( 'stdhome', 'home-dir' )
 
203
 
 
204
 
220
205
        class UsageError( Exception ):
221
206
 
222
207
                def __init__( self, error_message ):