app.py

import logging as log
import time
from getpass import getpass
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.common.exceptions import NoSuchWindowException, TimeoutException
from termcolor import colored
from base import (
    clear_cmd,
    site_login,
    get_driver,
    close_shop,
    click_element,
    random_time,
    get_length_of_page,
    start_end_log,
    check_session,
)
from config import Settings


def delete_post(driver):
    """
    Delete a post from the user's profile.
    This function finds and clicks the more options, delete, and delete confirmation button.

    Args:
        driver: The Selenium WebDriver instance.
    Returns:
        bool: True if the post was deleted successfully, False otherwise.
    """
    try:
        # Wait for load and find the three-dot menu (more options)
        wait = WebDriverWait(driver, 15)
        menu_btn = wait.until(
            EC.element_to_be_clickable(
                (By.CSS_SELECTOR, "svg[aria-label='More options']")
            )
        )
        if not menu_btn:
            log.info("Options button not found.\nSkipping post...")
            return False
        click_element(driver, menu_btn, "More options")
        random_time()

        # Click delete option
        delete_btn = wait.until(
            EC.element_to_be_clickable(
                (By.CSS_SELECTOR, "button:not([disabled])[tabindex='0']")
            )
        )
        if not delete_btn or "Delete" not in delete_btn.text:
            log.info("Delete button not found.\nSkipping post...")
            return False
        click_element(driver, delete_btn, "Delete")
        random_time()

        # Click confirm deletion option
        confirm_btn = wait.until(
            EC.element_to_be_clickable(
                (By.CSS_SELECTOR, "button:not([disabled])[tabindex='0']")
            )
        )
        if not confirm_btn or "Delete" not in confirm_btn.text:
            log.info("Confirm delete button not found or incorrect.")
            return False
        click_element(driver, confirm_btn, "Confirm Delete")
        random_time()
        log.info("Post deleted successfully.")
        return True
    except Exception as ex:
        log.error(f"Error deleting post: {ex}", exc_info=True)
        return False


def main():
    start_end_log(__file__)
    driver = None
    username = None
    password = None
    posts_deleted = 0
    settings = Settings()
    max_posts = settings.max_posts
    max_scrolls = settings.max_scrolls
    try:
        clear_cmd()
        # ASCII art banner
        banner = colored(
            """
     ___           _      __        ___            
    |_ _|_ __  ___| |_ __ \\ \\      / (_)_ __   ___ 
     | || '_ \\/ __| __/ _` \\ \\ /\\ / /| | '_ \\ / _ \\
     | || | | \\__ \\ || (_| |\\ V  V / | | |_) |  __/
    |___|_| |_|___/\\__\\__,_| \\_/\\_/  |_| .__/ \\___|
                                       |_|         
              https://github.com/ajwdd
        """,
            color="magenta",
            attrs=["bold"],
        )
        print(banner)

        # Get user's Instagram credentials
        print("Enter your Instagram credentials.")
        username = input("Username: ").strip().lower()
        password = getpass("Password: ").strip()
        if not username or not password:
            warning = colored(
                "Username or password cannot be empty!", color="red", attrs=["blink"]
            )
            print(warning)
            input("Press any key to retry...")
            main()

        driver = get_driver()
        driver = site_login(driver, username, password)

        # Navigate to profile and wait for load
        driver.get(f"https://www.instagram.com/{username}/")
        wait = WebDriverWait(driver, 20)
        wait.until(
            EC.presence_of_element_located(
                (By.XPATH, f"//a[contains(@href, '/{username}/p/')]")
            )
        )
        random_time()

        while posts_deleted < max_posts:
            # Check session validity
            if not check_session(driver):
                log.info("Session invalid, restarting driver")
                close_shop(driver)
                driver = get_driver()
                driver = site_login(driver, username, password)
                driver.get(f"https://www.instagram.com/{username}/")
                wait.until(
                    EC.presence_of_element_located(
                        (By.XPATH, f"//a[contains(@href, '/{username}/p/')]")
                    )
                )
                random_time()

            # Scroll multiple times to load posts
            last_height = get_length_of_page(driver)
            for _ in range(max_scrolls):
                driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
                time.sleep(2)
                new_height = get_length_of_page(driver)
                if new_height == last_height:
                    log.info("No new posts loaded after scrolling.")
                    break
                last_height = new_height

            # Wait for post links to load (with retry)
            posts = []
            for attempt in range(2):
                try:
                    wait = WebDriverWait(driver, 20)
                    posts = wait.until(
                        EC.presence_of_all_elements_located(
                            (By.XPATH, f"//a[contains(@href, '/{username}/p/')]")
                        )
                    )
                    break
                except Exception as ex:
                    log.warning(
                        f"Attempt {attempt + 1}: No posts found, retrying... {ex}"
                    )
                    time.sleep(5)
            log.info(f"Found {len(posts)} posts")

            if not posts:
                log.info("No more posts found")
                break

            # Process the first available post
            try:
                post = posts[0]
                click_element(driver, post, "Post")
                random_time()
                if delete_post(driver):
                    posts_deleted += 1
            except Exception as ex:
                log.error(f"Error processing post: {ex}", exc_info=True)

            # Navigate back and wait for profile to reload
            driver.get(f"https://www.instagram.com/{username}/")
            try:
                wait.until(
                    EC.presence_of_element_located(
                        (By.XPATH, f"//a[contains(@href, '/{username}/p/')]")
                    )
                )
            except TimeoutException:
                log.info("No posts found after navigation.\nExiting...")
                break
            random_time()

        log.info(f"Deleted {posts_deleted} posts")

    except NoSuchWindowException as ex:
        log.error(f"Browser window closed unexpectedly:\n{ex}", exc_info=True)
    except Exception as ex:
        log.error(f"Error in main:\n{ex}", exc_info=True)
    finally:
        if driver:
            close_shop(driver)
        start_end_log(__file__, end_log=True)


if __name__ == "__main__":
    main()