Business Outreach Automation

sudo apt update
sudo apt install golang-go

def unzip_file(self, zip_link: str) -> None:
    if self._find_scraper_dir():
        info("=> Scraper already unzipped. Skipping unzip.")
        return

    r = requests.get(zip_link)
    z = zipfile.ZipFile(io.BytesIO(r.content))
    for member in z.namelist():
        if ".." in member or member.startswith("/"):
            warning(f"Skipping suspicious path in archive: {member}")
            continue
        z.extract(member)

def start(self) -> None:
    # Check if go is installed
    if not self.is_go_installed():
        error("Go is not installed. Please install go and try again.")
        return

    # Unzip the scraper
    self.unzip_file(get_google_maps_scraper_zip_url())

    # Build the scraper
    self.build_scraper()

    # Write the niche to a file
    with open("niche.txt", "w") as f:
        f.write(self.niche)

    output_path = get_results_cache_path()
    message_subject = get_outreach_message_subject()
    message_body = get_outreach_message_body_file()

    # Run
    self.run_scraper_with_args_for_30_seconds(
        f'-input niche.txt -results "{output_path}"', timeout=get_scraper_timeout()
    )

def run_scraper_with_args_for_30_seconds(self, args: str, timeout=300) -> None:
    info(" => Running scraper...")
    binary_name = (
        "google-maps-scraper.exe"
        if platform.system() == "Windows"
        else "google-maps-scraper"
    )
    command = [os.path.join(os.getcwd(), binary_name)] + shlex.split(args)
    try:
        scraper_process = subprocess.run(command, timeout=float(timeout))

        if scraper_process.returncode == 0:
            print(colored("=> Scraper finished successfully.", "green"))
        else:
            print(colored("=> Scraper finished with an error.", "red"))
    except subprocess.TimeoutExpired:
        print(colored("=> Scraper timed out.", "red"))
    except Exception as e:
        print(colored("An error occurred while running the scraper:", "red"))
        print(str(e))

def set_email_for_website(self, index: int, website: str, output_file: str):
    # Extract and set an email for a website
    email = ""

    r = requests.get(website)
    if r.status_code == 200:
        # Define a regular expression pattern to match email addresses
        email_pattern = r"\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,7}\b"

        # Find all email addresses in the HTML string
        email_addresses = re.findall(email_pattern, r.text)

        email = email_addresses[0] if len(email_addresses) > 0 else ""

    if email:
        print(f"=> Setting email {email} for website {website}")
        with open(output_file, "r", newline="", errors="ignore") as csvfile:
            csvreader = csv.reader(csvfile)
            items = list(csvreader)
            items[index].append(email)

        with open(output_file, "w", newline="", errors="ignore") as csvfile:
            csvwriter = csv.writer(csvfile)
            csvwriter.writerows(items)

# Create a yagmail SMTP client outside the loop
yag = yagmail.SMTP(
    user=self.email_creds["username"],
    password=self.email_creds["password"],
    host=self.email_creds["smtp_server"],
    port=self.email_creds["smtp_port"],
)

for index, item in enumerate(items, start=1):
    try:
        # Check if the item's website is valid
        website = item.split(",")
        website = [w for w in website if w.startswith("http")]
        website = website[0] if len(website) > 0 else ""
        if website != "":
            test_r = requests.get(website)
            if test_r.status_code == 200:
                self.set_email_for_website(index, website, output_path)

                # Send emails using the existing SMTP connection
                receiver_email = item.split(",")[-1]

                if "@" not in receiver_email:
                    warning(f" => No email provided. Skipping...")
                    continue

                company_name = item.split(",")[0]
                subject = message_subject.replace(
                    "{{COMPANY_NAME}}", company_name
                )
                body = (
                    open(message_body, "r")
                    .read()
                    .replace("{{COMPANY_NAME}}", company_name)
                )

                info(f" => Sending email to {receiver_email}...")

                yag.send(
                    to=receiver_email,
                    subject=subject,
                    contents=body,
                )

                success(f" => Sent email to {receiver_email}")