Added a func total_downloads to check all possible package name like flask, Flask, FLask etc. When package name = flask, returned download count is 0. The new function will try create all possible package name from flask to FLASK with incremental capitalization of each character

This commit is contained in:
kracekumar
2012-01-31 23:01:23 +05:30
parent b793704db8
commit 1fb1f0e5b2
+14 -2
View File
@@ -33,17 +33,29 @@ def index():
return jsonify(d)
def total_downloads(package):
count = downloads_total(package)
package = package.swapcase() if package.isupper() else package
while not count:
for pos, value in enumerate(package):
temp_package = package[:pos].swapcase() + package[pos:]
count = downloads_total(temp_package)
if count:
return count
else:
return count
@app.route('/pypi/<package>')
@cache.memoize(timeout=6*60*60)
def package_stats(package):
d = {
'package': package,
'downloads': downloads_total(package)
'downloads': total_downloads(package)
}
return jsonify(d)
if __name__ == '__main__':
app.run()
app.run()