/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-05 22:30:23 UTC
  • Revision ID: tim@ed.am-20140405223023-3co87mbvnxwk1l65
added revert command

Show diffs side-by-side

added added

removed removed

Lines of Context:
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',
125
124
 
126
125
 
127
126
        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
127
                """
 
128
                Find the first program argument what isn't an option argument.
132
129
 
 
130
        Arguments:
 
131
        - `args`: the program arguments
 
132
        """
133
133
                while args:
134
134
                        if args[ 0 ] == '--':
135
135
                                return args[ 1 ] if len( args ) > 1 else None
141
141
 
142
142
        def run( self ):
143
143
                # 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
 
144
                # --help and --version, so that they have the chance to run without a
 
145
                # command being specified
146
146
                try:
147
147
                        opts, args = getopt.gnu_getopt(
148
 
                                sys.argv[ 1: ], "v",
149
 
                                [ "verbose", "help", "version" ] )
 
148
                                sys.argv[ 1: ], "",
 
149
                                [ "help", "version" ] )
150
150
 
151
151
                        for opt, optarg in opts:
152
152
                                # we only show help if there are no non-option arguments (e.g.,
153
153
                                # a command) specified.  If a command has been specified it will
154
154
                                # 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:
 
155
                                if opt == "--help" and len( args ) == 0:
158
156
                                        self.print_help()
159
157
                                elif opt == "--version":
160
158
                                        self.print_version()
164
162
                        pass
165
163
 
166
164
                # read program configuration
167
 
                the.config = Config()
 
165
                self.read_config()
168
166
 
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)
 
167
                # find the first non-option argument (the command)
174
168
                the.command = self.get_command_argument( sys.argv[ 1: ] )
175
169
                if the.command == None:
176
170
                        self.print_usage( "missing command" )
210
204
                        instance.run()
211
205
                except Vcs.VcsError as e:
212
206
                        message = e.msg.rstrip()
213
 
                        if the.verbose >= 1:
 
207
                        if the.verbose:
214
208
                                message += '\n\nOUTPUT:\n' + e.output.rstrip()
215
209
                        self.die( message )
216
210
                except self.FatalError as e:
217
211
                        self.die( e.msg )
218
212
 
219
213
 
 
214
        def read_config( self ):
 
215
                config = ConfigParser.SafeConfigParser( allow_no_value = True )
 
216
                config.read( the.config_file )
 
217
                if config.has_option( 'stdhome', 'home-dir' ):
 
218
                        the.home_dir = config.get( 'stdhome', 'home-dir' )
 
219
 
 
220
 
220
221
        class UsageError( Exception ):
221
222
 
222
223
                def __init__( self, error_message ):