/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-13 14:21:53 UTC
  • Revision ID: tim@ed.am-20160213142153-68g7pzq0au6bejep
fixed init command

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
 
                # --help and --version, so that they have the chance to run without a
130
 
                # command being specified
 
144
                # --help, --version and --verbose, so that they have the chance to run
 
145
                # without a command being specified
131
146
                try:
132
147
                        opts, args = getopt.gnu_getopt(
133
 
                                sys.argv[ 1: ], "",
134
 
                                [ "help", "version" ] )
 
148
                                sys.argv[ 1: ], "v",
 
149
                                [ "verbose", "help", "version" ] )
135
150
 
136
151
                        for opt, optarg in opts:
137
152
                                # we only show help if there are no non-option arguments (e.g.,
138
153
                                # a command) specified.  If a command has been specified it will
139
154
                                # have to be parsed and --help will be handled by it, instead)
140
 
                                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:
141
158
                                        self.print_help()
142
159
                                elif opt == "--version":
143
160
                                        self.print_version()
149
166
                # read program configuration
150
167
                the.config = Config()
151
168
 
152
 
                # the first argument should be the command
153
 
                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: ] )
154
175
                if the.command == None:
155
176
                        self.print_usage( "missing command" )
156
177