/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-01-05 11:51:35 UTC
  • Revision ID: tim@ed.am-20140105115135-6ses87ggwyjrxzfj
added global objects (the.repo, the.program), deployment object and implemented
init command

Show diffs side-by-side

added added

removed removed

lib/stdhome/command_init.py
1
 
# init.py
 
1
# command_init.py
2
2
#
3
3
# Copyright (C) 2013 Tim Marston <tim@edm.am>
4
4
#
20
20
 
21
21
 
22
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 ):
 
23
import the, deployment
 
24
from subprocess import Popen
 
25
 
 
26
 
 
27
class CommandInit:
 
28
 
 
29
 
 
30
        def __init__( self ):
 
31
                self.repo = None
29
32
 
30
33
 
31
34
        def print_help( self ):
42
45
                print
43
46
                print "Options:"
44
47
                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
48
                print "      --help       display help and exit"
47
49
                exit( 0 )
48
50
 
52
54
                        sys.argv[ 1: ], "r:v",
53
55
                        [ "repo=", "verbose", "help" ] )
54
56
                for opt, optarg in opts:
55
 
                        if opt in [ '--repo', '-r' ]:
 
57
                        if [ '--repo', '-r' ].count( opt ):
56
58
                                if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
57
 
                                        raise the.program.FatalError(
 
59
                                        raise the.FatalError(
58
60
                                                'invalid repository name: ' + optarg )
59
 
                                the.repo = optarg
 
61
                                self.repo = optarg
 
62
                        elif [ '--verbose', '-v' ].count( opt ):
 
63
                                the.verbose = True
60
64
                        elif opt == "--help":
61
65
                                self.print_help()
62
 
 
 
66
                
63
67
                # discard first argument (the command)
64
68
                args.pop( 0 )
65
69
 
66
70
                # URL argument
67
71
                self.url = args[ 0 ].strip() if len( args ) else None
68
72
 
69
 
                # check remaining arguments
 
73
                # remaining arguments
70
74
                if len( args ) > 1:
71
75
                        raise the.program.UsageError( 'too many arguments' )
72
76
 
73
77
 
74
78
        def run( self ):
 
79
                the.set_repo( self.repo )
75
80
 
76
 
                # set up repo and check it *doesn't* already exists
 
81
                # repo dir must not already exist
77
82
                the.repo.check_dir_exists( False )
78
83
 
79
84
                # ensure our top-level directory exists
80
 
                if not os.path.exists( the.full_dir ):
81
 
                        os.mkdir( the.full_dir )
 
85
                if not os.path.exists( the.expanded_dir ):
 
86
                        os.mkdir( the.expanded_dir )
82
87
 
83
 
                # checkout a remote repo, or create an empty local one?
84
88
                if self.url:
85
89
 
86
90
                        # expand url if it's a simple hostname
87
91
                        if re.match( '^[0-9a-zA-z.]+$', self.url ):
88
 
                                self.url = 'bzr+ssh://%s/%s/%s' % \
 
92
                                self.url = 'sftp://%s/%s/%s' % \
89
93
                                                   ( self.url, the.dir, the.repo.name )
90
94
 
91
95
                        # initialise deployment (with an empty repo)
92
 
                        deployment = Deployment()
 
96
                        deployment = Deployment( self.repo )
93
97
                        deployment.copy_in()
94
98
 
95
99
                        # perform bzr checkout
96
 
                        if the.verbose >= 1: print "checking out %s" % the.repo.dir
97
 
                        try:
98
 
                                the.repo.vcs.checkout( self.url )
99
 
                        except Exception as e:
 
100
                        print "checking out %s" % the.repo.dir
 
101
                        p = Popen( [ 'bzr', 'co', self.url, the.repo.name ],
 
102
                                           cwd = the.expanded_dir,
 
103
                                           stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
 
104
                        out = p.communicate()[ 0 ]
 
105
                        if p.returncode > 0:
100
106
 
101
 
                                # attempt to clean-up repo dir
 
107
                                # attempt to clean-up repo dir, and die
102
108
                                try:
103
 
                                        shutil.rmtree( the.repo.full_dir )
104
 
                                except OSError:
 
109
                                        shutil.rmtree( the.repo.expanded_dir )
 
110
                                except OSError as e:
105
111
                                        pass
106
 
 
107
 
                                raise e
 
112
                                raise the.program.FatalError( 'checkout failed', out )
108
113
 
109
114
                        # perform deployment
110
 
                        try:
111
 
                                deployment.copy_out()
112
 
                        except deployment.Conflict as e:
113
 
                                raise the.program.FatalError( e.msg )
 
115
                        deployment.copy_out()
114
116
 
115
117
                else:
116
118
 
117
119
                        # perform bzr init
118
 
                        if the.verbose >= 1: print 'initialising %s' % the.repo.dir
119
 
                        try:
120
 
                                the.repo.vcs.init()
121
 
                        except Exception as e:
 
120
                        print 'creating %s' % the.repo.dir
 
121
                        p = Popen( [ 'bzr', 'init', the.repo.name ],
 
122
                                           cwd = the.expanded_dir,
 
123
                                           stdout = subprocess.PIPE, stderr = subprocess.STDOUT )
 
124
                        out = p.communicate()[ 0 ]
 
125
                        if p.returncode > 0:
122
126
 
123
127
                                # attempt to clean-up repo dir, and die
124
128
                                try:
125
 
                                        shutil.rmtree( the.repo.full_dir )
126
 
                                except OSError:
 
129
                                        shutil.rmtree( the.repo.expanded_dir )
 
130
                                except OSError as e:
127
131
                                        pass
128
 
 
129
 
                                raise e
 
132
                                raise the.program.FatalError( 'init failed', out )