generate-mock-info

#!/usr/bin/env python3
from __future__ import annotations

import os
import sys
import unittest.mock


def main() -> int:
    assert sys.version_info[:2] == (3, 6), 'needs py36'

    mock = __import__('mock')  # avoid self-rewriting!

    mock_all, mock_version = mock.__all__, mock.__version__
    unittest_mock_all = unittest.mock.__all__

    attrs = set(mock_all).intersection(unittest_mock_all)

    renames_s = '\n'.join(
        sorted(
            f"    '{orig_mod}=unittest.mock:{attr}',"
            for orig_mod in ('mock', 'mock.mock')
            for attr in attrs
        ),
    )

    print(
        f'# GENERATED VIA {os.path.basename(sys.argv[0])}\n'
        f'# Using mock=={mock_version}\n'
        f'REPLACES[(3,)].update((\n{renames_s}\n))\n'
        f'# END GENERATED\n',
    )

    return 0


if __name__ == '__main__':
    raise SystemExit(main())