main.py

import json
import sys
import black

def check_code_snippet(code_snippet: str):
    try:
        formatted_code = black.format_str(code_snippet, mode=black.FileMode())
    except Exception as e:
        return {
            'status': 'error',
            'error': str(e),
        }
    if formatted_code.strip() == code_snippet.strip():
        return {
            'status': 'success',
        }
    return {
        'status': 'updated',
        'newCode': formatted_code,
    }


def main():
    code_snippets_path = sys.argv[1]
    if not code_snippets_path:
        print("No code snippets path provided")
        return
    code_snippets = json.load(open(code_snippets_path))
    formatted_codes = [check_code_snippet(snippet["code"]) for snippet in code_snippets]
    print(json.dumps(formatted_codes))


if __name__ == "__main__":
    main()