Fix crash related to non existent primary key

Author: julien-duponchelle

pymysqlreplication/column.py

@@ -24,7 +24,10 @@ def __parse_column_definition(self, column_type, column_schema, packet):
         self.data["comment"] = column_schema["COLUMN_COMMENT"]
         self.data["unsigned"] = False
         self.data["type_is_bool"] = False
-        self.data["is_primary"] = column_schema["COLUMN_KEY"] == "PRI"
+        if "COLUMN_KEY" in column_schema and column_schema["COLUMN_KEY"] == "PRI":
+            self.data["is_primary"] = True
+        else:
+            self.data["is_primary"] = False
 
         if column_schema["COLUMN_TYPE"].find("unsigned") != -1:
             self.data["unsigned"] = True

pymysqlreplication/table.py

@@ -2,14 +2,15 @@
 
 
 class Table(object):
-    def __init__(self, column_schemas, table_id, schema, table, columns):
-        primary_key = [c.data["name"] for c in columns if c.data["is_primary"]]
-        if len(primary_key) == 0:
-            primary_key = ''
-        elif len(primary_key) == 1:
-            primary_key, = primary_key
-        else:
-            primary_key = tuple(primary_key)
+    def __init__(self, column_schemas, table_id, schema, table, columns, primary_key = None):
+        if primary_key is None:
+            primary_key = [c.data["name"] for c in columns if c.data["is_primary"]]
+            if len(primary_key) == 0:
+                primary_key = ''
+            elif len(primary_key) == 1:
+                primary_key, = primary_key
+            else:
+                primary_key = tuple(primary_key)
 
         self.data = {
             "column_schemas": column_schemas,