From 00ce4d227aef4ccde15f8cfa5b1a5582195eb639 Mon Sep 17 00:00:00 2001
From: Michael Vogt <michael.vogt@ubuntu.com>
Date: Mon, 30 Jan 2012 15:33:24 +0100
Subject: GREEN: policy should suppors PkgVerFile now

---
 python/policy.cc | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'python')

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;
-- 
cgit v1.2.3