import shutil
import unittest
import os.path
import os
import sh

from test.helpers.md5 import md5_file

class Qusb(unittest.TestCase):
    params = None
    __resultlist = None  # resultlist is a python list of python dictionaries
    __QusbName = None

    def __init__(self, testname, testfunc, varlist):
        self.params = varlist
        super(Qusb, self).__init__(testfunc)
        self._testMethodDoc = testname
        self.__xmlObj = varlist["xml"]

        self.__QusbName = varlist.get('name', 'qusb')
        self.__loops = varlist.get('loops', self.__xmlObj.getKeyVal(self.__QusbName, "loops", "3"))
        self.__filepath_from = varlist.get('path', self.__xmlObj.getKeyVal(self.__QusbName, "path", "/root/hwtest-files"))
        self.__file_source = varlist.get('file', self.__xmlObj.getKeyVal(self.__QusbName, "file", "usb-test.bin"))
        self.__file_md5 = varlist.get('file_md5', self.__xmlObj.getKeyVal(self.__QusbName, "file_md5", "usb-test.bin.md5"))
        self.__path_to = varlist.get('pathto', self.__xmlObj.getKeyVal(self.__QusbName, "pathto", "/mnt/test/disk2"))
        self.__check_mounted = varlist.get('check_mounted', self.__xmlObj.getKeyVal(self.__QusbName, "check_mounted", ""))

        self.__resultlist = []

    def removefileIfExist(self, fname):
        if os.path.exists(fname):
            try:
                os.remove(fname)
            except OSError as error:
                return False, str(error)
        return True, ''

    def checkfiles(self):
        if not os.path.isfile('{}/{}'.format(self.__filepath_from, self.__file_source)):
            return False, 'File test binary Not found {}/{}'.format(self.__filepath_from, self.__file_source)
        if not os.path.isfile('{}/{}'.format(self.__filepath_from, self.__file_md5)):
            return False, 'File test binary md5 Not found {}/{}'.format(self.__filepath_from, self.__file_md5)
        if self.__check_mounted != '':
            if not os.path.ismount('{}'.format(self.__check_mounted)):
                return False, 'Required mounted failed: {}'.format(self.__path_to)
        r, s = self.removefileIfExist('{}/{}'.format(self.__path_to, self.__file_source))
        if not r:
            return r, s
        return True, ''

    def getTestFile_md5(self, fname):
        t = ''
        with open('{}'.format(fname), 'r') as outfile:
            t = outfile.read().rstrip("\n")
        outfile.close()
        return t

    def ExecuteTranfer(self):
        try:
            originalMD5 = self.getTestFile_md5('{}/{}'.format(self.__filepath_from, self.__file_md5))
            if originalMD5 == '':
                self.fail('MD5 file {}/{} Not contain any md5'.format(self.__filepath_from, self.__file_md5))
            # shutil.copy2('{}/{}'.format(self.__filepath_from, self.__file_source), '{}'.format(self.__path_to))
            sh.cp('{}/{}'.format(self.__filepath_from, self.__file_source), '{}'.format(self.__path_to))
            sh.sync()
            md5val = md5_file('{}/{}'.format(self.__path_to, self.__file_source))
            self.removefileIfExist('{}/{}'.format(self.__path_to, self.__file_source))
            if originalMD5 != md5val:
                return False, 'md5 check failed {}<>{}'.format(originalMD5, md5val)
        except OSError as why:
            return False,'Error: {} tranfer failed'.format(why.errno)
        except sh.ErrorReturnCode as shError:
            return False, 'USB Exception: tranfer failed'
        except Exception as details:
            return False, 'USB Exception: {} tranfer failed'.format(details)
        return True, ''

    def execute(self):
        v, m = self.checkfiles()
        if not v:
            self.fail(m)
        countLoops = int(self.__loops)
        while countLoops > 0:
            res, msg = self.ExecuteTranfer()
            if not res:
                res, msg = self.ExecuteTranfer()
                if not res:
                    self.fail(msg)
            countLoops = countLoops - 1

    def getresults(self):
        return self.__resultlist

    def gettextresult(self):
        return ""