Python loader_forの例

コード例 #1

ファイル: test_loaders.py プロジェクト: calpt/flask-filealchemy

def test_model_for_single_file(db, tmpdir):
    authors = tmpdir.mkdir('authors')
    authors.join('_all.yml').write('does-not-matter')

    class Author(db.Model):
        __tablename__ = 'authors'

        slug = Column(String(255), primary_key=True)
        name = Column(String(255), nullable=False)

    assert len(db.metadata.sorted_tables) == 1
    assert db.metadata.sorted_tables[0].name == 'authors'

    author_table = db.metadata.sorted_tables[0]

    assert isinstance(loader_for(Path(tmpdir.strpath), author_table),
                      YAMLSingleFileLoader)

コード例 #2

ファイル: test_loaders.py プロジェクト: calpt/flask-filealchemy

def test_model_for_directory_extra_files(db, tmpdir):
    authors = tmpdir.mkdir('authors')

    authors.join('invalid.md').write('does-not-matter')
    authors.join('valid.yml').write('does-not-matter')

    class Author(db.Model):
        __tablename__ = 'authors'

        slug = Column(String(255), primary_key=True)
        name = Column(String(255), nullable=False)

    assert len(db.metadata.sorted_tables) == 1
    assert db.metadata.sorted_tables[0].name == 'authors'

    author_table = db.metadata.sorted_tables[0]

    assert not loader_for(Path(tmpdir.strpath), author_table)

コード例 #3

ファイル: test_loaders.py プロジェクト: siddhantgoel/flask-filealchemy

def test_yaml_directory_loader(db, tmpdir):
    authors = tmpdir.mkdir('authors')

    authors.join('first.yml').write('does-not-matter')
    authors.join('second.yml').write('does-not-matter')

    class Author(db.Model):
        __tablename__ = 'authors'

        slug = Column(String(255), primary_key=True)
        name = Column(String(255), nullable=False)

    assert len(db.metadata.sorted_tables) == 1
    assert db.metadata.sorted_tables[0].name == 'authors'

    author_table = db.metadata.sorted_tables[0]

    assert isinstance(loader_for(Path(tmpdir.strpath), author_table),
                      YAMLDirectoryLoader)

コード例 #4

ファイル: test_loaders.py プロジェクト: calpt/flask-filealchemy

def test_model_for_directory_extra_extensions(db, tmpdir):
    authors = tmpdir.mkdir('authors')

    for index, extension in enumerate(YAMLDirectoryLoader.extensions):
        authors.join('authors-{}.{}'.format(
            index, extension)).write('does-not-matter')

    class Author(db.Model):
        __tablename__ = 'authors'

        slug = Column(String(255), primary_key=True)
        name = Column(String(255), nullable=False)

    assert len(db.metadata.sorted_tables) == 1
    assert db.metadata.sorted_tables[0].name == 'authors'

    author_table = db.metadata.sorted_tables[0]

    assert isinstance(loader_for(Path(tmpdir.strpath), author_table),
                      YAMLDirectoryLoader)