Proposal.py

import csv
import time
from selenium import webdriver
from selenium.webdriver.common.by import By

import Const
import UserConfig.UserConfig as UserConfig

from Filter import Filter
from ProposalInfo import ProposalInfo

from DynamoDb.DynamoDb import DynamoDb
from Models.PofileInfo import ProfileInfo
from UserConfig.Script import Script
from Utils.DateTime import DateTime


class Proposal:
    def __init__(self, driver: webdriver.Chrome):
        self.driver = driver

    def open_latest_proposal(self):
        self.driver.find_element(
            By.CSS_SELECTOR, "[data-test-service-request-details__accept]"
        ).click()

    def decline_latest_proposal(self):
        self.driver.find_element(
            By.CSS_SELECTOR, "[data-test-service-requests-detail__decline]"
        ).click()

        buttons = self.driver.find_elements(By.TAG_NAME, "button")

        for button in buttons:
            if (
                "Decline".lower().strip()
                in button.get_attribute("innerText").lower().strip()
            ):
                button.click()
                break

    def enter_proposal_script(self):
        self.driver.find_element(By.TAG_NAME, "textarea").send_keys(
            Script().getScript()
        )

    def check_consultation(self):
        if UserConfig.FREE_CONSULTATION:
            self.driver.find_element(
                By.CSS_SELECTOR, "[data-test-text-selectable-option__label]"
            ).click()
            print("Consultation clicked")

    def submit(self):
        self.driver.find_element(
            By.CSS_SELECTOR, "[data-test-proposal-submission-modal__submit-button]"
        ).click()

    def get_proposal_link(self) -> str:
        try:
            links = self.driver.find_element(
                By.CSS_SELECTOR,
                "[data-test-service-marketplace-proposal-submission-success-modal__engagement-success]",
            ).find_elements(By.TAG_NAME, "a")
        except Exception as exception:
            print(exception)
            return ""

        for link in links:
            if (
                "View project".lower().strip()
                in link.get_attribute("innerText").lower().strip()
            ):
                return link.get_attribute("href").lower().strip()

    def save_proposal_details(self, info: ProfileInfo, status: str, notes: str):
        # Pass same time to both updates?
        self.save_to_text_file(info, status, notes)
        self.save_to_database(info, status, notes)

    def save_to_database(self, info: ProfileInfo, status: str, notes: str):
        dynamo_db = DynamoDb(UserConfig.USER_ID, Const.TABLE_NAME, UserConfig.REGION)
        dynamo_db.put_item(info, status, notes)

    def save_to_text_file(self, info: ProfileInfo, status: str, notes: str):
        with open("sent_proposals.csv", "a", newline="", encoding="utf-8") as file:
            writer = csv.writer(file)
            writer.writerow(
                [
                    UserConfig.USER_ID,
                    DateTime().get_time_formatted(),
                    info.name,
                    info.service_type,
                    info.location,
                    info.profile_description,
                    info.profile_link,
                    status,
                    notes,
                ]
            )

    def main(self):
        proposal_info = ProposalInfo(self.driver).get_proposal_info()

        print("Proposal Details:")
        print(f"Name: {proposal_info.name}")
        print(f"Service Type: {proposal_info.service_type}")
        print(f"Location: {proposal_info.location}")
        print(f"Profile Description: {proposal_info.profile_description}")
        print(f"Profile Link: {proposal_info.profile_link}")
        print("\n")

        filter_result = Filter(self.driver).filter_questions()

        if filter_result:
            print("Proposal filters not met\n")
            self.decline_latest_proposal()
            self.save_proposal_details(proposal_info, "Rejected", filter_result)
            print("Recorded to text file and database\n")
            print(f"Proposal rejected on {DateTime().get_time_formatted()}")
        else:
            print("Proposal filters met\n")
            self.open_latest_proposal()
            time.sleep(5)
            self.enter_proposal_script()
            print("Script entered")
            self.check_consultation()
            self.submit()
            time.sleep(5)
            self.save_proposal_details(
                proposal_info, "Accepted", self.get_proposal_link()
            )
            print("Recorded to text file and database\n")
            print(f"Proposal submitted on {DateTime().get_time_formatted()}")