42
40
def print_usage( self, error_message ):
43
command = ' ' + the.command if the.command else ''
41
suffix = ' ' + the.command if the.command else ''
44
42
self.die( error_message + \
45
43
"\nTry '%s%s --help' for more information." % \
46
( self.name, command ) )
44
( self.name, suffix ) )
49
47
def print_help( self ):
50
48
print "Usage: " + self.name + " COMMAND [OPTION]..."
52
50
# 01234567890123456789012345678901234567890123456789012345678901234567890123456789
53
print "Tool to manage a set of files in your home directory and distribute them across"
54
print "multiple computers, merging local changes (in the same way as you would manage"
55
print "source code under version control)."
51
print "Manage your home directories, across multiple computers, similar to how you"
52
print "would software under version control."
57
54
print "Global options (for all commands):"
58
55
print " --help display help and exit"
63
60
print " update update files in your home directory"
64
61
print " resolve try to finish an update (that had conflicts)"
65
62
print " add add local files/changes to the repository"
66
# print " remove remove a local file from the repository"
67
print " status list files that have changed locally"
68
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"
63
print " remove remove a local file from the repository"
64
print " revert undo changes made to a local file"
65
print " stage-add stage local files/changes"
66
print " stage-remove stage the removal of files"
67
print " stage-revert revert staged changes"
68
print " stage-status show status of staging area"
69
print " stage-commit commit staged changes to repository"
77
71
print "For help about a particular command (including the additional options that the"
78
72
print "command accepts) try typing:"
83
77
def print_version( self ):
84
78
print "stdhome " + self.version
86
print "Copyright (C) 2013 to 2014 Tim Marston"
80
print "Copyright (C) 2013 Tim Marston"
88
82
# 01234567890123456789012345678901234567890123456789012345678901234567890123456789
89
83
print "This program is free software, and you may use, modify and redistribute it"
100
94
def check_command( self, command ):
101
"""Check that the given command is valid and return the full name of the
104
@param command the given command
96
Check that the given command is valid and return the full name of the command.
99
- `command`: the given command
108
if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert', 'status',
109
'diff', 'stage-add', 'stage-remove', 'stage-revert',
110
'stage-status', 'stage-diff', 'stage-commit'
111
].count( command ) == 1:
102
if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert',
103
'stage-add', 'stage-remove', 'stage-revert', 'stage-status',
104
'stage-commit' ].count( command ) == 1:
114
107
# resolve aliases
120
}.get( command, False )
121
if alias: return alias
108
elif command == 'up':
110
elif command == 'rm':
127
118
def get_command_argument( self, args ):
128
"""Find the first program argument what isn't an option argument.
130
@param args the program arguments
120
Find the first program argument what isn't an option argument.
123
- `args`: the program arguments
134
126
if args[ 0 ] == '--':
135
127
return args[ 1 ] if len( args ) > 1 else None
143
135
# 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
136
# --help and --version, so that they have the chance to run without a
137
# command being specified
147
139
opts, args = getopt.gnu_getopt(
149
[ "verbose", "help", "version" ] )
141
[ "help", "version" ] )
151
143
for opt, optarg in opts:
152
144
# we only show help if there are no non-option arguments (e.g.,
153
145
# a command) specified. If a command has been specified it will
154
146
# have to be parsed and --help will be handled by it, instead)
155
if opt in [ '--verbose', '-v' ]:
157
elif opt == "--help" and len( args ) == 0:
147
if opt == "--help" and len( args ) == 0:
158
148
self.print_help()
159
149
elif opt == "--version":
160
150
self.print_version()
163
153
# ignore errors -- we aren't parsing the command line properly yet
166
# read program configuration
167
the.config = Config()
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)
156
# find the first non-option argument (the command)
174
157
the.command = self.get_command_argument( sys.argv[ 1: ] )
175
158
if the.command == None:
176
159
self.print_usage( "missing command" )
180
163
if the.command == None:
181
164
self.print_usage( "bad command" )
183
# calculate module and class name
184
class_name = module_name = ''
166
# calculate class name
185
167
bits = the.command.split( '-' )
168
class_name = 'Command'
187
170
class_name += bit[ 0 ].upper() + bit[ 1 : ]
188
if module_name: module_name += '_'
190
class_name += 'Command'
192
# import module and instantiate the class
193
module = __import__( 'stdhome.command.' + module_name,
172
# instantiate the command class
173
module = __import__( 'stdhome.command_' + the.command,
194
174
fromlist = [ class_name ] )
195
175
instance = getattr( module, class_name )()
197
# fully parse the command line, as per the command
199
179
instance.parse_command_line()
200
except( getopt.GetoptError, self.UsageError ) as e:
181
except ( getopt.GetoptError, self.UsageError ) as e:
201
182
self.print_usage( e.msg )
202
183
except self.FatalError as e:
205
# do late initialisation
211
except Vcs.VcsError as e:
212
message = e.msg.rstrip()
214
message += '\n\nOUTPUT:\n' + e.output.rstrip()
185
if the.verbose and e.output:
186
message += '\n\nOUTPUT:\n' + e.output
215
187
self.die( message )
216
except self.FatalError as e:
220
190
class UsageError( Exception ):
226
196
class FatalError( Exception ):
228
def __init__( self, message ):
198
def __init__( self, message, output = None ):
229
199
self.msg = message