




import re
import tempfile
from pathlib import Path

import pytest
import yaml

from datasets.utils.readme import ReadMe


# @pytest.fixture
# def example_yaml_structure():

UpperCAmelCase_						=yaml.safe_load(
    """\
name: \"\"
allow_empty: false
allow_empty_text: true
subsections:
  - name: \"Dataset Card for X\" # First-level markdown heading
    allow_empty: false
    allow_empty_text: true
    subsections:
      - name: \"Table of Contents\"
        allow_empty: false
        allow_empty_text: false
        subsections: null
      - name: \"Dataset Description\"
        allow_empty: false
        allow_empty_text: false
        subsections:
          - name: \"Dataset Summary\"
            allow_empty: false
            allow_empty_text: false
            subsections: null
          - name: \"Supported Tasks and Leaderboards\"
            allow_empty: true
            allow_empty_text: true
            subsections: null
          - name: Languages
            allow_empty: false
            allow_empty_text: true
            subsections: null
"""
)


UpperCAmelCase_						={
    """name""": """root""",
    """text""": """""",
    """is_empty_text""": True,
    """subsections""": [
        {
            """name""": """Dataset Card for My Dataset""",
            """text""": """""",
            """is_empty_text""": True,
            """subsections""": [
                {"""name""": """Table of Contents""", """text""": """Some text here.""", """is_empty_text""": False, """subsections""": []},
                {
                    """name""": """Dataset Description""",
                    """text""": """Some text here.""",
                    """is_empty_text""": False,
                    """subsections""": [
                        {
                            """name""": """Dataset Summary""",
                            """text""": """Some text here.""",
                            """is_empty_text""": False,
                            """subsections""": [],
                        },
                        {
                            """name""": """Supported Tasks and Leaderboards""",
                            """text""": """""",
                            """is_empty_text""": True,
                            """subsections""": [],
                        },
                        {"""name""": """Languages""", """text""": """Language Text""", """is_empty_text""": False, """subsections""": []},
                    ],
                },
            ],
        }
    ],
}


UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""


UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
#### Extra Ignored Subsection
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						={
    """name""": """root""",
    """text""": """""",
    """is_empty_text""": True,
    """subsections""": [
        {
            """name""": """Dataset Card for My Dataset""",
            """text""": """""",
            """is_empty_text""": True,
            """subsections""": [
                {"""name""": """Table of Contents""", """text""": """Some text here.""", """is_empty_text""": False, """subsections""": []},
                {
                    """name""": """Dataset Description""",
                    """text""": """Some text here.""",
                    """is_empty_text""": False,
                    """subsections""": [
                        {
                            """name""": """Dataset Summary""",
                            """text""": """Some text here.""",
                            """is_empty_text""": False,
                            """subsections""": [
                                {
                                    """name""": """Extra Ignored Subsection""",
                                    """text""": """""",
                                    """is_empty_text""": True,
                                    """subsections""": [],
                                }
                            ],
                        },
                        {
                            """name""": """Supported Tasks and Leaderboards""",
                            """text""": """""",
                            """is_empty_text""": True,
                            """subsections""": [],
                        },
                        {"""name""": """Languages""", """text""": """Language Text""", """is_empty_text""": False, """subsections""": []},
                    ],
                },
            ],
        }
    ],
}

UpperCAmelCase_						="""\
---
---
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						=(
    """The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README."""
)

UpperCAmelCase_						="""\
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						=(
    """The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README."""
)

UpperCAmelCase_						="""\
---
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
### Supported Tasks and Leaderboards
### Languages
Language Text
"""
UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Summary` but it is empty.\n-\tExpected some text in section `Dataset Summary` but it is empty (text in subsections are ignored)."""


UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
"""
UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Card for My Dataset` but it is empty.\n-\tSection `Dataset Card for My Dataset` expected the following subsections: `Table of Contents`, `Dataset Description`. Found 'None'."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Languages
Language Text
"""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`."""


UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
"""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Languages` but it is empty."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""
UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
# Dataset Card My Dataset
"""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tThe README has several first-level headings: `Dataset Card for My Dataset`, `Dataset Card My Dataset`. Only one heading is expected. Skipping further validation for this README."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tNo first-level heading starting with `Dataset Card for` found in README. Skipping further validation for this README."""

UpperCAmelCase_						=""""""

UpperCAmelCase_						="""The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README.\n-\tNo YAML markers are present in the README."""

UpperCAmelCase_						="""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

UpperCAmelCase_						="""The following issues were found while parsing the README at `{path}`:\n-\tMultiple sections with the same heading `Dataset Card for My Dataset` have been found. Please keep only one of these sections."""



@pytest.mark.parametrize(
    '''readme_md, expected_dict'''						, [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							assert ReadMe.from_string(_snake_case						, _snake_case							).to_dict() == expected_dict



@pytest.mark.parametrize(
    '''readme_md, expected_error'''						, [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							with pytest.raises(_snake_case						, match=re.escape(expected_error.format(path='''root'''							)							)							):
														lowerCAmelCase   =		ReadMe.from_string(_snake_case						, _snake_case							)
														readme.validate()



@pytest.mark.parametrize(
    '''readme_md, expected_error'''						, [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							with pytest.raises(_snake_case						, match=re.escape(expected_error.format(path='''root'''							)							)							):
														ReadMe.from_string(_snake_case						, _snake_case							)



@pytest.mark.parametrize(
    '''readme_md,'''						, [
        (README_MULTIPLE_SAME_HEADING_1),
    ]						, )
def       UpperCAmelCase       (     _snake_case							):
							ReadMe.from_string(_snake_case						, _snake_case						, suppress_parsing_errors=_snake_case							)



@pytest.mark.parametrize(
    '''readme_md, expected_dict'''						, [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							with tempfile.TemporaryDirectory() as tmp_dir:
														lowerCAmelCase   =		Path(_snake_case							) / '''README.md'''
														with open(_snake_case						, '''w+'''							) as readme_file:
																					readme_file.write(_snake_case							)
														lowerCAmelCase   =		ReadMe.from_readme(_snake_case						, _snake_case							).to_dict()
														assert out["name"] == path
														assert out["text"] == ""
														assert out["is_empty_text"]
														assert out["subsections"] == expected_dict["subsections"]



@pytest.mark.parametrize(
    '''readme_md, expected_error'''						, [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							with tempfile.TemporaryDirectory() as tmp_dir:
														lowerCAmelCase   =		Path(_snake_case							) / '''README.md'''
														with open(_snake_case						, '''w+'''							) as readme_file:
																					readme_file.write(_snake_case							)
														lowerCAmelCase   =		expected_error.format(path=_snake_case							)
														with pytest.raises(_snake_case						, match=re.escape(_snake_case							)							):
																					lowerCAmelCase   =		ReadMe.from_readme(_snake_case						, _snake_case							)
																					readme.validate()



@pytest.mark.parametrize(
    '''readme_md, expected_error'''						, [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]						, )
def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							with tempfile.TemporaryDirectory() as tmp_dir:
														lowerCAmelCase   =		Path(_snake_case							) / '''README.md'''
														with open(_snake_case						, '''w+'''							) as readme_file:
																					readme_file.write(_snake_case							)
														lowerCAmelCase   =		expected_error.format(path=_snake_case							)
														with pytest.raises(_snake_case						, match=re.escape(_snake_case							)							):
																					ReadMe.from_readme(_snake_case						, _snake_case							)



@pytest.mark.parametrize(
    '''readme_md,'''						, [
        (README_MULTIPLE_SAME_HEADING_1),
    ]						, )
def       UpperCAmelCase       (     _snake_case							):
							with tempfile.TemporaryDirectory() as tmp_dir:
														lowerCAmelCase   =		Path(_snake_case							) / '''README.md'''
														with open(_snake_case						, '''w+'''							) as readme_file:
																					readme_file.write(_snake_case							)
														ReadMe.from_readme(_snake_case						, _snake_case						, suppress_parsing_errors=_snake_case							)









from collections import Counter
from pathlib import Path
from typing import Optional, Tuple

import yaml


class 							__UpperCamelCase			( yaml.SafeLoader  ):





				'''simple docstring'''



				def       __snake_case		( self       , UpperCAmelCase_						):
											lowerCAmelCase   =		[self.constructed_objects[key_node] for key_node, _ in node.value]
											lowerCAmelCase   =		[tuple(UpperCAmelCase_						) if isinstance(UpperCAmelCase_       , UpperCAmelCase_						) else key for key in keys]
											lowerCAmelCase   =		Counter(UpperCAmelCase_						)
											lowerCAmelCase   =		[key for key in counter if counter[key] > 1]
											if duplicate_keys:
																		raise TypeError(F"""Got duplicate yaml keys: {duplicate_keys}"""						)






				def       __snake_case		( self       , UpperCAmelCase_       , UpperCAmelCase_=False						):
											lowerCAmelCase   =		super().construct_mapping(UpperCAmelCase_       , deep=UpperCAmelCase_						)
											self._check_no_duplicates_on_constructed_node(UpperCAmelCase_						)
											return mapping




def       UpperCAmelCase       (     _snake_case							):
							lowerCAmelCase   =		list(readme_content.splitlines()							)
							if full_content and full_content[0] == "---" and "---" in full_content[1:]:
														lowerCAmelCase   =		full_content[1:].index('''---'''							) + 1
														lowerCAmelCase   =		'''\n'''.join(full_content[1:sep_idx]							)
														return yamlblock, "\n".join(full_content[sep_idx + 1 :]							)

							return None, "\n".join(_snake_case							)


class 							__UpperCamelCase			( __UpperCAmelCase  ):





				'''simple docstring'''



				__a							: Any										={"""train_eval_index"""}  # train-eval-index in the YAML metadata



				@classmethod
				def       __snake_case		( cls       , UpperCAmelCase_						):
											with open(UpperCAmelCase_       , encoding='''utf-8'''						) as readme_file:
																		lowerCAmelCase   ,     lowerCAmelCase   =		_split_yaml_from_readme(readme_file.read()						)
											if yaml_string is not None:
																		return cls.from_yaml_string(UpperCAmelCase_						)
											else:
																		return cls()



				def       __snake_case		( self       , UpperCAmelCase_						):
											if path.exists():
																		with open(UpperCAmelCase_       , encoding='''utf-8'''						) as readme_file:
																									lowerCAmelCase   =		readme_file.read()
											else:
																		lowerCAmelCase   =		None
											lowerCAmelCase   =		self._to_readme(UpperCAmelCase_						)
											with open(UpperCAmelCase_       , '''w'''       , encoding='''utf-8'''						) as readme_file:
																		readme_file.write(UpperCAmelCase_						)



				def       __snake_case		( self       , UpperCAmelCase_ = None						):
											if readme_content is not None:
																		lowerCAmelCase   ,     lowerCAmelCase   =		_split_yaml_from_readme(UpperCAmelCase_						)
																		lowerCAmelCase   =		'''---\n''' + self.to_yaml_string() + '''---\n''' + content
											else:
																		lowerCAmelCase   =		'''---\n''' + self.to_yaml_string() + '''---\n'''
											return full_content



				@classmethod
				def       __snake_case		( cls       , UpperCAmelCase_						):
											lowerCAmelCase   =		yaml.load(UpperCAmelCase_       , Loader=_NoDuplicateSafeLoader						) or {}

											# Convert the YAML keys to DatasetMetadata fields
											lowerCAmelCase   =		{
											    (key.replace('''-'''       , '''_'''						) if key.replace('''-'''       , '''_'''						) in cls._FIELDS_WITH_DASHES else key): value
											    for key, value in metadata_dict.items()
											}
											return cls(**UpperCAmelCase_						)






				def       __snake_case		( self						):
											return yaml.safe_dump(
											    {
											        (key.replace('''_'''       , '''-'''						) if key in self._FIELDS_WITH_DASHES else key): value
											        for key, value in self.items()
											    }       , sort_keys=UpperCAmelCase_       , allow_unicode=UpperCAmelCase_       , encoding='''utf-8'''       , ).decode('''utf-8'''						)




UpperCAmelCase_						={
    """image-classification""": [],
    """translation""": [],
    """image-segmentation""": [],
    """fill-mask""": [],
    """automatic-speech-recognition""": [],
    """token-classification""": [],
    """sentence-similarity""": [],
    """audio-classification""": [],
    """question-answering""": [],
    """summarization""": [],
    """zero-shot-classification""": [],
    """table-to-text""": [],
    """feature-extraction""": [],
    """other""": [],
    """multiple-choice""": [],
    """text-classification""": [],
    """text-to-image""": [],
    """text2text-generation""": [],
    """zero-shot-image-classification""": [],
    """tabular-classification""": [],
    """tabular-regression""": [],
    """image-to-image""": [],
    """tabular-to-text""": [],
    """unconditional-image-generation""": [],
    """text-retrieval""": [],
    """text-to-speech""": [],
    """object-detection""": [],
    """audio-to-audio""": [],
    """text-generation""": [],
    """conversational""": [],
    """table-question-answering""": [],
    """visual-question-answering""": [],
    """image-to-text""": [],
    """reinforcement-learning""": [],
    """voice-activity-detection""": [],
    """time-series-forecasting""": [],
    """document-question-answering""": [],
}


if __name__ == "__main__":
		from argparse import ArgumentParser

		UpperCAmelCase_						=ArgumentParser(usage="""Validate the yaml metadata block of a README.md file.""")
		ap.add_argument("""readme_filepath""")
		UpperCAmelCase_						=ap.parse_args()

		UpperCAmelCase_						=Path(args.readme_filepath)
		UpperCAmelCase_						=DatasetMetadata.from_readme(readme_filepath)
		print(dataset_metadata)
		dataset_metadata.to_readme(readme_filepath)



