merged from lp:~mvo/apt/mvo

1 files changed, 4 insertions, 1 deletions

diff --git a/python/policy.cc b/python/policy.cc
index b11e4dde..96b83abd 100644
--- a/python/policy.cc
+++ b/python/policy.cc
@@ -46,8 +46,11 @@ 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().");
+        PyErr_SetString(PyExc_TypeError,"Argument must be of Package() or PackageFile().");
         return 0;
     }
 }