From 5c219e07aea347f652cf7949dc74f37282a17144 Mon Sep 17 00:00:00 2001
From: Julian Andres Klode <jak@debian.org>
Date: Sun, 19 Jul 2009 14:39:41 +0200
Subject: python/cdromprogress.cc: Add apt_pkg.CdromProgress.

---
 python/progress.cc | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'python/progress.cc')

diff --git a/python/progress.cc b/python/progress.cc
index 123a1805..b19ab0c7 100644
--- a/python/progress.cc
+++ b/python/progress.cc
@@ -518,10 +518,14 @@ pkgPackageManager::OrderResult PyInstallProgress::Run(pkgPackageManager *pm)
 void PyCdromProgress::Update(string text, int current)
 {
    PyObject *arglist = Py_BuildValue("(si)", text.c_str(), current);
-
-   PyObject *o = Py_BuildValue("i", totalSteps);
-   PyObject_SetAttrString(callbackInst, "totalSteps", o);
-   Py_XDECREF(o);
+   if (PyObject_TypeCheck(callbackInst, &PyCdromProgress_Type)) {
+      ((PyCdromProgressObject *)callbackInst)->total_steps = totalSteps;
+   }
+   else {
+      PyObject *o = Py_BuildValue("i", totalSteps);
+      PyObject_SetAttrString(callbackInst, "totalSteps", o);
+      Py_XDECREF(o);
+   }
 
    RunSimpleCallback("update", arglist);
 }
-- 
cgit v1.2.3