/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-03-09 01:56:01 UTC
  • Revision ID: tim@ed.am-20140309015601-ptqyjqrngb7i6dmj
changed use of python

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
#
20
20
 
21
21
 
22
22
import sys, os, re, getopt, shutil, subprocess
23
 
import the, deployment
24
 
from subprocess import Popen
25
 
 
26
 
 
27
 
class CommandInit:
 
23
from command import Command
 
24
import stdhome.the as the
 
25
from stdhome.deployment import Deployment
 
26
from stdhome.subprocess import Popen
 
27
 
 
28
 
 
29
class InitCommand( Command ):
28
30
 
29
31
 
30
32
        def __init__( self ):
45
47
                print
46
48
                print "Options:"
47
49
                print "  -r, --repo=REPO  select the repo to check-out or create (defaults to 'home')"
 
50
                print "  -v, --verbose    display information about what is being done"
48
51
                print "      --help       display help and exit"
49
52
                exit( 0 )
50
53
 
54
57
                        sys.argv[ 1: ], "r:v",
55
58
                        [ "repo=", "verbose", "help" ] )
56
59
                for opt, optarg in opts:
57
 
                        if [ '--repo', '-r' ].count( opt ):
 
60
                        if opt in [ '--repo', '-r' ]:
58
61
                                if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
59
 
                                        raise the.FatalError(
 
62
                                        raise the.program.FatalError(
60
63
                                                'invalid repository name: ' + optarg )
61
64
                                self.repo = optarg
62
 
                        elif [ '--verbose', '-v' ].count( opt ):
63
 
                                the.verbose = True
 
65
                        elif opt in [ '--verbose', '-v' ]:
 
66
                                the.verbose += 1
64
67
                        elif opt == "--help":
65
68
                                self.print_help()
66
69
                
76
79
 
77
80
 
78
81
        def run( self ):
 
82
 
 
83
                # set up repo and check it *doesn't* already exists
79
84
                the.set_repo( self.repo )
80
 
 
81
 
                # repo dir must not already exist
82
85
                the.repo.check_dir_exists( False )
83
86
 
84
87
                # ensure our top-level directory exists
85
 
                if not os.path.exists( the.expanded_dir ):
86
 
                        os.mkdir( the.expanded_dir )
 
88
                if not os.path.exists( the.full_dir ):
 
89
                        os.mkdir( the.full_dir )
87
90
 
 
91
                # checkout a remote repo, or create an empty local one?
88
92
                if self.url:
89
93
 
90
94
                        # expand url if it's a simple hostname
93
97
                                                   ( self.url, the.dir, the.repo.name )
94
98
 
95
99
                        # initialise deployment (with an empty repo)
96
 
                        deployment = Deployment( self.repo )
 
100
                        deployment = Deployment()
97
101
                        deployment.copy_in()
98
102
 
99
103
                        # perform bzr checkout
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:
 
104
                        if the.verbose: print "checking out %s" % the.repo.dir
 
105
                        try:
 
106
                                the.repo.vcs.checkout( self.url )
 
107
                        except Exception as e:
106
108
 
107
 
                                # attempt to clean-up repo dir, and die
 
109
                                # attempt to clean-up repo dir
108
110
                                try:
109
 
                                        shutil.rmtree( the.repo.expanded_dir )
110
 
                                except OSError as e:
 
111
                                        shutil.rmtree( the.repo.full_dir )
 
112
                                except OSError:
111
113
                                        pass
112
 
                                raise the.program.FatalError( 'checkout failed', out )
 
114
 
 
115
                                raise e
113
116
 
114
117
                        # perform deployment
115
 
                        deployment.copy_out()
 
118
                        try:
 
119
                                deployment.copy_out()
 
120
                        except deployment.Conflict as e:
 
121
                                raise the.program.FatalError( e.msg )
116
122
 
117
123
                else:
118
124
 
119
125
                        # perform bzr init
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:
 
126
                        if the.verbose: print 'initialising %s' % the.repo.dir
 
127
                        try:
 
128
                                the.repo.vcs.init()
 
129
                        except Exception as e:
126
130
 
127
131
                                # attempt to clean-up repo dir, and die
128
132
                                try:
129
 
                                        shutil.rmtree( the.repo.expanded_dir )
130
 
                                except OSError as e:
 
133
                                        shutil.rmtree( the.repo.full_dir )
 
134
                                except OSError:
131
135
                                        pass
132
 
                                raise the.program.FatalError( 'init failed', out )
 
136
 
 
137
                                raise e