pip_version_check.py

#-*- coding:utf-8 -*-
"""
    This script compare a package installed by pip and the latest version
    in PyPI. And it is also possible to upgrade it then and there.

    For now work on Windows and Linux.
    And I'm not MACer.(Probably run.)

    require: Python 2.6, 2.7 or 3.x
    If you using Python 2.5 or below, check at gist:1153625
"""

from pip import get_installed_distributions, call_subprocess
from subprocess import call
import sys

try:
    # Python 2.6, 2.7
    import xmlrpclib
except ImportError:
    # Python 3.x
    raw_input = input
    import xmlrpc.client as xmlrpclib


def check_versions(pypi_ver, installed):
    if installed.has_version():
        print("Checking:  ** {0} **".format(installed.project_name))
        print("IN PyPI:   {0}".format(pypi_ver[0]))
        print("INSTALLED: {0}".format(installed.version))
        if pypi_ver[0] == installed.version:
            print('Using latest version.\n')
        else:
            print("Versions don't match.\n")
            pip_upgrade(installed)
    else:
        print("{0} doesn't have version.\n".format(installed.project_name))

def pip_upgrade(dist):
    qstring1 = 'Do you want to upgrade "{0}"?: (y/n)'
    qstring2 = 'Really? (y/n):'
    upgrade_string = 'pip install --upgrade {0}'
    ans1 = raw_input(qstring1.format(dist.project_name))
    if ans1 == 'y':
        ans2 = raw_input(qstring2)
        if ans2 == 'y':
            try:
                # Windows
                call_subprocess(upgrade_string.format(dist.project_name))
            except OSError:
                # Linux
                call(upgrade_string.format(dist.project_name), shell=True)
            finally:
                print('Done.\n')
        else:
            print("Upgrading Canceled: {0}.\n".format(dist.project_name))
    else:
        print("Upgrading Canceled: {0}.\n".format(dist.project_name))

def main():
    client = xmlrpclib.ServerProxy('http://pypi.python.org/pypi')
    for installed in get_installed_distributions():
        pypi_ver = client.package_releases(installed.project_name)
        if len(pypi_ver) > 0:
            check_versions(pypi_ver, installed)
        else:
            capitalized = installed.project_name.capitalize()
            pypi_ver = client.package_releases(capitalized)
            if pypi_ver:
                check_versions(pypi_ver, installed)
            else:
                print('"{0}" is not found in PyPI\n'.format(
                                            installed.project_name))


if __name__ == '__main__':
    main()