Skip to content

Commit

Permalink
Optimize imports (#93)
Browse files Browse the repository at this point in the history
* CUT WatermarkAdd from pdfconduit.conduit.watermark modules __all__

* FIX pdfconduit.conduit module imports

* CUT convert function from pdfconduit.convert module

* REFACTOR pdfconduit.modify module imports

* CUT function from pdfconduit.transform modules

* ADD `PdfDriver` to pdfconduit.utils module

* CUT `WatermarkAdd` from pdfconduit modules

* Revert "REFACTOR pdfconduit.modify module imports"

This reverts commit 2773a21.

* FIX issue with type hinting

* OPTIMIZE imports

* REFORMAT code style (via black)

---------

Co-authored-by: sfneal <sfneal@users.noreply.github.com>
  • Loading branch information
sfneal and sfneal authored Jul 11, 2024
1 parent bce2de4 commit 01ce0fe
Show file tree
Hide file tree
Showing 11 changed files with 21 additions and 18 deletions.
3 changes: 1 addition & 2 deletions pdfconduit/__init__.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
from pdfconduit.conduit import Encrypt, Watermark, WatermarkAdd
from pdfconduit.conduit import Encrypt, Watermark
from pdfconduit.convert import IMG2PDF, PDF2IMG, Flatten
from pdfconduit.transform import Merge, Rotate, Upscale, slicer
from pdfconduit.utils import Info

__all__ = [
"Encrypt",
"Watermark",
"WatermarkAdd",
"IMG2PDF",
"PDF2IMG",
"Flatten",
Expand Down
4 changes: 2 additions & 2 deletions pdfconduit/conduit/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from pdfconduit.conduit.encrypt import Encrypt
from pdfconduit.conduit.watermark import WatermarkAdd, Watermark
from pdfconduit.conduit.watermark import Watermark

__all__ = ["Encrypt", "Watermark", "WatermarkAdd"]
__all__ = ["Encrypt", "Watermark"]
3 changes: 1 addition & 2 deletions pdfconduit/conduit/watermark/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
from pdfconduit.conduit.watermark.add import WatermarkAdd
from pdfconduit.conduit.watermark.watermark import Watermark

__all__ = ["Watermark", "WatermarkAdd"]
__all__ = ["Watermark"]
6 changes: 3 additions & 3 deletions pdfconduit/convert/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from pdfconduit.convert.flatten import Flatten
from pdfconduit.convert.img2pdf import img2pdf, IMG2PDF
from pdfconduit.convert.pdf2img import pdf2img, PDF2IMG
from pdfconduit.convert.img2pdf import IMG2PDF
from pdfconduit.convert.pdf2img import PDF2IMG

# TODO: Create conversion modules pdf2docx, docx2pdf, html2pdf, csv2pdf, excel2pdf
__all__ = ["img2pdf", "pdf2img", "IMG2PDF", "PDF2IMG", "Flatten"]
__all__ = ["IMG2PDF", "PDF2IMG", "Flatten"]
5 changes: 4 additions & 1 deletion pdfconduit/modify/draw/pdf.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,10 @@ def split_str(string: str) -> Tuple[str, str]:

class DrawPDF:
def __init__(
self, tempdir: Optional[str] = None, compress: int = 0, pagesize: float = LETTER
self,
tempdir: Optional[str] = None,
compress: int = 0,
pagesize: Tuple[float, float] = LETTER,
):
if tempdir:
self.dir = tempdir
Expand Down
6 changes: 3 additions & 3 deletions pdfconduit/transform/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from pdfconduit.transform.merge import Merge
from pdfconduit.transform.rotate import rotate, Rotate
from pdfconduit.transform.rotate import Rotate
from pdfconduit.transform.slice import slicer
from pdfconduit.transform.upscale import upscale, Upscale
from pdfconduit.transform.upscale import Upscale

__all__ = ["Merge", "rotate", "Rotate", "slicer", "upscale", "Upscale"]
__all__ = ["Merge", "Rotate", "slicer", "Upscale"]
2 changes: 2 additions & 0 deletions pdfconduit/utils/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
from pdfconduit.utils.driver import PdfDriver
from pdfconduit.utils.info import Info
from pdfconduit.utils.path import set_destination, add_suffix
from pdfconduit.utils.read import pypdf_reader
Expand All @@ -13,4 +14,5 @@
"Info",
"Receipt",
"pypdf_reader",
"PdfDriver",
]
2 changes: 1 addition & 1 deletion samples/samples.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
import os

from pdfconduit.conduit import Watermark
from pdfconduit.conduit.watermark.label import Label
from pdfconduit.conduit.lib import available_images
from pdfconduit.conduit.watermark.label import Label
from pdfconduit.convert import Flatten
from pdfconduit.transform import slicer, upscale, Merge
from pdfconduit.utils import Info
Expand Down
2 changes: 1 addition & 1 deletion tests/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import inspect
import os
import shutil
import inspect

test_data_dir = os.path.join(os.path.dirname(__file__), "data")
# pdf_name = 'plan_l.pdf'
Expand Down
2 changes: 1 addition & 1 deletion tests/test_utils_path.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

from looptools import Timer

from pdfconduit.utils import add_suffix, set_destination
from pdfconduit.utils import add_suffix
from tests import *


Expand Down
4 changes: 2 additions & 2 deletions tests/test_utils_read.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
import unittest

from looptools import Timer
from pypdf import PdfReader
from pypdf.errors import PdfReadError
from looptools import Timer
from pdfconduit.utils import pypdf_reader

from pdfconduit.utils import pypdf_reader
from tests import *


Expand Down

0 comments on commit 01ce0fe

Please sign in to comment.