19
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
import os, sys, getopt
24
from config import Config
25
from vcs.vcs import Vcs
31
29
def __init__( self, version ):
32
self.name = os.path.basename( sys.argv[ 0 ] )
33
30
self.version = version
36
def die( self, error_message ):
37
prefix = self.name + ( ' ' + the.command if the.command else '' )
38
print >> sys.stderr, '%s: %s' % ( prefix, error_message )
31
self.program = os.path.basename( sys.argv[ 0 ] )
34
def print_usage( self, error_message ):
35
print >> sys.stderr, self.program + ": " + error_message
36
print >> sys.stderr, "Try '" + self.program + " --help' for more information."
42
def print_usage( self, error_message ):
43
command = ' ' + the.command if the.command else ''
44
self.die( error_message + \
45
"\nTry '%s%s --help' for more information." % \
46
( self.name, command ) )
49
40
def print_help( self ):
50
print "Usage: " + self.name + " COMMAND [OPTION]..."
41
print "Usage: " + self.program + " COMMAND [OPTION]..."
52
43
# 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)."
44
print "Manage your home directories, across multiple computers, similar to how you"
45
print "would software under version control."
57
47
print "Global options (for all commands):"
58
48
print " --help display help and exit"
63
53
print " update update files in your home directory"
64
54
print " resolve try to finish an update (that had conflicts)"
65
55
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"
56
print " remove remove a local file from the repository"
57
print " revert undo changes made to a local file"
58
print " stage-add stage local files/changes"
59
print " stage-remove stage the removal of files"
60
print " stage-revert revert staged changes"
61
print " stage-status show status of staging area"
62
print " stage-commit commit staged changes to repository"
77
64
print "For help about a particular command (including the additional options that the"
78
65
print "command accepts) try typing:"
79
print " $ " + self.name + " COMMAND --help"
66
print " $ " + self.program + " COMMAND --help"
83
70
def print_version( self ):
84
71
print "stdhome " + self.version
86
print "Copyright (C) 2013 to 2014 Tim Marston"
73
print "Copyright (C) 2013 Tim Marston"
88
75
# 01234567890123456789012345678901234567890123456789012345678901234567890123456789
89
76
print "This program is free software, and you may use, modify and redistribute it"
100
87
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
89
Check that the given command is valid and return the full name of the command.
92
- `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:
95
if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert',
96
'stage-add', 'stage-remove', 'stage-revert', 'stage-status',
97
'stage-commit' ].count( command ) == 1:
120
}.get( command, False )
121
if alias: return alias
101
elif command == 'up':
103
elif command == 'rm':
127
111
def get_command_argument( self, args ):
128
"""Find the first program argument what isn't an option argument.
130
@param args the program arguments
113
Find the first program argument what isn't an option argument.
116
- `args`: the program arguments
134
119
if args[ 0 ] == '--':
135
120
return args[ 1 ] if len( args ) > 1 else None
138
123
args = args[ 1 : ]
143
127
# make an initial attempt to parse the command line, looking only for
144
# --help and --version so that they have the chance to run without a
128
# --help and --version, so that they have the chance to run without a
145
129
# command being specified
147
131
opts, args = getopt.gnu_getopt(
148
132
sys.argv[ 1: ], "",
149
133
[ "help", "version" ] )
151
for opt, optarg in opts:
135
for opt, optargs in opts:
152
136
# we only show help if there are no non-option arguments (e.g.,
153
137
# a command) specified. If a command has been specified it will
154
138
# have to be parsed and --help will be handled by it, instead)
158
142
self.print_version()
160
144
except getopt.GetoptError as e:
161
# ignore errors -- we aren't parsing the command line properly yet
145
# ignore any errors -- we aren't parsing the command line properly
164
# read program configuration
165
the.config = Config()
167
# find the command by grabbing the first program argument that doesn't
168
# look like an option (note: the first argument that doesn't look like
169
# an option may actually be an argument to an option, so this is far
170
# from perfect, but, provided the user is mindful, this is preferable to
171
# forcing the user to always specify options after the command)
172
the.command = self.get_command_argument( sys.argv[ 1: ] )
173
if the.command == None:
148
# find the first non-option argument (the command)
149
self.command = self.get_command_argument( sys.argv[ 1: ] )
150
if self.command == None:
174
151
self.print_usage( "missing command" )
176
153
# check command is valid
177
the.command = self.check_command( the.command )
178
if the.command == None:
154
self.command = self.check_command( self.command )
155
if self.command == None:
179
156
self.print_usage( "bad command" )
181
# calculate module and class name
182
class_name = module_name = ''
183
bits = the.command.split( '-' )
158
# calculate class name
159
bits = self.command.split( '-' )
160
class_name = 'Command'
185
162
class_name += bit[ 0 ].upper() + bit[ 1 : ]
186
if module_name: module_name += '_'
188
class_name += 'Command'
190
# import module and instantiate the class
191
module = __import__( 'stdhome.command.' + module_name,
192
fromlist = [ class_name ] )
193
instance = getattr( module, class_name )()
195
# fully parse the command line, as per the command
197
instance.parse_command_line()
198
except( getopt.GetoptError, self.UsageError ) as e:
199
self.print_usage( e.msg )
200
except self.FatalError as e:
203
# do late initialisation
209
except Vcs.VcsError as e:
210
message = e.msg.rstrip()
212
message += '\n\nOUTPUT:\n' + e.output.rstrip()
214
except self.FatalError as e:
218
class UsageError( Exception ):
220
def __init__( self, error_message ):
221
self.msg = error_message
224
class FatalError( Exception ):
226
def __init__( self, message ):
164
# instantiate and run the command class
165
module = __import__( 'stdhome.command_' + self.command )
166
the_class = getattr( module, class_name )
167
instance = the_class()