draw方法更新

Author: brucehe3

lib/generator.py

@@ -158,6 +158,20 @@ def _table_end(self):
         </TABLE>
         """
 
+    def relation(self, table_a, table_b, label):
+        """
+        样式暂时不处理
+        :param table_a:
+        :param table_b:
+        :param label:
+        :return:
+        """
+
+        return """
+        %s -> %s
+        [label="%s"] [arrowhead=none, arrowtail=dot, dir=both];
+        """ % (table_a, table_b, label)
+
     def label(self, *args, **kwargs):
         """
         :return:
@@ -182,7 +196,10 @@ def table(self, *args, **kwargs):
         if not data or not table_name:
             raise AttributeError('param is empty')
 
-        title = "%s(%s)" % (table_name, table_comment)
+        title = table_name
+        if table_comment:
+            title = "%s(%s)" % (table_name, table_comment)
+
         # 字段长度要保持一致
         output = [self._table_start(), self.title(title)]
 

pydbdoc.py

@@ -181,7 +181,7 @@ def output_graph(self, dest, override):
                 'table_name': table_name,
                 'table_comment': table_comment,
                 # 'columns': ['字段名', '类型', '是否可NULL', '描述'],
-                'data': [[column.name, column.field_type, column.if_null, column.comment.replace(os.linesep, '')] for
+                'data': [[column.name, column.field_type, column.if_null, column.comment] for
                          column in columns],
             }
             _output.append(generator.label(**_table_data))