Derive all api classes from Request / Response base classesrequest_response_classes

1 files changed, 9 insertions, 9 deletions

diff --git a/kafka/protocol/fetch.py b/kafka/protocol/fetch.py
index 6a9ad5b..b441e63 100644
--- a/kafka/protocol/fetch.py
+++ b/kafka/protocol/fetch.py
@@ -1,11 +1,11 @@
 from __future__ import absolute_import
 
+from .api import Request, Response
 from .message import MessageSet
-from .struct import Struct
 from .types import Array, Int16, Int32, Int64, Schema, String
 
 
-class FetchResponse_v0(Struct):
+class FetchResponse_v0(Response):
     API_KEY = 1
     API_VERSION = 0
     SCHEMA = Schema(
@@ -19,7 +19,7 @@ class FetchResponse_v0(Struct):
     )
 
 
-class FetchResponse_v1(Struct):
+class FetchResponse_v1(Response):
     API_KEY = 1
     API_VERSION = 1
     SCHEMA = Schema(
@@ -34,19 +34,19 @@ class FetchResponse_v1(Struct):
     )
 
 
-class FetchResponse_v2(Struct):
+class FetchResponse_v2(Response):
     API_KEY = 1
     API_VERSION = 2
     SCHEMA = FetchResponse_v1.SCHEMA  # message format changed internally
 
 
-class FetchResponse_v3(Struct):
+class FetchResponse_v3(Response):
     API_KEY = 1
     API_VERSION = 3
     SCHEMA = FetchResponse_v2.SCHEMA
 
 
-class FetchRequest_v0(Struct):
+class FetchRequest_v0(Request):
     API_KEY = 1
     API_VERSION = 0
     RESPONSE_TYPE = FetchResponse_v0
@@ -63,21 +63,21 @@ class FetchRequest_v0(Struct):
     )
 
 
-class FetchRequest_v1(Struct):
+class FetchRequest_v1(Request):
     API_KEY = 1
     API_VERSION = 1
     RESPONSE_TYPE = FetchResponse_v1
     SCHEMA = FetchRequest_v0.SCHEMA
 
 
-class FetchRequest_v2(Struct):
+class FetchRequest_v2(Request):
     API_KEY = 1
     API_VERSION = 2
     RESPONSE_TYPE = FetchResponse_v2
     SCHEMA = FetchRequest_v1.SCHEMA
 
 
-class FetchRequest_v3(Struct):
+class FetchRequest_v3(Request):
     API_KEY = 1
     API_VERSION = 3
     RESPONSE_TYPE = FetchResponse_v3