diff --git a/script/sequel-model-graph.rb b/script/sequel-model-graph.rb index 1c971a6402..fa0052829a 100644 --- a/script/sequel-model-graph.rb +++ b/script/sequel-model-graph.rb @@ -13,6 +13,7 @@ require 'set' OUTPUT_FILE = '/tmp/sequel-graph.dot' +INCLUDE_ASSOCIATION_TYPE = false # Load all the model classes (autoloader might not have loaded all of # them yet) @@ -27,7 +28,8 @@ Sequel::Model.subclasses.each do |model| associations = model.association_reflections.values associations.each do |assoc| # Graphviz dot output format - graph << "#{model.to_s.demodulize} -> #{assoc.associated_class.to_s} [ label=\"#{assoc[:type].to_s}\" ];" + association_type = INCLUDE_ASSOCIATION_TYPE ? %Q([ label=\"#{assoc[:type].to_s}\" ]) : '' + graph << "#{model.to_s.demodulize} -> #{assoc.associated_class.to_s} #{association_type};" end graph << "#{model.to_s.demodulize} -> {}" if associations.empty? end