table init changed

julien-duponchelle · Sep 16, 2023 · 6aa3872 · 6aa3872
1 parent aa1ecdf
commit 6aa3872
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pymysqlreplication/table.py b/pymysqlreplication/table.py
@@ -2,8 +2,9 @@
 
 
 class Table(object):
-    def __init__(self, table_id, schema, table, columns, primary_key=None):
-        self.column_name_flag = False
+    def __init__(
+        self, table_id, schema, table, columns, primary_key=None, column_name_flag=False
+    ):
         if primary_key is None:
             primary_key = [c.data["name"] for c in columns if c.data["is_primary"]]
             if len(primary_key) == 0:
@@ -20,6 +21,7 @@ def __init__(self, table_id, schema, table, columns, primary_key=None):
                 "table": table,
                 "columns": columns,
                 "primary_key": primary_key,
+                "column_name_flag": column_name_flag,
             }
         )