1 files changed, 57 insertions, 0 deletions
diff --git a/io_import_dxf/dxfgrabber/blockssection.py b/io_import_dxf/dxfgrabber/blockssection.py
new file mode 100755
index 00000000..d8f6b057
--- /dev/null
+++ b/io_import_dxf/dxfgrabber/blockssection.py
@@ -0,0 +1,57 @@
+# Purpose: blocks section
+# Created: 09.08.2012, taken from my package ezdxf
+# Copyright (C) 2011, Manfred Moitzi
+# License: MIT-License
+from __future__ import unicode_literals
+__author__ = "mozman <mozman@gmx.at>"
+
+from itertools import islice
+
+from .tags import TagGroups
+from .entitysection import build_entities
+
+
+class BlocksSection(object):
+    name = 'blocks'
+
+    def __init__(self):
+        self._blocks = dict()
+
+    @staticmethod
+    def from_tags(tags, drawing):
+        blocks_section = BlocksSection()
+        if drawing.grab_blocks:
+            blocks_section._build(tags, drawing.dxfversion)
+        return blocks_section
+
+    def _build(self, tags, dxfversion):
+        if len(tags) == 3:  # empty block section
+            return
+        groups = list()
+        for group in TagGroups(islice(tags, 2, len(tags)-1)):
+            groups.append(group)
+            if group[0].value == 'ENDBLK':
+                entities = build_entities(groups, dxfversion)
+                block = entities[0]
+                block.set_entities(entities[1:-1])
+                self._add(block)
+                groups = list()
+
+    def _add(self, block):
+        self._blocks[block.name] = block
+
+    # start of public interface
+    def __len__(self):
+        return len(self._blocks)
+
+    def __iter__(self):
+        return iter(self._blocks.values())
+
+    def __contains__(self, name):
+        return name in self._blocks
+
+    def __getitem__(self, name):
+        return self._blocks[name]
+
+    def get(self, name, default=None):
+        return self._blocks.get(name, default)