/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/command/init.py

  • Committer: Tim Marston
  • Date: 2014-07-17 18:17:18 UTC
  • Revision ID: tim@ed.am-20140717181718-gxi5lpi8qszsigxf
switch to using bzr+ssh, rather than sftp (to work around broken pimiko)

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/stdhome/command/init.py
1
 
# command_init.py
 
1
# init.py
2
2
#
3
3
# Copyright (C) 2013 Tim Marston <tim@edm.am>
4
4
#
19
19
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
 
21
21
 
22
 
import sys, os
23
 
import getopt
24
 
 
25
 
 
26
 
class CommandInit:
 
22
import sys, os, re, getopt, shutil, subprocess
 
23
from command import Command
 
24
import stdhome.the as the
 
25
from stdhome.deployment import Deployment
 
26
 
 
27
 
 
28
class InitCommand( Command ):
 
29
 
27
30
 
28
31
        def print_help( self ):
29
 
                print "Usage: " + self.program + " init [URL] [--repo=REPO]"
 
32
                print "Usage: " + the.program.name + " init [URL] [--repo=REPO]"
30
33
                print
31
34
                #      01234567890123456789012345678901234567890123456789012345678901234567890123456789
32
35
                print "Initialise a local repository."
38
41
                print "hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO)."
39
42
                print
40
43
                print "Options:"
41
 
                print "     --repo     select the repo to check-out or create (defaults to 'home')"
42
 
                print "     --help     display help and exit"
 
44
                print "  -r, --repo=REPO  select the repo to check-out or create (defaults to 'home')"
 
45
                print "  -v, --verbose    display information about what is being done"
 
46
                print "      --help       display help and exit"
43
47
                exit( 0 )
44
48
 
45
49
 
46
 
        def run( self, stdhome ):
47
 
                # parse command line
48
 
                try:
49
 
                        opts, args = getopt.gnu_getopt(
50
 
                                sys.argv[ 1: ], "",
51
 
                                [ "repo:", "help" ] )
52
 
                except getopt.GetoptError as e:
53
 
                        stdhome.print_usage( e )
54
 
                for opt, optargs in opts:
55
 
                        if opt == "--help":
 
50
        def parse_command_line( self ):
 
51
                opts, args = getopt.gnu_getopt(
 
52
                        sys.argv[ 1: ], "r:v",
 
53
                        [ "repo=", "verbose", "help" ] )
 
54
                for opt, optarg in opts:
 
55
                        if opt in [ '--repo', '-r' ]:
 
56
                                if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
 
57
                                        raise the.program.FatalError(
 
58
                                                'invalid repository name: ' + optarg )
 
59
                                the.repo = optarg
 
60
                        elif opt in [ '--verbose', '-v' ]:
 
61
                                the.verbose += 1
 
62
                        elif opt == "--help":
56
63
                                self.print_help()
57
 
                        elif aopt == "--repo":
58
 
                                self.repo = optarg
59
 
 
60
 
                print "init"
 
64
                
 
65
                # discard first argument (the command)
 
66
                args.pop( 0 )
 
67
 
 
68
                # URL argument
 
69
                self.url = args[ 0 ].strip() if len( args ) else None
 
70
 
 
71
                # check remaining arguments
 
72
                if len( args ) > 1:
 
73
                        raise the.program.UsageError( 'too many arguments' )
 
74
 
 
75
 
 
76
        def run( self ):
 
77
 
 
78
                # set up repo and check it *doesn't* already exists
 
79
                the.repo.check_dir_exists( False )
 
80
 
 
81
                # ensure our top-level directory exists
 
82
                if not os.path.exists( the.full_dir ):
 
83
                        os.mkdir( the.full_dir )
 
84
 
 
85
                # checkout a remote repo, or create an empty local one?
 
86
                if self.url:
 
87
 
 
88
                        # expand url if it's a simple hostname
 
89
                        if re.match( '^[0-9a-zA-z.]+$', self.url ):
 
90
                                self.url = 'bzr+ssh://%s/%s/%s' % \
 
91
                                                   ( self.url, the.dir, the.repo.name )
 
92
 
 
93
                        # initialise deployment (with an empty repo)
 
94
                        deployment = Deployment()
 
95
                        deployment.copy_in()
 
96
 
 
97
                        # perform bzr checkout
 
98
                        if the.verbose >= 1: print "checking out %s" % the.repo.dir
 
99
                        try:
 
100
                                the.repo.vcs.checkout( self.url )
 
101
                        except Exception as e:
 
102
 
 
103
                                # attempt to clean-up repo dir
 
104
                                try:
 
105
                                        shutil.rmtree( the.repo.full_dir )
 
106
                                except OSError:
 
107
                                        pass
 
108
 
 
109
                                raise e
 
110
 
 
111
                        # perform deployment
 
112
                        try:
 
113
                                deployment.copy_out()
 
114
                        except deployment.Conflict as e:
 
115
                                raise the.program.FatalError( e.msg )
 
116
 
 
117
                else:
 
118
 
 
119
                        # perform bzr init
 
120
                        if the.verbose >= 1: print 'initialising %s' % the.repo.dir
 
121
                        try:
 
122
                                the.repo.vcs.init()
 
123
                        except Exception as e:
 
124
 
 
125
                                # attempt to clean-up repo dir, and die
 
126
                                try:
 
127
                                        shutil.rmtree( the.repo.full_dir )
 
128
                                except OSError:
 
129
                                        pass
 
130
 
 
131
                                raise e