/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-03-08 00:40:42 UTC
  • Revision ID: tim@ed.am-20140308004042-j5rco14bjp3teegj
updated .bzrignore

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 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"
66
 
#               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"
67
66
                print "  status        list files that have changed locally"
68
67
                print "  diff          shows changes made to local files"
69
 
                print "  revert        undo changes made to local files"
70
 
#               print "  stage-add     add (but don't commit) files/changes to local repository"
71
 
#               print "  stage-remove  delete *but don't comit) files from the local repository"
72
 
                print "  stage-revert  revert changes in the local repository"
73
 
#               print "  stage-status  show status of local repository"
74
 
#               print "  stage-diff    shows changes in local repository"
75
 
#               print "  stage-commit  commit changes in the local repository"
 
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"
 
71
                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"
76
75
                print
77
76
                print "For help about a particular command (including the additional options that the"
78
77
                print "command accepts) try typing:"
83
82
        def print_version( self ):
84
83
                print "stdhome " + self.version
85
84
                print
86
 
                print "Copyright (C) 2013 to 2014 Tim Marston"
 
85
                print "Copyright (C) 2013 Tim Marston"
87
86
                print
88
87
                #      01234567890123456789012345678901234567890123456789012345678901234567890123456789
89
88
                print "This program is free software, and you may use, modify and redistribute it"
112
111
                        return command
113
112
 
114
113
                # resolve aliases
115
 
                alias = {
116
 
                        'up': 'update',
117
 
                        'rm': 'remove',
118
 
                        'st': 'status',
119
 
                        'co': 'init',
120
 
                }.get( command, False )
121
 
                if alias: return alias
 
114
                elif command == 'up':
 
115
                        return 'update'
 
116
                elif command == 'rm':
 
117
                        return 'remove'
122
118
 
123
119
                # invalid
 
120
                else:
 
121
                        return None
 
122
 
 
123
 
 
124
        def get_command_argument( self, args ):
 
125
                """
 
126
                Find the first program argument what isn't an option argument.
 
127
 
 
128
        Arguments:
 
129
        - `args`: the program arguments
 
130
        """
 
131
                while args:
 
132
                        if args[ 0 ] == '--':
 
133
                                return args[ 1 ] if len( args ) > 1 else None
 
134
                        if args[ 0 ][ 0 : 1 ] != '-':
 
135
                                return args[ 0 ]
 
136
                        args = args[ 1 : ]
124
137
                return None
125
138
 
126
139
 
146
159
                        # ignore errors -- we aren't parsing the command line properly yet
147
160
                        pass
148
161
 
149
 
                # read program configuration
150
 
                the.config = Config()
151
 
 
152
 
                # the first argument should be the command
153
 
                the.command = sys.argv[ 1 ] if len( sys.argv ) > 1 else None
 
162
                # find the first non-option argument (the command)
 
163
                the.command = self.get_command_argument( sys.argv[ 1: ] )
154
164
                if the.command == None:
155
165
                        self.print_usage( "missing command" )
156
166
 
160
170
                        self.print_usage( "bad command" )
161
171
 
162
172
                # calculate module and class name
163
 
                class_name = module_name = ''
164
173
                bits = the.command.split( '-' )
 
174
                class_name = 'Command'
 
175
                module_name = 'command'
165
176
                for bit in bits:
166
177
                        class_name += bit[ 0 ].upper() + bit[ 1 : ]
167
 
                        if module_name: module_name += '_'
168
 
                        module_name += bit
169
 
                class_name += 'Command'
 
178
                        module_name += '_' + bit
170
179
 
171
180
                # import module and instantiate the class
172
 
                module = __import__( 'stdhome.command.' + module_name,
 
181
                module = __import__( 'stdhome.' + module_name,
173
182
                                                         fromlist = [ class_name ] )
174
183
                instance = getattr( module, class_name )()
175
184
 
176
 
                # fully parse the command line, as per the command
 
185
                # run the command
177
186
                try:
178
187
                        instance.parse_command_line()
 
188
                        instance.run()
179
189
                except( getopt.GetoptError, self.UsageError ) as e:
180
190
                        self.print_usage( e.msg )
181
 
                except self.FatalError as e:
182
 
                        self.die( e.msg )
183
 
 
184
 
                # do late initialisation
185
 
                the.late_init()
186
 
 
187
 
                # run the command
188
 
                try:
189
 
                        instance.run()
190
191
                except Vcs.VcsError as e:
191
192
                        message = e.msg.rstrip()
192
 
                        if the.verbose >= 1:
 
193
                        if the.verbose and hasattr( e, 'output' ) and e.output:
193
194
                                message += '\n\nOUTPUT:\n' + e.output.rstrip()
194
195
                        self.die( message )
195
196
                except self.FatalError as e: