import pdb

import pytest
import warnings

from .fixtures import nwb_core_fixture, tmp_output_dir
from linkml_runtime.dumpers import yaml_dumper
from linkml.generators import PydanticGenerator

from nwb_linkml.lang_elements import NwbLangSchema

def test_generate_nwblang(tmp_output_dir):
    output_file = (tmp_output_dir / NwbLangSchema.name).with_suffix('.yml')
    yaml_dumper.dump(NwbLangSchema, output_file)

def test_generate_core(nwb_core_fixture, tmp_output_dir):
    schemas = nwb_core_fixture.build().schemas
    for schema in schemas:
        output_file = tmp_output_dir / (schema.name + '.yaml')
        yaml_dumper.dump(schema, output_file)

@pytest.mark.depends(on=['test_generate_core'])
def test_generate_pydantic(tmp_output_dir):


    # core_file = tmp_output_dir / 'core.yaml'
    # pydantic_file = tmp_output_dir / 'core.py'

    for schema in tmp_output_dir.glob('*.yaml'):
        pydantic_file = (schema.parent / schema.name).with_suffix('.py')

        generator = PydanticGenerator(
            str(schema),
            pydantic_version='1',
            emit_metadata=True,
            gen_classvars=True,
            gen_slots=True
        )
        gen_pydantic = generator.serialize()
        with open(pydantic_file, 'w') as pfile:
            pfile.write(gen_pydantic)