19
19
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
import os, sys, getopt, ConfigParser
24
from vcs.vcs import Vcs
29
30
def __init__( self, version ):
31
self.name = os.path.basename( sys.argv[ 0 ] )
30
32
self.version = version
31
self.program = os.path.basename( sys.argv[ 0 ] )
35
def die( self, error_message ):
36
prefix = self.name + ( ' ' + the.command if the.command else '' )
37
print >> sys.stderr, '%s: %s' % ( prefix, error_message )
34
41
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
command = ' ' + the.command if the.command else ''
43
self.die( error_message + \
44
"\nTry '%s%s --help' for more information." % \
45
( self.name, command ) )
40
48
def print_help( self ):
41
print "Usage: " + self.program + " COMMAND [OPTION]..."
49
print "Usage: " + self.name + " COMMAND [OPTION]..."
43
51
# 01234567890123456789012345678901234567890123456789012345678901234567890123456789
44
print "Manage your home directories, across multiple computers, similar to how you"
45
print "would software under version control."
52
print "Tool to manage a set of files in your home directory and distribute them across"
53
print "multiple computers, merging local changes (in the same way as you would manage"
54
print "source code under version control)."
47
56
print "Global options (for all commands):"
48
57
print " --help display help and exit"
52
61
print " init initialise a local copy of your repositories"
53
62
print " update update files in your home directory"
54
63
print " resolve try to finish an update (that had conflicts)"
55
print " add add local files/changes to the 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"
64
# print " add add local files/changes to the repository"
65
# print " remove remove a local file from the repository"
66
print " status list files that have changed locally"
67
print " diff shows changes made to local files"
68
# print " revert undo changes made to a local file"
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"
64
76
print "For help about a particular command (including the additional options that the"
65
77
print "command accepts) try typing:"
66
print " $ " + self.program + " COMMAND --help"
78
print " $ " + self.name + " COMMAND --help"
70
82
def print_version( self ):
71
83
print "stdhome " + self.version
73
print "Copyright (C) 2013 Tim Marston"
85
print "Copyright (C) 2013 to 2014 Tim Marston"
75
87
# 01234567890123456789012345678901234567890123456789012345678901234567890123456789
76
88
print "This program is free software, and you may use, modify and redistribute it"
92
104
- `command`: the given command
95
if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert',
96
'stage-add', 'stage-remove', 'stage-revert', 'stage-status',
97
'stage-commit' ].count( command ) == 1:
107
if [ 'init', 'update', 'resolve', 'add', 'remove', 'revert', 'status',
108
'diff', 'stage-add', 'stage-remove', 'stage-revert',
109
'stage-status', 'stage-diff', 'stage-commit'
110
].count( command ) == 1:
101
elif command == 'up':
103
elif command == 'rm':
119
}.get( command, False )
120
if alias: return alias
111
126
def get_command_argument( self, args ):
132
148
sys.argv[ 1: ], "",
133
149
[ "help", "version" ] )
135
for opt, optargs in opts:
151
for opt, optarg in opts:
136
152
# we only show help if there are no non-option arguments (e.g.,
137
153
# a command) specified. If a command has been specified it will
138
154
# have to be parsed and --help will be handled by it, instead)
142
158
self.print_version()
144
160
except getopt.GetoptError as e:
145
# ignore any errors -- we aren't parsing the command line properly
161
# ignore errors -- we aren't parsing the command line properly yet
164
# read program configuration
148
167
# find the first non-option argument (the command)
149
self.command = self.get_command_argument( sys.argv[ 1: ] )
150
if self.command == None:
168
the.command = self.get_command_argument( sys.argv[ 1: ] )
169
if the.command == None:
151
170
self.print_usage( "missing command" )
153
172
# check command is valid
154
self.command = self.check_command( self.command )
155
if self.command == None:
173
the.command = self.check_command( the.command )
174
if the.command == None:
156
175
self.print_usage( "bad command" )
158
# calculate class name
159
bits = self.command.split( '-' )
160
class_name = 'Command'
177
# calculate module and class name
178
class_name = module_name = ''
179
bits = the.command.split( '-' )
162
181
class_name += bit[ 0 ].upper() + bit[ 1 : ]
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()
182
if module_name: module_name += '_'
184
class_name += 'Command'
186
# import module and instantiate the class
187
module = __import__( 'stdhome.command.' + module_name,
188
fromlist = [ class_name ] )
189
instance = getattr( module, class_name )()
191
# fully parse the command line, as per the command
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:
199
# do late initialisation
205
except Vcs.VcsError as e:
206
message = e.msg.rstrip()
208
message += '\n\nOUTPUT:\n' + e.output.rstrip()
210
except self.FatalError as e:
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' )
221
class UsageError( Exception ):
223
def __init__( self, error_message ):
224
self.msg = error_message
227
class FatalError( Exception ):
229
def __init__( self, message ):