X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=lib%2Ferd_handler%2Ferd.rb;h=bf740072d432b883cf7e58d891d56c4431418051;hb=df86d6dec3f5944842cf8e5cdcaa909bfc7e0406;hp=142f575d8851044d9299090b069599526aad7480;hpb=5cb26987412346fc34cb344d7b169c0784f1da77;p=erd-marker.git diff --git a/lib/erd_handler/erd.rb b/lib/erd_handler/erd.rb index 142f575..bf74007 100644 --- a/lib/erd_handler/erd.rb +++ b/lib/erd_handler/erd.rb @@ -10,7 +10,7 @@ module ErdHandler doc = Document.new(source) raise InvalidErdFile unless doc.elements.to_a.length == 1 and doc.elements[1].name.downcase == 'drawing' self.mark = doc.elements['Drawing'].attributes["mark"].to_f - self.name = doc.elements['Drawing'].attributes["name"] + self.name = Label.new doc.elements['Drawing'].attributes["name"] doc.elements.each('Drawing/box') do |box_element| self << Box.new(box_element) end