/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-02-12 21:51:49 UTC
  • Revision ID: tim@ed.am-20140212215149-msaxl7vo98il5i4a
added more commands

Show diffs side-by-side

added added

removed removed

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