1 """pyversioncheck - Module to help with checking versions"""

     2 import types

     3 import rfc822

     4 import urllib

     5 import sys

     6 

     7 # Verbose options

     8 VERBOSE_SILENT=0        # Single-line reports per package

     9 VERBOSE_NORMAL=1        # Single-line reports per package, more info if outdated

    10 VERBOSE_EACHFILE=2      # Report on each URL checked

    11 VERBOSE_CHECKALL=3      # Check each URL for each package

    12 

    13 # Test directory

    14 ## urllib bug: _TESTDIR="ftp://ftp.cwi.nl/pub/jack/python/versiontestdir/"

    15 _TESTDIR="http://www.cwi.nl/~jack/versiontestdir/"

    16 

    17 def versioncheck(package, url, version, verbose=0):

    18     ok, newversion, fp = checkonly(package, url, version, verbose)

    19     if verbose > VERBOSE_NORMAL:

    20         return ok

    21     if ok < 0:

    22         print '%s: No correctly formatted current version file found'%(package)

    23     elif ok == 1:

    24         print '%s: up-to-date (version %s)'%(package, version)

    25     else:

    26         print '%s: version %s installed, version %s found:' % \

    27                         (package, version, newversion)

    28         if verbose > VERBOSE_SILENT:

    29             while 1:

    30                 line = fp.readline()

    31                 if not line: break

    32                 sys.stdout.write('\t'+line)

    33     return ok

    34 

    35 def checkonly(package, url, version, verbose=0):

    36     if verbose >= VERBOSE_EACHFILE:

    37         print '%s:'%package

    38     if type(url) == types.StringType:

    39         ok, newversion, fp = _check1version(package, url, version, verbose)

    40     else:

    41         for u in url:

    42             ok, newversion, fp = _check1version(package, u, version, verbose)

    43             if ok >= 0 and verbose < VERBOSE_CHECKALL:

    44                 break

    45     return ok, newversion, fp

    46 

    47 def _check1version(package, url, version, verbose=0):

    48     if verbose >= VERBOSE_EACHFILE:

    49         print '  Checking %s'%url

    50     try:

    51         fp = urllib.urlopen(url)

    52     except IOError, arg:

    53         if verbose >= VERBOSE_EACHFILE:

    54             print '    Cannot open:', arg

    55         return -1, None, None

    56     msg = rfc822.Message(fp, seekable=0)

    57     newversion = msg.getheader('current-version')

    58     if not newversion:

    59         if verbose >= VERBOSE_EACHFILE:

    60             print '    No "Current-Version:" header in URL or URL not found'

    61         return -1, None, None

    62     version = version.lower().strip()

    63     newversion = newversion.lower().strip()

    64     if version == newversion:

    65         if verbose >= VERBOSE_EACHFILE:

    66             print '    Version identical (%s)'%newversion

    67         return 1, version, fp

    68     else:

    69         if verbose >= VERBOSE_EACHFILE:

    70             print '    Versions different (installed: %s, new: %s)'% \

    71                         (version, newversion)

    72         return 0, newversion, fp

    73 

    74 

    75 def _test():

    76     print '--- TEST VERBOSE=1'

    77     print '--- Testing existing and identical version file'

    78     versioncheck('VersionTestPackage', _TESTDIR+'Version10.txt', '1.0', verbose=1)

    79     print '--- Testing existing package with new version'

    80     versioncheck('VersionTestPackage', _TESTDIR+'Version11.txt', '1.0', verbose=1)

    81     print '--- Testing package with non-existing version file'

    82     versioncheck('VersionTestPackage', _TESTDIR+'nonexistent.txt', '1.0', verbose=1)

    83     print '--- Test package with 2 locations, first non-existing second ok'

    84     versfiles = [_TESTDIR+'nonexistent.txt', _TESTDIR+'Version10.txt']

    85     versioncheck('VersionTestPackage', versfiles, '1.0', verbose=1)

    86     print '--- TEST VERBOSE=2'

    87     print '--- Testing existing and identical version file'

    88     versioncheck('VersionTestPackage', _TESTDIR+'Version10.txt', '1.0', verbose=2)

    89     print '--- Testing existing package with new version'

    90     versioncheck('VersionTestPackage', _TESTDIR+'Version11.txt', '1.0', verbose=2)

    91     print '--- Testing package with non-existing version file'

    92     versioncheck('VersionTestPackage', _TESTDIR+'nonexistent.txt', '1.0', verbose=2)

    93     print '--- Test package with 2 locations, first non-existing second ok'

    94     versfiles = [_TESTDIR+'nonexistent.txt', _TESTDIR+'Version10.txt']

    95     versioncheck('VersionTestPackage', versfiles, '1.0', verbose=2)

    96 

    97 if __name__ == '__main__':

    98     _test()