#! /usr/bin/python

import sys
import os.path
import pygtk
import gtk
import os

class App (gtk.Window):
        def __init__(self):
                super(App, self).__init__()

                if len(sys.argv)!=3:
                        print "Need 2 args\n",
                        sys.exit()

                self.set_title(" ")
                self.set_size_request(600, 200)
                self.set_position(gtk.WIN_POS_CENTER)

                fixed = gtk.Fixed()

                txt = sys.argv[1]

###########################################################################################################
###### Verificarea fisierelor err. ########################################################################
###########################################################################################################
        listOfDirs_filename = "/tmp/CEVA"
        listOfFiles = ""
        dirName = ""
        errFile = ""
        partitionsNames = ""
        unexpectedError = 0
        try:
            f = open(listOfDirs_filename)
            lines = f.readlines()

            for dirName in lines:
                dirName = dirName.replace("\n","")
                listOfFiles = os.listdir(dirName)
                #print listOfFiles, "\n",
                for fileOrDirName in listOfFiles:
                    errFile = os.path.join(dirName,fileOrDirName)
                    if((fileOrDirName.rfind(".err") != -1) and (os.path.isfile(errFile))):
                        #print "Fisier gasit: ", errFile, "\n",
                        fisier = open(errFile)
                        l1 = fisier.readlines()
                        if(len(l1)>=2):
                            if(l1[0].split("+")[0] != l1[1].split("+")[0]):
                                partitionsNames = partitionsNames + fileOrDirName.split(".err")[0] + ", "
        except IOError:
            print "IO error on file ", listOfDirs_filename, "!\n",
            unexpectedError = 1
        except OSError:
            print "OS error on file ", dirName, "!\n",
            unexpectedError = 2

                labelPWE = gtk.Label("")
                if(unexpectedError == 0):
                        if(len(partitionsNames)>1):
                                labelPWE = gtk.Label("Device "+partitionsNames+" contained corrupt sectors!")
                        else:
                                labelPWE = gtk.Label("No Corrupt Sectors were found.")

###########################################################################################################
###########################################################################################################



                label = gtk.Label("The process is now finished! There were " + txt + " copied.")
                txt = sys.argv[2]
                label0 = gtk.Label("There are "+txt+" left on the destination drive. Press ENTER to Shutdown")
                label1 = gtk.Label("")
                testf = os.path.isfile("/tmp/FALSE")
                if(testf==False):
                        try:
                                f = open("/tmp/LOG","r+")
                                l = f.readlines()
                                loe = ""
                                for s in l:
                                        print s,
                                        lw=s.split()
                                        if(s.find("failed")!=-1):
                                                loe=loe+lw[3]+", ";
                                print "loe=", loe
                                if(loe==""):
                                        label1 = gtk.Label("Image Verification is successful.")
                                else:
                                        label1 = gtk.Label("Image verification was not successful for "+loe)
                                        label2 = gtk.Label("please refer to the logs for more detail.")
                                        fixed.put(label2, 50, 90)
                        except IOError:
                                print "Eroare",

                self.btn = gtk.Button("ENTER")
                self.btn.set_size_request(100, 30)
                self.btn.connect("clicked", self.onclick)


                fixed.put(self.btn, 450, 150)
                fixed.put(label, 50, 50)
                fixed.put(label0, 50, 110)
                fixed.put(label1, 50, 70)
                fixed.put(labelPWE,50,30)

                self.connect("destroy", gtk.main_quit)


                self.add(fixed)
                self.show_all()



        def onclick (self, widget):
                gtk.main_quit()

        def main (self, argv):
                gtk.main()

if __name__ ==  "__main__":
        app = App()
        app.main(sys.argv[1])