bstinson / centos / t_functional

Forked from centos/t_functional 4 years ago
Clone

Blame tests/z_rpminfo/check-rpminfo.py

0b568f
#!/usr/bin/python -tt
0b568f
'''
0b568f
Author: Brian Stinson <brian@bstinson.com>
0b568f
Check rpminfo attributes that must be set before release in the repositories (e.g. Vendor, Packager)
0b568f
'''
0b568f
0b568f
from __future__ import print_function, with_statement
0b568f
import os
0b568f
import sys
0b568f
import yum
0b568f
0b568f
from datetime import datetime
0b568f
from fnmatch import fnmatch
0b568f
0b568f
0b568f
def log(methodstring):
0b568f
    localtime = datetime.now()
0b568f
    print("[+] {0} -> {1}".format(localtime.strftime("%a %b %e %H:%M:%S %Z %Y"), methodstring))
0b568f
0b568f
0b568f
def fail(failstring):
0b568f
    log('FAIL {0}'.format(failstring))
0b568f
0b568f
0b568f
def skip(skipstring):
0b568f
    log('SKIP {0}'.format(skipstring))
0b568f
0b568f
log('Running check-rpminfo.py - Check rpminfo Attributes')
0b568f
0b568f
if str(os.getenv('PRE_UPDATES')) == '1':
0b568f
    checkrepos = ['preupdates']
0b568f
else:
0b568f
    log('Not running in preupdates, checking all repos')
0b568f
    checkrepos = ['extras', 'base', 'updates', 'centosplus']
0b568f
0b568f
# Set the required attributes and their values here. You may use shell-style
0b568f
# globs in the value if necessary.
0b568f
required_attrs = {
0b568f
    'vendor': 'CentOS',
0b568f
    'packager': '*centos.org*',
0b568f
    }
0b568f
0b568f
# special_overrides is a dictionary of packages and a list of attributes
0b568f
# to ignore for that package. When updating this list add a comment to
0b568f
# the line describing why the override is in place
0b568f
special_overrides = {
0b568f
    'epel-release': ['vendor', 'packager'],                   #Rebuilt directly from Fedora, so the vendor remains Fedora Project
0b568f
    'redhat-support-lib-python': ['vendor'],
0b568f
    'redhat-support-tool': ['vendor'],
0b568f
    }
0b568f
0b568f
# you can also add the NVR, and the attribute to ignore to the special_overrides.txt file
0b568f
with open(os.path.join(sys.path[0],'special_overrides.txt'),'r') as thefile:
0b568f
    for line in thefile.readlines():
0b568f
        if not line.strip() or line.startswith('#'):
0b568f
            continue
0b568f
        pkg, ignoreattr = map(str.strip,line.split(':'))
0b568f
        special_overrides.setdefault(pkg, []).append(ignoreattr)
0b568f
0b568f
yb = yum.YumBase()
0b568f
yb.conf.cache = 0
0b568f
0b568f
yb.repos.disableRepo('*')
0b568f
0b568f
for repo in checkrepos:
0b568f
    yb.repos.enableRepo(repo)
0b568f
yb.repos.doSetup()
0b568f
0b568f
sack = yb.pkgSack
0b568f
0b568f
finalret = 0
0b568f
for pkg in sack:
0b568f
    for attr, val in required_attrs.iteritems():
0b568f
        nvr = '{0}-{1}-{2}'.format(pkg['name'], pkg['version'], pkg['release'])
0b568f
        if pkg.name in special_overrides:
0b568f
            if attr in special_overrides[pkg.name]:
0b568f
                skip('{0}: {1} listed in special_overrides'.format(pkg.remote_path, attr))
0b568f
                continue
0b568f
        elif nvr in special_overrides:
0b568f
            if attr in special_overrides[nvr]:
0b568f
                skip('{0}: {1} listed in special_overrides'.format(pkg.remote_path, attr))
0b568f
                continue
0b568f
0b568f
        if not pkg[attr]:
0b568f
            fail('{0}: Missing {1}'.format(pkg.remote_path, attr))
0b568f
            finalret = 1
0b568f
            continue
0b568f
0b568f
        if not fnmatch(str.upper(pkg[attr]), str.upper(val)):
0b568f
            fail('{0}: {1}: {2} does not match {3}'.format(pkg.remote_path, attr, pkg[attr], val))
0b568f
            finalret = 1
0b568f
0b568f
sys.exit(finalret)