/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-09-09 19:13:33 UTC
  • Revision ID: tim@ed.am-20140909191333-0b7hrzonxce8mgy3
attempt to allow arguments before main command by treating the first thing that
doesn't look like an option as the command

Show diffs side-by-side

added added

removed removed

lib/stdhome/program.py
98
98
 
99
99
 
100
100
        def check_command( self, command ):
101
 
                """
102
 
                Check that the given command is valid and return the full name of the command.
103
 
        
104
 
                Arguments:
105
 
                - `command`: the given command
106
 
                """
 
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
 
107
107
                # commands
108
108
                if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert', 'status',
109
109
                         'diff', 'stage-add', 'stage-remove', 'stage-revert',
124
124
                return None
125
125
 
126
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
 
127
142
        def run( self ):
128
143
                # make an initial attempt to parse the command line, looking only for
129
144
                # --help and --version, so that they have the chance to run without a
149
164
                # read program configuration
150
165
                the.config = Config()
151
166
 
152
 
                # the first argument should be the command
153
 
                the.command = sys.argv[ 1 ] if len( sys.argv ) > 1 else None
 
167
                # find the command by grabbing the first program argument that doesn't
 
168
                # look like an option (note: the first argument that doesn't look like
 
169
                # an option may actually be an argument to an option, so this is far
 
170
                # from perfect, but, provided the user is mindful, this is preferable to
 
171
                # forcing the user to always specify options after the command)
 
172
                the.command = self.get_command_argument( sys.argv[ 1: ] )
154
173
                if the.command == None:
155
174
                        self.print_usage( "missing command" )
156
175