import sys
from nose.plugins.base import Plugin

from StringIO import StringIO

import pygtk
pygtk.require('2.0')
import pynotify
from new import instancemethod

def writeln(self,*args):
    for i in args:
        self.write(i)
    self.write("\n")

class NotifyPlugin(Plugin):

    enabled = True
    score = 1000
    name = "NotifyPlugin"
    
    def __init__(self):
        self.stdout = []
        self._buf = StringIO()
        self._buf.writeln = instancemethod(writeln,self._buf,StringIO)
    
    def configure(self, options, conf):
        pass

    def options(self, parser, env):
        pass

    def finalize(self, result):
        result.stream = self._buf
        result.printSummary(0,2)
        pynotify.init("Basics")
        n = pynotify.Notification("Tests", self._buf.getvalue())
        n.show()
        return None