Implement protocol

esque/cli/output.py

@@ -105,21 +105,23 @@ def pretty_bytes(value: bytes) -> str:
     return value.decode("UTF-8")
 
 
-def pretty_duration(value: Any, *, multiplier: int = 1) -> str:
-    if not value:
+def pretty_duration(orig_value: Any, *, multiplier: int = 1) -> str:
+    if not orig_value:
         return ""
 
-    if type(value) != int:
-        value = int(value)
+    if type(orig_value) != int:
+        value = int(orig_value)
+    else:
+        value = orig_value
 
     value *= multiplier
 
     # Fix for conversion errors of ms > C_MAX_INT in some internal lib
     if value > MILLISECONDS_PER_YEAR:
         value = int(value / MILLISECONDS_PER_YEAR)
-        return pendulum.duration(years=value).in_words()
+        return f"{orig_value} ({pendulum.duration(years=value).in_words()})"
 
-    return pendulum.duration(milliseconds=value).in_words()
+    return f"{orig_value} ({pendulum.duration(milliseconds=value).in_words()})"
 
 
 def pretty_topic_diffs(topics_config_diff: Dict[str, TopicDiff]) -> str:
@@ -173,7 +175,7 @@ def pretty_size(value: Any) -> str:
     ]
     for sign, size in units:
         if value >= size:
-            return f"{pretty_float(value / size)} {sign}"
+            return f"{value} ({pretty_float(value / size)} {sign})"
 
 
 def bold(s: str) -> str:

esque/protocol/__init__.py

@@ -0,0 +1,100 @@
+from .connection import BrokerConnection
+
+from .api import (
+    ApiKey,
+    ApiVersions,
+    Request,
+    RequestData,
+    ResponseData,
+    SUPPORTED_API_VERSIONS,
+    ProduceRequestData,
+    ProduceResponseData,
+    FetchRequestData,
+    FetchResponseData,
+    ListOffsetsRequestData,
+    ListOffsetsResponseData,
+    MetadataRequestData,
+    MetadataResponseData,
+    LeaderAndIsrRequestData,
+    LeaderAndIsrResponseData,
+    StopReplicaRequestData,
+    StopReplicaResponseData,
+    UpdateMetadataRequestData,
+    UpdateMetadataResponseData,
+    ControlledShutdownRequestData,
+    ControlledShutdownResponseData,
+    OffsetCommitRequestData,
+    OffsetCommitResponseData,
+    OffsetFetchRequestData,
+    OffsetFetchResponseData,
+    FindCoordinatorRequestData,
+    FindCoordinatorResponseData,
+    JoinGroupRequestData,
+    JoinGroupResponseData,
+    HeartbeatRequestData,
+    HeartbeatResponseData,
+    LeaveGroupRequestData,
+    LeaveGroupResponseData,
+    SyncGroupRequestData,
+    SyncGroupResponseData,
+    DescribeGroupsRequestData,
+    DescribeGroupsResponseData,
+    ListGroupsRequestData,
+    ListGroupsResponseData,
+    SaslHandshakeRequestData,
+    SaslHandshakeResponseData,
+    ApiVersionsRequestData,
+    ApiVersionsResponseData,
+    CreateTopicsRequestData,
+    CreateTopicsResponseData,
+    DeleteTopicsRequestData,
+    DeleteTopicsResponseData,
+    DeleteRecordsRequestData,
+    DeleteRecordsResponseData,
+    InitProducerIdRequestData,
+    InitProducerIdResponseData,
+    OffsetForLeaderEpochRequestData,
+    OffsetForLeaderEpochResponseData,
+    AddPartitionsToTxnRequestData,
+    AddPartitionsToTxnResponseData,
+    AddOffsetsToTxnRequestData,
+    AddOffsetsToTxnResponseData,
+    EndTxnRequestData,
+    EndTxnResponseData,
+    WriteTxnMarkersRequestData,
+    WriteTxnMarkersResponseData,
+    TxnOffsetCommitRequestData,
+    TxnOffsetCommitResponseData,
+    DescribeAclsRequestData,
+    DescribeAclsResponseData,
+    CreateAclsRequestData,
+    CreateAclsResponseData,
+    DeleteAclsRequestData,
+    DeleteAclsResponseData,
+    DescribeConfigsRequestData,
+    DescribeConfigsResponseData,
+    AlterConfigsRequestData,
+    AlterConfigsResponseData,
+    AlterReplicaLogDirsRequestData,
+    AlterReplicaLogDirsResponseData,
+    DescribeLogDirsRequestData,
+    DescribeLogDirsResponseData,
+    SaslAuthenticateRequestData,
+    SaslAuthenticateResponseData,
+    CreatePartitionsRequestData,
+    CreatePartitionsResponseData,
+    CreateDelegationTokenRequestData,
+    CreateDelegationTokenResponseData,
+    RenewDelegationTokenRequestData,
+    RenewDelegationTokenResponseData,
+    ExpireDelegationTokenRequestData,
+    ExpireDelegationTokenResponseData,
+    DescribeDelegationTokenRequestData,
+    DescribeDelegationTokenResponseData,
+    DeleteGroupsRequestData,
+    DeleteGroupsResponseData,
+    ElectPreferredLeadersRequestData,
+    ElectPreferredLeadersResponseData,
+    IncrementalAlterConfigsRequestData,
+    IncrementalAlterConfigsResponseData,
+)