/stdhome : revision 67

28

class InitCommand( Command ):

29

30

31

def __init__( self ):

32

self.quiet = False

33

self.vcs = the.config.default_vcs

34

35

31

36

def print_help( self ):

32

37

print "Usage: " + the.program.name + " init [URL] [--repo=REPO]"

33

38

print

41

46

print "hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO)."

42

47

print

43

48

print "Options:"

49

print " --quiet do not report changes to the home directory"

44

50

print " -r, --repo=REPO select the repo to check-out or create (defaults to 'home')"

45

51

print " -v, --verbose display information about what is being done"

52

print " --vcs=VCS select the version control system"

46

53

print " --help display help and exit"

47

54

exit( 0 )

48

55

50

57

def parse_command_line( self ):

51

58

opts, args = getopt.gnu_getopt(

52

59

sys.argv[ 1: ], "r:v",

53

[ "repo=", "verbose", "help" ] )

60

[ "quiet", "repo=", "vcs=", "verbose", "help" ] )

54

61

for opt, optarg in opts:

55

if opt in [ '--repo', '-r' ]:

56

if not re.match( '^[-a-zA-z0-9.]+$', optarg ):

62

if opt == "--quiet":

63

self.quiet = True

64

elif opt in [ '--repo', '-r' ]:

65

if not re.search( '^[-a-zA-z0-9.]+$', optarg ):

57

66

raise the.program.FatalError(

58

67

'invalid repository name: ' + optarg )

59

68

the.repo = optarg

69

elif opt == "--vcs":

70

self.vcs = optarg

71

elif opt in [ '--verbose', '-v' ]:

72

the.verbose += 1

60

73

elif opt == "--help":

61

74

self.print_help()

62

75

88

76

89

# set up repo and check it *doesn't* already exists

77

90

the.repo.check_dir_exists( False )

91

the.repo.set_vcs( self.vcs )

78

92

79

93

# ensure our top-level directory exists

80

94

if not os.path.exists( the.full_dir ):

84

98

if self.url:

85

99

86

100

# expand url if it's a simple hostname

87

if re.match( '^[0-9a-zA-z.]+$', self.url ):

88

self.url = 'bzr+ssh://%s/%s/%s' % \

89

( self.url, the.dir, the.repo.name )

101

if re.search( '^[0-9a-zA-z.]+$', self.url ):

102

self.url = the.repo.vcs.expand_repo_url( self.url )

90

103

91

104

# initialise deployment (with an empty repo)

92

105

deployment = Deployment()

96

109

if the.verbose >= 1: print "checking out %s" % the.repo.dir

97

110

try:

98

111

the.repo.vcs.checkout( self.url )

99

except Exception as e:

112

except Exception:

100

113

101

114

# attempt to clean-up repo dir

102

try:

103

shutil.rmtree( the.repo.full_dir )

104

except OSError:

105

pass

106

107

raise e

115

if os.path.exists( the.repo.full_dir ):

116

try:

117

shutil.rmtree( the.repo.full_dir )

118

except OSError:

119

pass

120

121

raise

122

123

message = ''

124

125

# check for deployment conclicts

126

conflicts = deployment.get_conflicts()

127

if conflicts:

128

message += 'deployment conflicts:\n %s' % \

129

'\n '.join( conflicts )

130

131

# stop if there are conflicts

132

if message:

133

raise the.program.FatalError(

134

'there were conflicts...\n' + message )

108

135

109

136

# perform deployment

110

try:

111

deployment.copy_out()

112

except deployment.Conflict as e:

113

raise the.program.FatalError( e.msg )

137

deployment.copy_out( self.quiet )

114

138

115

139

else:

116

140

118

142

if the.verbose >= 1: print 'initialising %s' % the.repo.dir

119

143

try:

120

144

the.repo.vcs.init()

121

except Exception as e:

145

except Exception:

122

146

123

147

# attempt to clean-up repo dir, and die

124

try:

125

shutil.rmtree( the.repo.full_dir )

126

except OSError:

127

pass

148

if os.path.exists( the.repo.full_dir ):

149

try:

150

shutil.rmtree( the.repo.full_dir )

151

except OSError:

152

pass

128

153

129

raise e

154

raise