import mysql.connector

# Database connection details
db_config = {
    "host": "127.0.0.1",  # e.g., "localhost"
    "user": "sql_seo2024_ir",
    "database": "sql_seo2024_ir",
    "password": "1bd0d20735dfd8",
}

# Output file
output_file = "importer/imovieids.txt"

def fetch_and_save_postmeta():
    try:
        # Connect to the database
        connection = mysql.connector.connect(**db_config)
        cursor = connection.cursor()

        # Query to fetch rows where meta_value starts with 'tt'
        query = """
        SELECT post_id, meta_value 
        FROM wp_postmeta
        WHERE meta_value LIKE 'tt%'
        """
        cursor.execute(query)

        # Fetch all results
        results = cursor.fetchall()

        # Write results to the file
        with open(output_file, "w", encoding="utf-8") as file:
            for post_id, meta_value in results:
                file.write(f"{post_id};{meta_value}\n")

        print(f"Data successfully written to {output_file}")

    except mysql.connector.Error as err:
        print(f"Error: {err}")

    finally:
        # Clean up
        if cursor:
            cursor.close()
        if connection:
            connection.close()

if __name__ == "__main__":
    fetch_and_save_postmeta()