1 files changed, 47 insertions, 0 deletions
diff --git a/io_import_dxf/dxfgrabber/dxfattr.py b/io_import_dxf/dxfgrabber/dxfattr.py
new file mode 100755
index 00000000..a98f038d
--- /dev/null
+++ b/io_import_dxf/dxfgrabber/dxfattr.py
@@ -0,0 +1,47 @@
+# Purpose: define dxf attributes
+# Created: 21.07.2012, taken from my ezdxf project
+# Copyright (C) 2012, Manfred Moitzi
+# License: MIT License
+from __future__ import unicode_literals
+__author__ = "mozman <mozman@gmx.at>"
+
+from collections import namedtuple
+from .const import XTYPE_NONE
+
+
+def DXFAttr(code, xtype=XTYPE_NONE):
+    # assert type(xtype) is int
+    return _DXFAttr(code, xtype)
+
+_DXFAttr = namedtuple('DXFAttr', 'code xtype')
+DXFAttr3 = namedtuple('DXFAttr3', 'code xtype subclass')
+DefSubclass = namedtuple('DefSubclass', 'name attribs')
+
+
+class DXFAttributes(object):
+    def __init__(self, *subclassdefs):
+        self._subclasses = []
+        self._attribs = {}
+        for subclass in subclassdefs:
+            self.add_subclass(subclass)
+            
+    def add_subclass(self, subclass):
+        subclass_index = len(self._subclasses)
+        self._subclasses.append(subclass)
+        self._add_subclass_attribs(subclass, subclass_index)
+        
+    def _add_subclass_attribs(self, subclass, subclass_index):
+        for name, dxfattrib in subclass.attribs.items():
+            self._attribs[name] = DXFAttr3(dxfattrib.code, dxfattrib.xtype, subclass_index)
+
+    def __getitem__(self, name):
+        return self._attribs[name]
+
+    def __contains__(self, name):
+        return name in self._attribs
+
+    def keys(self):
+        return iter(self._attribs.keys())
+    
+    def subclasses(self):
+        return iter(self._subclasses)