lint

Author: John Lyu

dataframe_image/__init__.py

@@ -1,3 +1,5 @@
 from ._convert import convert
 from ._pandas_accessor import export, export_async
 from ._version import __version__
+
+__all__ = ["export", "export_async", "convert", "__version__"]

dataframe_image/_browser_pdf.py

@@ -5,14 +5,12 @@
 import os
 import platform
 import socket
-import urllib.parse
 from pathlib import Path
 from subprocess import Popen
-from tempfile import TemporaryDirectory, mkstemp
+from tempfile import mkstemp
 
 import aiohttp
-from nbconvert import TemplateExporter
-from nbconvert.exporters import Exporter, HTMLExporter
+from nbconvert.exporters import HTMLExporter
 
 from .converter.browser.chrome_converter import get_chrome_path
 
@@ -141,7 +139,7 @@ def get_pdf_data(file_name):
 
 
 def get_pdf_data_chromecontroller(file_name):
-    import ChromeController
+    import ChromeController  # type: ignore
 
     additional_options = get_launch_args()
     # ChromeContext will shlex.split binary, so add quote to it
@@ -150,10 +148,10 @@ def get_pdf_data_chromecontroller(file_name):
     ) as cr:
         # Do a blocking navigate to a URL, and get the page content as served by the remote
         # server, with no modification by local javascript (if applicable)
-        raw_source = cr.blocking_navigate_and_get_source(file_name)
+        # raw_source = cr.blocking_navigate_and_get_source(file_name)
         # Since the page is now rendered by the blocking navigate, we can
         # get the page source after any javascript has modified it.
-        rendered_source = cr.get_rendered_page_source()
+        # rendered_source = cr.get_rendered_page_source()
         # # Or take a screenshot
         # # The screenshot is the size of the remote browser's configured viewport,
         # # which by default is set to 1024 * 1366. This size can be changed via the

dataframe_image/_bundler.py

@@ -1,8 +1,6 @@
 import base64
-import json
 from pathlib import Path
 
-from tornado import gen
 
 
 def _jupyter_bundlerextension_paths():

dataframe_image/_convert.py

@@ -1,11 +1,7 @@
-import base64
 import io
 import logging
 import os
-import re
 import shutil
-import tempfile
-import time
 import urllib.parse
 import warnings
 from pathlib import Path

dataframe_image/_latex_pdf.py

@@ -1,6 +1,6 @@
 import shutil
 
-from nbconvert.exporters import Exporter, HTMLExporter, PDFExporter
+from nbconvert.exporters import PDFExporter
 
 from ._preprocessors import (
     ChangeOutputTypePreprocessor,

dataframe_image/_preprocessors.py

@@ -1,5 +1,4 @@
 import base64
-import io
 import re
 from pathlib import Path
 

dataframe_image/converter/__init__.py

@@ -1 +1 @@
-from .browser import *
+from .browser import *  # noqa: F403

dataframe_image/converter/browser/__init__.py

@@ -2,3 +2,11 @@
 from .html2image_converter import Html2ImageConverter
 from .playwright_converter import AsyncPlayWrightConverter, PlayWrightConverter
 from .selenium_converter import SeleniumConverter
+
+__all__ = [
+    "ChromeConverter",
+    "Html2ImageConverter",
+    "PlayWrightConverter",
+    "AsyncPlayWrightConverter",
+    "SeleniumConverter",
+]

dataframe_image/converter/browser/base.py

@@ -1,10 +1,8 @@
 import base64
 import io
 import logging
-import subprocess
 from abc import ABC
 from pathlib import Path
-from tempfile import TemporaryDirectory
 
 import numpy as np
 from PIL import Image, ImageOps

dataframe_image/converter/browser/playwright_converter.py

@@ -1,7 +1,5 @@
 import math
 from io import BytesIO
-from pathlib import Path
-from tempfile import TemporaryDirectory
 
 from PIL import Image