chore: update typing in textfsm parse helper func

carlmontanari · Oct 20, 2024 · 5829eff · 5829eff
1 parent 6dd94db
commit 5829eff
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/scrapli/helper.py b/scrapli/helper.py
@@ -4,7 +4,7 @@
 import importlib.resources
 import sys
 import urllib.request
-from io import BytesIO, TextIOWrapper
+from io import BufferedReader, BytesIO, TextIOWrapper
 from pathlib import Path
 from shutil import get_terminal_size
 from typing import Any, Dict, List, Optional, TextIO, Tuple, Union
@@ -119,6 +119,8 @@ def textfsm_parse(
     """
     import textfsm  # pylint: disable=C0415
 
+    template_file: Union[BufferedReader, TextIOWrapper]
+
     if not isinstance(template, TextIOWrapper):
         if template.startswith("http://") or template.startswith("https://"):
             with urllib.request.urlopen(template) as response:
@@ -127,7 +129,7 @@ def textfsm_parse(
                     encoding=response.headers.get_content_charset(),
                 )
         else:
-            template_file = TextIOWrapper(open(template, mode="rb"))  # pylint: disable=R1732
+            template_file = open(template, mode="rb")  # pylint: disable=R1732
     else:
         template_file = template
     re_table = textfsm.TextFSM(template_file)

diff --git a/tests/unit/driver/generic/test_generic_async_driver.py b/tests/unit/driver/generic/test_generic_async_driver.py
@@ -1,4 +1,5 @@
 import logging
+
 import pytest
 
 from scrapli.driver.generic.base_driver import ReadCallback

diff --git a/tests/unit/driver/generic/test_generic_sync_driver.py b/tests/unit/driver/generic/test_generic_sync_driver.py
@@ -1,4 +1,5 @@
 import logging
+
 import pytest
 
 from scrapli.driver.generic.base_driver import ReadCallback