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'

    generator = PydanticGenerator(
        str(core_file),
        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)