/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 vcs.vcs import Vcs
24
25
 
25
26
 
26
27
class Program:
60
61
                print "  init          initialise a local copy of your repositories"
61
62
                print "  update        update files in your home directory"
62
63
                print "  resolve       try to finish an update (that had conflicts)"
63
 
                print "  add           add local files/changes to the repository"
64
 
                print "  remove        remove a local file from the repository"
 
64
#               print "  add           add local files/changes to the repository"
 
65
#               print "  remove        remove a local file from the repository"
65
66
                print "  status        list files that have changed locally"
66
67
                print "  diff          shows changes made to local files"
67
 
                print "  revert        undo changes made to a local file"
68
 
                print "  stage-add     stage local files/changes"
69
 
                print "  stage-remove  stage the removal of files"
70
 
                print "  stage-revert  revert staged changes"
71
 
                print "  stage-status  show status of staging area"
72
 
                print "  stage-diff    shows staged changes"
73
 
                print "  stage-commit  commit staged changes to repository"
 
68
                print "  revert        undo changes made to local files"
 
69
#               print "  stage-add     add (but don't commit) files/changes to local repository"
 
70
#               print "  stage-remove  delete *but don't comit) files from the local repository"
 
71
                print "  stage-revert  revert changes in the local repository"
 
72
#               print "  stage-status  show status of local repository"
 
73
#               print "  stage-diff    shows changes in local repository"
 
74
#               print "  stage-commit  commit changes in the local repository"
74
75
                print
75
76
                print "For help about a particular command (including the additional options that the"
76
77
                print "command accepts) try typing:"
81
82
        def print_version( self ):
82
83
                print "stdhome " + self.version
83
84
                print
84
 
                print "Copyright (C) 2013 Tim Marston"
 
85
                print "Copyright (C) 2013 to 2014 Tim Marston"
85
86
                print
86
87
                #      01234567890123456789012345678901234567890123456789012345678901234567890123456789
87
88
                print "This program is free software, and you may use, modify and redistribute it"
110
111
                        return command
111
112
 
112
113
                # resolve aliases
113
 
                elif command == 'up':
114
 
                        return 'update'
115
 
                elif command == 'rm':
116
 
                        return 'remove'
 
114
                alias = {
 
115
                        'up': 'update',
 
116
                        'rm': 'remove',
 
117
                        'st': 'status',
 
118
                        'co': 'init',
 
119
                }.get( command, False )
 
120
                if alias: return alias
117
121
 
118
122
                # invalid
119
 
                else:
120
 
                        return None
 
123
                return None
121
124
 
122
125
 
123
126
        def get_command_argument( self, args ):
158
161
                        # ignore errors -- we aren't parsing the command line properly yet
159
162
                        pass
160
163
 
 
164
                # read program configuration
 
165
                self.read_config()
 
166
 
161
167
                # find the first non-option argument (the command)
162
168
                the.command = self.get_command_argument( sys.argv[ 1: ] )
163
169
                if the.command == None:
169
175
                        self.print_usage( "bad command" )
170
176
 
171
177
                # calculate module and class name
 
178
                class_name = module_name = ''
172
179
                bits = the.command.split( '-' )
173
 
                class_name = 'Command'
174
 
                module_name = 'command'
175
180
                for bit in bits:
176
181
                        class_name += bit[ 0 ].upper() + bit[ 1 : ]
177
 
                        module_name += '_' + bit
 
182
                        if module_name: module_name += '_'
 
183
                        module_name += bit
 
184
                class_name += 'Command'
178
185
 
179
186
                # import module and instantiate the class
180
 
                module = __import__( 'stdhome.' + module_name,
 
187
                module = __import__( 'stdhome.command.' + module_name,
181
188
                                                         fromlist = [ class_name ] )
182
189
                instance = getattr( module, class_name )()
183
190
 
 
191
                # fully parse the command line, as per the command
 
192
                try:
 
193
                        instance.parse_command_line()
 
194
                except( getopt.GetoptError, self.UsageError ) as e:
 
195
                        self.print_usage( e.msg )
 
196
                except self.FatalError as e:
 
197
                        self.die( e.msg )
 
198
 
 
199
                # do late initialisation
 
200
                the.late_init()
 
201
 
184
202
                # run the command
185
203
                try:
186
 
                        instance.parse_command_line()
187
204
                        instance.run()
188
 
                except ( getopt.GetoptError, self.UsageError ) as e:
189
 
                        self.print_usage( e.msg )
190
 
                except self.FatalError as e:
 
205
                except Vcs.VcsError as e:
191
206
                        message = e.msg.rstrip()
192
 
                        if the.verbose and hasattr( e, 'output' ) and e.output:
 
207
                        if the.verbose:
193
208
                                message += '\n\nOUTPUT:\n' + e.output.rstrip()
194
209
                        self.die( message )
 
210
                except self.FatalError as e:
 
211
                        self.die( e.msg )
 
212
 
 
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' )
195
219
 
196
220
 
197
221
        class UsageError( Exception ):
202
226
 
203
227
        class FatalError( Exception ):
204
228
 
205
 
                def __init__( self, message, output = None ):
 
229
                def __init__( self, message ):
206
230
                        self.msg = message
207
 
                        self.output = output