Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036 NAME="download_checkmd5.py"
00037
00038 import urllib, hashlib, os, sys
00039 from optparse import OptionParser
00040
00041 def main():
00042 parser = OptionParser(usage="usage: %prog URI dest [md5sum]", prog=NAME)
00043 options, args = parser.parse_args()
00044 md5sum = None
00045 if len(args) == 2:
00046 uri, dest = args
00047 elif len(args) == 3:
00048 uri, dest, md5sum = args
00049 else:
00050 parser.error("wrong number of arguments")
00051
00052
00053 d = os.path.dirname(dest)
00054 if len(d) and not os.path.exists(d):
00055 os.makedirs(d)
00056
00057 fresh = False
00058 if not os.path.exists(dest):
00059 sys.stdout.write('[rosbuild] Downloading %s to %s...'%(uri, dest))
00060 sys.stdout.flush()
00061 urllib.urlretrieve(uri, dest)
00062 sys.stdout.write('Done\n')
00063 fresh = True
00064
00065 if md5sum:
00066 m = hashlib.md5(open(dest).read())
00067 d = m.hexdigest()
00068
00069 print('[rosbuild] Checking md5sum on %s'%(dest))
00070
00071 if d != md5sum:
00072 if not fresh:
00073 print('[rosbuild] WARNING: md5sum mismatch (%s != %s); re-downloading file %s' % (d, md5sum, dest))
00074 os.remove(dest)
00075
00076
00077 urllib.urlretrieve(uri, dest)
00078 m = hashlib.md5(open(dest).read())
00079 d = m.hexdigest()
00080
00081 if d != md5sum:
00082 print('[rosbuild] ERROR: md5sum mismatch (%s != %s) on %s; aborting' % (d, md5sum, dest))
00083 return 1
00084
00085 return 0
00086
00087
00088 if __name__ == '__main__':
00089 sys.exit(main())