technical_drawings_extraction/venv/lib/python3.7/site-packages/ezdxf/tools/test.py

# Purpose: test tools
# Created: 27.03.2011
# Copyright (c) 2011-2018, Manfred Moitzi
# License: MIT License
from typing import Sequence, TYPE_CHECKING, Iterable, List, Set
from ezdxf.lldxf.tagger import internal_tag_compiler
from ezdxf.modern import ModernDXFFactory

if TYPE_CHECKING:  # import forward declarations
    from ezdxf.eztypes import DXFTag, EntityDB, ExtendedTags, DXFEntity


def compile_tags_without_handles(text: str) -> Iterable['DXFTag']:
    return (tag for tag in internal_tag_compiler(text) if tag.code not in (5, 105))


def normlines(text: str) -> Sequence[str]:
    lines = text.split('\n')
    return [line.strip() for line in lines]


def load_section(text: str, name: str, database: 'EntityDB' = None, dxfversion='AC1009') -> List['ExtendedTags']:
    from ezdxf.lldxf.loader import load_dxf_structure, fill_database
    dxf = load_dxf_structure(internal_tag_compiler(text), ignore_missing_eof=True)
    if database is not None:
        fill_database(database, dxf, dxfversion)
    return dxf[name]


SUPPORTED_ENTITIES = ModernDXFFactory(None).ENTITY_WRAPPERS


def find_unsupported_entities(container: Iterable['DXFEntity']) -> Set[str]:
    unsupported_entities = set()
    for entity in container:
        dxftype = entity.dxftype()
        if dxftype not in SUPPORTED_ENTITIES:
            unsupported_entities.add(dxftype)
    return unsupported_entities