1 files changed, 20 insertions, 13 deletions
diff --git a/python/progress.h b/python/progress.h
index 6ea0b16b..a39ae41a 100644
--- a/python/progress.h
+++ b/python/progress.h
@@ -13,34 +13,39 @@
 #include <apt-pkg/acquire.h>
 #include <Python.h>
 
-struct PyOpProgress : public OpProgress
-{
+class PyCallbackObj {
+ protected:
    PyObject *callbackInst;
 
+ public:
    void setCallbackInst(PyObject *o) {
+      Py_INCREF(o);
       callbackInst = o;
    }
 
-   virtual void Update();
-   virtual void Done();
+   bool RunSimpleCallback(const char *method, PyObject *arglist=NULL);
 
-   PyOpProgress() : OpProgress(), callbackInst(0) {};
+   PyCallbackObj() : callbackInst(0) {};
+   ~PyCallbackObj()  {Py_DECREF(callbackInst); };
 };
 
-
-struct PyFetchProgress : public pkgAcquireStatus 
+struct PyOpProgress : public OpProgress, public PyCallbackObj
 {
-   PyObject *callbackInst;
 
-   void setCallbackInst(PyObject *o) {
-      callbackInst = o;
-   }
+   virtual void Update();
+   virtual void Done();
 
+   PyOpProgress() : OpProgress(), PyCallbackObj() {};
+};
 
-   void updateStatus(pkgAcquire::ItemDesc & Itm, int status);
 
-   // apt interface
+struct PyFetchProgress : public pkgAcquireStatus, public PyCallbackObj
+{
+   void UpdateStatus(pkgAcquire::ItemDesc & Itm, int status);
+
    virtual bool MediaChange(string Media, string Drive);
+
+   /* apt stuff */   
    virtual void IMSHit(pkgAcquire::ItemDesc &Itm);
    virtual void Fetch(pkgAcquire::ItemDesc &Itm);
    virtual void Done(pkgAcquire::ItemDesc &Itm);
@@ -49,6 +54,8 @@ struct PyFetchProgress : public pkgAcquireStatus
    virtual void Stop();
 
    bool Pulse(pkgAcquire * Owner);
+   PyFetchProgress() : PyCallbackObj() {};
+   
 };