/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: 2016-02-23 19:35:21 UTC
  • Revision ID: tim@ed.am-20160223193521-2vgtxbfos50rrpku
renamed version -> VERSION

Show diffs side-by-side

added added

removed removed

Lines of Context:
19
19
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
 
21
21
 
22
 
import os, sys, getopt, ConfigParser
 
22
import os, sys, getopt
23
23
import the
 
24
from config import Config
24
25
from vcs.vcs import Vcs
25
26
 
26
27
 
61
62
                print "  init          initialise a local copy of your repositories"
62
63
                print "  update        update files in your home directory"
63
64
                print "  resolve       try to finish an update (that had conflicts)"
64
 
#               print "  add           add local files/changes to the repository"
 
65
                print "  add           add local files/changes to the repository"
65
66
#               print "  remove        remove a local file from the repository"
66
67
                print "  status        list files that have changed locally"
67
68
                print "  diff          shows changes made to local files"
97
98
 
98
99
 
99
100
        def check_command( self, command ):
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
 
                """
 
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
 
106
107
                # commands
107
108
                if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert', 'status',
108
109
                         'diff', 'stage-add', 'stage-remove', 'stage-revert',
123
124
                return None
124
125
 
125
126
 
 
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
 
126
142
        def run( self ):
127
143
                # make an initial attempt to parse the command line, looking only for
128
 
                # --help and --version, so that they have the chance to run without a
129
 
                # command being specified
 
144
                # --help, --version and --verbose, so that they have the chance to run
 
145
                # without a command being specified
130
146
                try:
131
147
                        opts, args = getopt.gnu_getopt(
132
 
                                sys.argv[ 1: ], "",
133
 
                                [ "help", "version" ] )
 
148
                                sys.argv[ 1: ], "v",
 
149
                                [ "verbose", "help", "version" ] )
134
150
 
135
151
                        for opt, optarg in opts:
136
152
                                # we only show help if there are no non-option arguments (e.g.,
137
153
                                # a command) specified.  If a command has been specified it will
138
154
                                # have to be parsed and --help will be handled by it, instead)
139
 
                                if opt == "--help" and len( args ) == 0:
 
155
                                if opt in [ '--verbose', '-v' ]:
 
156
                                        the.verbose += 1
 
157
                                elif opt == "--help" and len( args ) == 0:
140
158
                                        self.print_help()
141
159
                                elif opt == "--version":
142
160
                                        self.print_version()
146
164
                        pass
147
165
 
148
166
                # read program configuration
149
 
                self.read_config()
 
167
                the.config = Config()
150
168
 
151
 
                # the first argument should be the command
152
 
                the.command = sys.argv[ 1 ] if len( sys.argv ) > 1 else None
 
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: ] )
153
175
                if the.command == None:
154
176
                        self.print_usage( "missing command" )
155
177
 
188
210
                        instance.run()
189
211
                except Vcs.VcsError as e:
190
212
                        message = e.msg.rstrip()
191
 
                        if the.verbose:
 
213
                        if the.verbose >= 1:
192
214
                                message += '\n\nOUTPUT:\n' + e.output.rstrip()
193
215
                        self.die( message )
194
216
                except self.FatalError as e:
195
217
                        self.die( e.msg )
196
218
 
197
219
 
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
 
 
205
220
        class UsageError( Exception ):
206
221
 
207
222
                def __init__( self, error_message ):