Last change
on this file since 1922 was
1363,
checked in by mar637, 17 years ago
|
ATNF doesn't allow http acces to ftp server anymore. Use ftp directly
|
-
Property svn:executable set to
*
|
File size:
1.4 KB
|
Rev | Line | |
---|
[1179] | 1 | #!/usr/bin/env python |
---|
| 2 | import os |
---|
| 3 | import tarfile |
---|
| 4 | import shutil |
---|
[1363] | 5 | from ftplib import FTP |
---|
[1179] | 6 | |
---|
[1363] | 7 | # globals |
---|
| 8 | md5suff = '.md5sum' |
---|
[1179] | 9 | name = 'asap_data.tar.bz2' |
---|
[1363] | 10 | dataurl = "ftp.atnf.csiro.au" |
---|
[1179] | 11 | tmpdata = '/tmp/asap_data.tar.bz2' |
---|
[1363] | 12 | datadir = 'pub/software/asap/data' |
---|
| 13 | tmpmd5 = tmpdata+md5suff |
---|
[1179] | 14 | |
---|
[1363] | 15 | ftp = FTP(dataurl) # connect to host, default port |
---|
| 16 | ftp.login() # user anonymous, passwd anonymous@ |
---|
| 17 | ftp.cwd(datadir) |
---|
[1180] | 18 | print "Checking if an update is required." |
---|
[1363] | 19 | ftp.retrbinary('RETR %s' % name+md5suff, open(tmpmd5, 'wb').write) |
---|
[1179] | 20 | md5file = file(tmpmd5) |
---|
| 21 | md5new = md5file.readlines()[0].split()[0] |
---|
| 22 | |
---|
[1180] | 23 | # use ASAPDATA if set - allows non-root update |
---|
| 24 | if os.environ.has_key("ASAPDATA"): |
---|
| 25 | asapbase = os.environ["ASAPDATA"] |
---|
[1183] | 26 | else: |
---|
| 27 | import asap |
---|
| 28 | # get asap module path |
---|
| 29 | asapbase = asap.__path__[0] |
---|
[1179] | 30 | try: |
---|
[1363] | 31 | fl = os.path.join(asapbase, name+md5suff) |
---|
[1179] | 32 | data_md5 = file(fl) |
---|
| 33 | ls = data_md5.readlines()[0] |
---|
| 34 | data_md5.close() |
---|
| 35 | md5old = ls.split()[0] |
---|
| 36 | except IOError: |
---|
| 37 | md5old ='' |
---|
| 38 | |
---|
| 39 | if md5new != md5old: |
---|
[1180] | 40 | print "Update required. Downloading asap data archive...." |
---|
[1363] | 41 | ftp.retrbinary('RETR %s' % name, open(tmpdata, 'wb').write) |
---|
[1179] | 42 | os.chdir(asapbase) |
---|
| 43 | tf = tarfile.TarFile.bz2open(tmpdata) |
---|
| 44 | print "Extracting data archive in %s.." % asapbase |
---|
| 45 | for member in tf.getmembers(): |
---|
| 46 | tf.extract(member) |
---|
| 47 | shutil.copy(tmpmd5, asapbase) |
---|
| 48 | else: |
---|
| 49 | print """Data already at latest available version. |
---|
| 50 | If you still get errors running asap, please report this.""" |
---|
[1363] | 51 | ftp.quit() |
---|
Note: See
TracBrowser
for help on using the repository browser.