[yum-commits] Branch 'yum-3_2_X' - yum/packages.py

[James Antill]

 yum/packages.py |    9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

New commits:
commit 3ff72f5413a6e82c00e4329d6130e0c8468009f9
Author: James Antill <james at and.org>
Date:   Wed Jul 7 17:10:21 2010 -0400

    Add the package we are operating on to YumVerifyPackage, as self.po

diff --git a/yum/packages.py b/yum/packages.py
index 8f160da..b5a7d40 100644
--- a/yum/packages.py
+++ b/yum/packages.py
@@ -1579,7 +1579,8 @@ class _RPMVerifyPackageFile(YUMVerifyPackageFile):
 
 class YUMVerifyPackage:
     """ A holder for YUMVerifyPackageFile objects. """
-    def __init__(self):
+    def __init__(self, po):
+        self.po = po
         self._files = {}
 
     def __contains__(self, fname):
@@ -1607,8 +1608,8 @@ class YUMVerifyPackage:
 
 
 class _RPMVerifyPackage(YUMVerifyPackage):
-    def __init__(self, fi, def_csum_type, patterns, all):
-        YUMVerifyPackage.__init__(self)
+    def __init__(self, po, fi, def_csum_type, patterns, all):
+        YUMVerifyPackage.__init__(self, po)
 
         self._presetup = (fi, def_csum_type, patterns, all)
     def _setup(self):
@@ -1681,7 +1682,7 @@ class YumInstalledPackage(YumHeaderPackage):
                     csum_type = RPM_CHECKSUM_TYPES[csum_num]
                 # maybe an else with an error code here? or even a verify issue?
 
-        pfs = _RPMVerifyPackage(fi, csum_type, patterns, all)
+        pfs = _RPMVerifyPackage(self, fi, csum_type, patterns, all)
 
         if callback is not None:
             pfs = callback(pfs)