import pydantic
import os
import fileinput

def replace_string_in_files(folder_path, old_str, new_str):
    for subdir, dirs, files in os.walk(folder_path):
        for file in files:
            file_path = os.path.join(subdir, file)

            # Check if the file is a text file (you can modify this condition based on your needs)
            if file.endswith(".txt") or file.endswith(".py"):
                # Open the file in place for editing
                with fileinput.FileInput(file_path, inplace=True) as f:
                    for line in f:
                        # Replace the old string with the new string
                        print(line.replace(old_str, new_str), end='')


def use_pydantic_v1():
    module_file_path = pydantic.__file__
    module_file_path = module_file_path.split('pydantic')[0] + 'haystack'  
    with open(module_file_path+'/schema.py','r') as f:
        haystack_schema_file = f.read()

    if 'from pydantic.v1' not in haystack_schema_file:
        replace_string_in_files(module_file_path, 'from pydantic', 'from pydantic.v1')