1 # Copyright (C) 2007, Thomas Leonard
2 # See the README file for details, or visit http://0install.net.
5 import os
, subprocess
, tarfile
6 import urlparse
, ftplib
, httplib
7 from zeroinstall
import SafeException
8 from zeroinstall
.injector
import model
, qdom
9 from zeroinstall
.support
import ro_rmtree
10 from logging
import info
12 release_status_file
= os
.path
.abspath('release-status')
14 def check_call(*args
, **kwargs
):
15 exitstatus
= subprocess
.call(*args
, **kwargs
)
17 if type(args
[0]) in (str, unicode):
20 cmd
= ' '.join(args
[0])
21 raise SafeException("Command failed with exit code %d:\n%s" % (exitstatus
, cmd
))
23 def show_and_run(cmd
, args
):
24 print "Executing: %s %s" % (cmd
, ' '.join("[%s]" % x
for x
in args
))
25 check_call(['sh', '-c', cmd
, '-'] + args
)
27 def suggest_release_version(snapshot_version
):
28 """Given a snapshot version, suggest a suitable release version.
29 >>> suggest_release_version('1.0-pre')
31 >>> suggest_release_version('0.9-post')
33 >>> suggest_release_version('3')
34 Traceback (most recent call last):
36 SafeException: Version '3' is not a snapshot version (should end in -pre or -post)
38 version
= model
.parse_version(snapshot_version
)
41 raise SafeException("Version '%s' is not a snapshot version (should end in -pre or -post)" % snapshot_version
)
43 # -post, so increment the number
45 version
[-1] = 0 # Remove the modifier
46 return model
.format_version(version
)
48 def publish(feed_path
, **kwargs
):
49 args
= [os
.environ
['0PUBLISH']]
53 args
+= ['--' + k
.replace('_', '-')]
54 elif value
is not None:
55 args
+= ['--' + k
.replace('_', '-'), value
]
56 args
.append(feed_path
)
57 info("Executing %s", args
)
60 def get_singleton_impl(feed
):
61 impls
= feed
.implementations
63 raise SafeException("Local feed '%s' contains %d versions! I need exactly one!" % (feed
.url
, len(impls
)))
64 return impls
.values()[0]
66 def backup_if_exists(name
):
67 if not os
.path
.exists(name
):
70 if os
.path
.exists(backup
):
71 print "(deleting old backup %s)" % backup
72 if os
.path
.isdir(backup
):
76 os
.rename(name
, backup
)
77 print "(renamed old %s as %s; will delete on next run)" % (name
, backup
)
79 def get_choice(options
):
81 choice
= raw_input('/'.join(options
) + ': ').lower()
82 if not choice
: continue
84 if o
.lower().startswith(choice
):
87 def make_archive_name(feed_name
, version
):
88 return feed_name
.lower().replace(' ', '-') + '-' + version
91 for x
in os
.environ
['PATH'].split(':'):
92 if os
.path
.isfile(os
.path
.join(x
, prog
)):
96 def show_diff(from_dir
, to_dir
):
97 for cmd
in [['meld'], ['xxdiff'], ['diff', '-ur']]:
99 code
= os
.spawnvp(os
.P_WAIT
, cmd
[0], cmd
+ [from_dir
, to_dir
])
101 print "WARNING: command %s failed with exit code %d" % (cmd
, code
)
104 class Status(object):
105 __slots__
= ['old_snapshot_version', 'release_version', 'head_before_release', 'new_snapshot_version',
106 'head_at_release', 'created_archive', 'src_tests_passed', 'tagged', 'verified_uploads', 'updated_master_feed']
108 for name
in self
.__slots
__:
109 setattr(self
, name
, None)
111 if os
.path
.isfile(release_status_file
):
112 for line
in file(release_status_file
):
113 assert line
.endswith('\n')
115 name
, value
= line
.split('=')
116 setattr(self
, name
, value
)
117 info("Loaded status %s=%s", name
, value
)
120 tmp_name
= release_status_file
+ '.new'
121 tmp
= file(tmp_name
, 'w')
123 lines
= ["%s=%s\n" % (name
, getattr(self
, name
)) for name
in self
.__slots
__ if getattr(self
, name
)]
124 tmp
.write(''.join(lines
))
126 os
.rename(tmp_name
, release_status_file
)
127 info("Wrote status to %s", release_status_file
)
133 if hasattr(address
, 'hostname'):
134 return address
.hostname
136 return address
[1].split(':', 1)[0]
139 if hasattr(address
, 'port'):
142 port
= address
[1].split(':', 1)[1:]
148 def get_http_size(url
, ttl
= 1):
149 assert url
.lower().startswith('http://')
151 address
= urlparse
.urlparse(url
)
152 http
= httplib
.HTTPConnection(host(address
), port(address
) or 80)
154 parts
= url
.split('/', 3)
160 http
.request('HEAD', '/' + path
, headers
= {'Host': host(address
)})
161 response
= http
.getresponse()
163 if response
.status
== 200:
164 return response
.getheader('Content-Length')
165 elif response
.status
in (301, 302):
166 new_url_rel
= response
.getheader('Location') or response
.getheader('URI')
167 new_url
= urlparse
.urljoin(url
, new_url_rel
)
169 raise SafeException("HTTP error: got status code %s" % response
.status
)
174 info("Resource moved! Checking new URL %s" % new_url
)
176 return get_http_size(new_url
, ttl
- 1)
178 raise SafeException('Too many redirections.')
180 def get_ftp_size(url
):
181 address
= urlparse
.urlparse(url
)
182 ftp
= ftplib
.FTP(host(address
))
185 return ftp
.size(url
.split('/', 3)[3])
190 scheme
= urlparse
.urlparse(url
)[0].lower()
191 if scheme
.startswith('http'):
192 return get_http_size(url
)
193 elif scheme
.startswith('ftp'):
194 return get_ftp_size(url
)
196 raise SafeException("Unknown scheme '%s' in '%s'" % (scheme
, url
))
198 def unpack_tarball(archive_file
):
199 tar
= tarfile
.open(archive_file
, 'r:bz2')
200 members
= [m
for m
in tar
.getmembers() if m
.name
!= 'pax_global_header']
201 #tar.extractall('.', members = members) # Python >= 2.5 only
202 for tarinfo
in members
:
203 tarinfo
= copy
.copy(tarinfo
)
206 tar
.extract(tarinfo
, '.')
209 with
open(path
, 'rb') as stream
:
210 return model
.ZeroInstallFeed(qdom
.parse(stream
), local_path
= path
)
212 def get_archive_basename(impl
):
213 # "2" means "path" (for Python 2.4)
214 return os
.path
.basename(urlparse
.urlparse(impl
.download_sources
[0].url
)[2])
216 def relative_path(ancestor
, dst
):
217 stem
= os
.path
.abspath(os
.path
.dirname(ancestor
))
218 dst
= os
.path
.abspath(dst
)
221 assert dst
.startswith(stem
)
222 return dst
[len(stem
):]
224 assert relative_path('/foo', '/foo') == 'foo'
225 assert relative_path('/foo', '/foo/bar') == 'foo/bar'
227 def make_readonly_recursive(path
):
228 for root
, dirs
, files
in os
.walk(path
):
229 for d
in dirs
+ files
:
230 full
= os
.path
.join(root
, d
)
231 mode
= os
.stat(full
).st_mode
232 os
.chmod(full
, mode
& 0o555)