run.py

""" Modules import """
from datetime import datetime
from cryptography.fernet import Fernet

from src.driver import DriverPreparation
from src.services import Linkedin, Vagas



def set_driver():
    """
    Returns url path from webdriver directory.

    """

    driver = DriverPreparation()

    chrome_driver = driver.chrome_driver()

    return chrome_driver


def credentials(service):
    """
    Returns login credentials.

    """

    with open('C:/Users/mekyl/OneDrive/Documentos/chave.key', 'rb') as key_file:
        key = key_file.read()

    with open('./src/credentials_' + service + '.txt', 'r') as credentials_file:
        lines = credentials_file.readlines()

    fernet = Fernet(key)

    return [fernet.decrypt(line.encode()).decode() for line in lines]


def use_linkedin(driver, email, password):
    """
    Run LinkedIn process.

    """

    linkedin = Linkedin(driver, email, password)
    linkedin.login()
    linkedin.logout()


def use_vagas(driver, email, password, job_description):
    """
    Run Vagas process.

    """

    vagas = Vagas(driver, email, password)
    vagas.login()
    applications = vagas.job_application(*job_description)
    save_info_application(applications)
    vagas.logout()


def save_info_application(informations):
    """
    Create a file with informations about job application.
    """
    date = datetime.now().strftime('%Y-%m-%d-%H%M%S')

    with open('./applications/applications' + date + '.txt', 'w') as file:
        for information in informations:
            for key, value in information.items():
                info = key + ': ' + value.replace('\n', '')
                file.writelines(info + '\n')
            file.writelines('\n')


def main(job_description, service='vagas'):
    """
    Main function.

    """

    driver = set_driver()
    email, password = [*credentials(service)]

    if service.lower().strip() == 'linkedin':
        return use_linkedin(driver, email, password)
    return use_vagas(driver, email, password, job_description)


if __name__ == '__main__':
    main(job_description=['Coordenador 3D'])