GREEN: policy should suppors PkgVerFile now

1 files changed, 3 insertions, 0 deletions

diff --git a/python/policy.cc b/python/policy.cc
index b11e4dde..b7e648b5 100644
--- a/python/policy.cc
+++ b/python/policy.cc
@@ -46,6 +46,9 @@ PyObject *policy_get_priority(PyObject *self, PyObject *arg) {
     if (PyObject_TypeCheck(arg, &PyPackage_Type)) {
         pkgCache::PkgIterator pkg = GetCpp<pkgCache::PkgIterator>(arg);
         return MkPyNumber(policy->GetPriority(pkg));
+    } else if (PyObject_TypeCheck(arg, &PyPackageFile_Type)) {
+        pkgCache::PkgFileIterator pkgfile = GetCpp<pkgCache::PkgFileIterator>(arg);
+        return MkPyNumber(policy->GetPriority(pkgfile));
     } else {
         PyErr_SetString(PyExc_TypeError,"Argument must be of Package().");
         return 0;