import sys
sys.path.insert(0, __file__.rsplit("/", 1)[0])

import golden


def pytest_addoption(parser):
    parser.addoption("--diff", action="store_true", help="Show diff for golden file mismatches")
    parser.addoption("--accept", action="store_true", help="Accept golden file changes")


def pytest_sessionfinish(session):
    if golden._FAILED:
        if session.config.getoption("--accept"):
            golden._accept_actual(golden._FAILED)
        if session.config.getoption("--diff"):
            golden._show_differences(golden._FAILED)