#!/usr/bin/env python3
import sys
import re
import os

def filter_inserts(input_file):
    insert_pattern = re.compile(r'^\s*INSERT\s+INTO', re.IGNORECASE)
    output_file = os.path.splitext(input_file)[0] + "_inserts.sql"

    with open(input_file, "r", encoding="utf-8") as infile, \
         open(output_file, "w", encoding="utf-8") as outfile:
        
        buffer = []
        for line in infile:
            buffer.append(line)
            if ";" in line:  # fin d'une requête SQL
                query = "".join(buffer).strip()
                if insert_pattern.match(query):
                    outfile.write(query + "\n")
                buffer = []  # réinitialiser

    print(f"✅ Fichier généré : {output_file}")


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python data-clear.py <fichier.sql>")
        sys.exit(1)

    input_file = sys.argv[1]
    filter_inserts(input_file)