Revision: 120671 https://trac.macports.org/changeset/120671 Author: gaurav@macports.org Date: 2014-06-05 05:00:26 -0700 (Thu, 05 Jun 2014) Log Message: ----------- Added support for rmd160 and sha256 checksums Modified Paths: -------------- branches/gsoc14-pip2port/tester.py Modified: branches/gsoc14-pip2port/tester.py =================================================================== --- branches/gsoc14-pip2port/tester.py 2014-06-05 11:41:21 UTC (rev 120670) +++ branches/gsoc14-pip2port/tester.py 2014-06-05 12:00:26 UTC (rev 120671) @@ -118,29 +118,57 @@ if name.split("/")[0] == "EGG-INFO": # print name zip.extract(name,src_dir) - + print "\n" + return file_name else: print 'Aborting due to inconsistency on checksums\n' try: os.remove(file_name) except OSError, e: print "Error: %s - %s." % (e.filename,e.strerror) + print "\n" + return False - -def fetch_url(pkg_name,pkg_version=None): - print "\n" +def fetch_url(pkg_name,pkg_version,checksum=False): values = client.release_urls(pkg_name,pkg_version) - for value in values: - fetch(pkg_name,value) - print "\n" + if checksum: + for value in values: + if value['filename'].split('.')[-1] == 'gz': + return fetch(pkg_name,value) + else: + print "\n" + for value in values: + return fetch(pkg_name,value) -def checksum_md5(dict2): - for value in dict2: - if not value['filename'].split('.')[-1] == 'egg': - return value['md5_digest'] - return +def checksum_rmd160(pkg_name,pkg_version): + file_name = fetch_url(pkg_name,pkg_version,True) + print file_name + if file_name: + try: + h = hashlib.new('ripemd160') + f = open(file_name) + h.update(f.read()) + checksum = h.hexdigest() + f.close() + return checksum + except: + print "Error\n" + return -def create_portfile(dict,file_name,dict2): +def checksum_sha256(pkg_name,pkg_version): + file_name = fetch_url(pkg_name,pkg_version,True) + print file_name + if file_name: + try: + f = open(file_name) + checksum = hashlib.sha256(f.read()).hexdigest() + f.close() + return checksum + except: + print "Error\n" + return + +def create_portfile(dict,file_name): file = open(file_name, 'w') file.write('# -*- coding: utf-8; mode: tcl; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- vim:fenc=utf-8:ft=tcl:et:sw=4:ts=4:sts=4\n') @@ -166,9 +194,8 @@ file.write('master_sites '+dict['download_url']+'\n') file.write('distname py-'+dict['name']+dict['version']+'\n\n') - checksum = checksum_md5(dict2) - if checksum: - file.write('checksums md5 '+checksum+'\n\n') + file.write('checksums rmd160 '+checksum_rmd160(dict['name'],dict['version'])+'\n\n') + file.write(' sha256 '+checksum_sha256(dict['name'],dict['version'])+'\n\n') file.write('python.versions 25 26 27\n\n') # file.write('if {${name} ne ${subport}} {\n') @@ -190,13 +217,13 @@ os.makedirs(src_dir) dict = client.release_data(pkg_name,pkg_version) - dict2 = client.release_urls(pkg_name,pkg_version) +# dict2 = client.release_urls(pkg_name,pkg_version) file_name = os.path.join(src_dir,"Test_Portfile") # try: # create_portfile(dict,file_name,dict2) # print "SUCCESS\n" - create_portfile(dict,file_name,dict2) + create_portfile(dict,file_name) print "SUCCESS\n" # except: # print "ERROR"