#maybe need to import selenium here as well since selenium-webbrowser container connects to here
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
import numpy as np
import pandas as pd


#import extracting Dobby databank
import source.Dobby.Dobby as Dobby


'''
get all Dobby data
'''
def main():
    #Dobby.testDriver()
    driver = Dobby.getDriver()
    ppiXGroupNames = Dobby.findGroupNames(driver)
    np.savetxt("/var/lib/data/test.txt", [1, 2], delimiter=",", fmt='%s')
    print(ppiXGroupNames)

    #newDriver = Dobby.getDriver()
    persons = Dobby.getPersons(driver, ppiXGroupNames)
    print(persons)
    

    mitarbeiterAnzahl = len(persons)
    print("Anzahl der Mitarbeiter: " + str(mitarbeiterAnzahl))
    personsArray = np.array(list(persons))
    #print(personsArray)
    personsArrayTrans = personsArray.transpose()
    #print(personsArrayTrans)
    

    allNumbers = Dobby.getDobbyPersonData(driver, personsArray)
    print(allNumbers)
    allNumbersTransposed = allNumbers.transpose()

    #TODO: solve that I only use one driver and never close it
    driver.close()

    dictDobby = {'kuerzel': personsArrayTrans[0], 'name': personsArrayTrans[1], 'mobile': allNumbersTransposed[0]}
    print(dictDobby)
    dataframeDobby = pd.DataFrame.from_dict(dictDobby)




    dataframeDobby.to_csv('/var/lib/data/fullDobby.csv', index=False)
    np.savetxt("/var/lib/data/personsDobby.csv", personsArray, delimiter=",", fmt='%s')
#TODO: also have to save it as csv file


if __name__ == '__main__':
    main()