Remove unused imports

snowflakedb · Nov 19, 2024 · b67548b · b67548b
1 parent 4b465c3
commit b67548b
Show file tree

Hide file tree

Showing 9 changed files with 52 additions and 85 deletions.
diff --git a/scripts/templates/template.py.jinja2 b/scripts/templates/template.py.jinja2
@@ -4,13 +4,12 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
 from snowflake.telemetry._internal.serialize import (
     Enum,
     MessageMarshaler,
     Varint,
 )
-from typing import List, Optional
+from typing import List
 
 {% for import in file_template.imports %}
 from {{ import }} import *
@@ -56,7 +55,7 @@ class {{ message.name }}(MessageMarshaler):
 {%- endfor %}
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
 {%- for field in message.fields %}
         {{ field.serialize_field_inline | indent(8) }}
 {%- endfor %}

diff --git a/...flake/telemetry/_internal/opentelemetry/proto/collector/logs/v1/logs_service_marshaler.py b/...flake/telemetry/_internal/opentelemetry/proto/collector/logs/v1/logs_service_marshaler.py
@@ -4,11 +4,7 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
-from typing import (
-    List,
-    Optional,
-)
+from typing import List
 
 from snowflake.telemetry._internal.opentelemetry.proto.logs.v1.logs_marshaler import *
 from snowflake.telemetry._internal.serialize import (
@@ -43,7 +39,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._resource_logs:
             for v in self._resource_logs:
                 out += b"\n"
@@ -75,7 +71,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._partial_success is not None:
             out += b"\n"
             Varint.write_varint_u32(out, self._partial_success._get_size())
@@ -104,7 +100,7 @@ def calculate_size(self) -> int:
             size += len(b"\x12") + Varint.size_varint_u32(len(v)) + len(v)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.rejected_log_records:
             out += b"\x08"
             Varint.write_varint_i64(out, self.rejected_log_records)

diff --git a/...telemetry/_internal/opentelemetry/proto/collector/metrics/v1/metrics_service_marshaler.py b/...telemetry/_internal/opentelemetry/proto/collector/metrics/v1/metrics_service_marshaler.py
@@ -4,11 +4,7 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
-from typing import (
-    List,
-    Optional,
-)
+from typing import List
 
 from snowflake.telemetry._internal.opentelemetry.proto.metrics.v1.metrics_marshaler import *
 from snowflake.telemetry._internal.serialize import (
@@ -43,7 +39,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._resource_metrics:
             for v in self._resource_metrics:
                 out += b"\n"
@@ -75,7 +71,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._partial_success is not None:
             out += b"\n"
             Varint.write_varint_u32(out, self._partial_success._get_size())
@@ -104,7 +100,7 @@ def calculate_size(self) -> int:
             size += len(b"\x12") + Varint.size_varint_u32(len(v)) + len(v)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.rejected_data_points:
             out += b"\x08"
             Varint.write_varint_i64(out, self.rejected_data_points)

diff --git a/...ake/telemetry/_internal/opentelemetry/proto/collector/trace/v1/trace_service_marshaler.py b/...ake/telemetry/_internal/opentelemetry/proto/collector/trace/v1/trace_service_marshaler.py
@@ -4,11 +4,7 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
-from typing import (
-    List,
-    Optional,
-)
+from typing import List
 
 from snowflake.telemetry._internal.opentelemetry.proto.trace.v1.trace_marshaler import *
 from snowflake.telemetry._internal.serialize import (
@@ -43,7 +39,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._resource_spans:
             for v in self._resource_spans:
                 out += b"\n"
@@ -75,7 +71,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._partial_success is not None:
             out += b"\n"
             Varint.write_varint_u32(out, self._partial_success._get_size())
@@ -104,7 +100,7 @@ def calculate_size(self) -> int:
             size += len(b"\x12") + Varint.size_varint_u32(len(v)) + len(v)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.rejected_spans:
             out += b"\x08"
             Varint.write_varint_i64(out, self.rejected_spans)

diff --git a/src/snowflake/telemetry/_internal/opentelemetry/proto/common/v1/common_marshaler.py b/src/snowflake/telemetry/_internal/opentelemetry/proto/common/v1/common_marshaler.py
@@ -4,11 +4,7 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
-from typing import (
-    List,
-    Optional,
-)
+from typing import List
 
 from snowflake.telemetry._internal.serialize import (
     Enum,
@@ -87,7 +83,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.string_value is not None:
             v = self.string_value.encode("utf-8")
             out += b"\n"
@@ -141,7 +137,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._values:
             for v in self._values:
                 out += b"\n"
@@ -174,7 +170,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._values:
             for v in self._values:
                 out += b"\n"
@@ -213,7 +209,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.key:
             v = self.key.encode("utf-8")
             out += b"\n"
@@ -269,7 +265,7 @@ def calculate_size(self) -> int:
             size += len(b" ") + Varint.size_varint_u32(self.dropped_attributes_count)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.name:
             v = self.name.encode("utf-8")
             out += b"\n"

diff --git a/src/snowflake/telemetry/_internal/opentelemetry/proto/logs/v1/logs_marshaler.py b/src/snowflake/telemetry/_internal/opentelemetry/proto/logs/v1/logs_marshaler.py
@@ -4,11 +4,7 @@
 from __future__ import annotations
 
 import struct
-from io import BytesIO
-from typing import (
-    List,
-    Optional,
-)
+from typing import List
 
 from snowflake.telemetry._internal.opentelemetry.proto.common.v1.common_marshaler import *
 from snowflake.telemetry._internal.opentelemetry.proto.resource.v1.resource_marshaler import *
@@ -77,7 +73,7 @@ def calculate_size(self) -> int:
             )
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._resource_logs:
             for v in self._resource_logs:
                 out += b"\n"
@@ -131,7 +127,7 @@ def calculate_size(self) -> int:
             size += len(b"\x1a") + Varint.size_varint_u32(len(v)) + len(v)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._resource is not None:
             out += b"\n"
             Varint.write_varint_u32(out, self._resource._get_size())
@@ -194,7 +190,7 @@ def calculate_size(self) -> int:
             size += len(b"\x1a") + Varint.size_varint_u32(len(v)) + len(v)
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self._scope is not None:
             out += b"\n"
             Varint.write_varint_u32(out, self._scope._get_size())
@@ -304,7 +300,7 @@ def calculate_size(self) -> int:
             size += len(b"Y") + 8
         return size
 
-    def write_to(self, out: BytesIO) -> None:
+    def write_to(self, out: bytearray) -> None:
         if self.time_unix_nano:
             out += b"\t"
             out += struct.pack("<Q", self.time_unix_nano)