diff --git a/pyedf/__init__.py b/pyedf/__init__.py index a0a9769..db7291a 100644 --- a/pyedf/__init__.py +++ b/pyedf/__init__.py @@ -3,8 +3,9 @@ from pprint import pprint as pp from .edfpdf import PDFReader + def main(): config = tomllib.load(open("config.toml", "rb")) pdf_reader = PDFReader(config["pdf_url"]) - pp(pdf_reader.tarifs) \ No newline at end of file + pp(pdf_reader.tarifs) diff --git a/pyedf/edfpdf.py b/pyedf/edfpdf.py index 7bb75d4..22b5ae8 100644 --- a/pyedf/edfpdf.py +++ b/pyedf/edfpdf.py @@ -4,6 +4,8 @@ import re import requests from pprint import pprint as pp + + class PDFReader: def __init__(self, url): self.url = url @@ -20,10 +22,12 @@ class PDFReader: if response.status_code == 200: self.pdf_content = response.content else: - raise Exception(f"Failed to retrieve PDF. Status code: {response.status_code}") - + raise Exception( + f"Failed to retrieve PDF. Status code: {response.status_code}" + ) + def read_pdf(self): - if hasattr(self, 'pdf_content'): + if hasattr(self, "pdf_content"): reader = pypdf.PdfReader(io.BytesIO(self.pdf_content)) text = reader.pages[0].extract_text() self.pdf_text = text @@ -31,11 +35,13 @@ class PDFReader: raise Exception("PDF content not available. Please call get_pdf() first.") def extract_base(self): - if hasattr(self, 'pdf_text'): + if hasattr(self, "pdf_text"): tarifs = dict() split_text = self.pdf_text.split("Option Base", 1)[1] for power in [3, 6, 9, 12, 15, 18, 24, 30, 36]: - extraction = re.search(rf"^{power}" + 2 * r"\s(\d+,\d+)", split_text, re.MULTILINE) + extraction = re.search( + rf"^{power}" + 2 * r"\s(\d+,\d+)", split_text, re.MULTILINE + ) tarifs[power] = { "abonnement": extraction.group(1), "kWh": extraction.group(2), @@ -43,13 +49,15 @@ class PDFReader: self.tarifs["base"] = tarifs else: raise Exception("PDF text not available. Please call read_pdf() first.") - + def extract_hc(self): - if hasattr(self, 'pdf_text'): + if hasattr(self, "pdf_text"): tarifs = dict() split_text = self.pdf_text.split("Option Heures Creuses", 1)[1] for power in [6, 9, 12, 15, 18, 24, 30, 36]: - extraction = re.search(rf"^{power}" + 3 * r"\s(\d+,\d+)", split_text, re.MULTILINE) + extraction = re.search( + rf"^{power}" + 3 * r"\s(\d+,\d+)", split_text, re.MULTILINE + ) tarifs[power] = { "abonnement": extraction.group(1), "kWh": { @@ -62,11 +70,13 @@ class PDFReader: raise Exception("PDF text not available. Please call read_pdf() first.") def extract_tempo(self): - if hasattr(self, 'pdf_text'): + if hasattr(self, "pdf_text"): tarifs = dict() split_text = self.pdf_text.split("Option Tempo", 1)[1] for power in [6, 9, 12, 15, 18, 30, 36]: - extraction = re.search(rf"^{power}" + 7 * r"\s(\d+,\d+)", split_text, re.MULTILINE) + extraction = re.search( + rf"^{power}" + 7 * r"\s(\d+,\d+)", split_text, re.MULTILINE + ) if extraction: self.tarifs.setdefault("tempo", dict())[power] = { "abonnement": extraction.group(1), @@ -87,4 +97,4 @@ class PDFReader: } self.tarifs["tempo"] = tarifs else: - raise Exception("PDF text not available. Please call read_pdf() first.") \ No newline at end of file + raise Exception("PDF text not available. Please call read_pdf() first.") diff --git a/pyproject.toml b/pyproject.toml index d5e55a6..18188f9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,3 +8,8 @@ dependencies = [ "pypdf>=6.7.0", "requests>=2.32.5", ] + +[dependency-groups] +dev = [ + "ruff>=0.15.1", +] diff --git a/uv.lock b/uv.lock index 256d36e..1ec7df8 100644 --- a/uv.lock +++ b/uv.lock @@ -86,12 +86,20 @@ dependencies = [ { name = "requests" }, ] +[package.dev-dependencies] +dev = [ + { name = "ruff" }, +] + [package.metadata] requires-dist = [ { name = "pypdf", specifier = ">=6.7.0" }, { name = "requests", specifier = ">=2.32.5" }, ] +[package.metadata.requires-dev] +dev = [{ name = "ruff", specifier = ">=0.15.1" }] + [[package]] name = "pypdf" version = "6.7.0" @@ -116,6 +124,31 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/1e/db/4254e3eabe8020b458f1a747140d32277ec7a271daf1d235b70dc0b4e6e3/requests-2.32.5-py3-none-any.whl", hash = "sha256:2462f94637a34fd532264295e186976db0f5d453d1cdd31473c85a6a161affb6", size = 64738, upload-time = "2025-08-18T20:46:00.542Z" }, ] +[[package]] +name = "ruff" +version = "0.15.1" +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/04/dc/4e6ac71b511b141cf626357a3946679abeba4cf67bc7cc5a17920f31e10d/ruff-0.15.1.tar.gz", hash = "sha256:c590fe13fb57c97141ae975c03a1aedb3d3156030cabd740d6ff0b0d601e203f", size = 4540855, upload-time = "2026-02-12T23:09:09.998Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/23/bf/e6e4324238c17f9d9120a9d60aa99a7daaa21204c07fcd84e2ef03bb5fd1/ruff-0.15.1-py3-none-linux_armv6l.whl", hash = "sha256:b101ed7cf4615bda6ffe65bdb59f964e9f4a0d3f85cbf0e54f0ab76d7b90228a", size = 10367819, upload-time = "2026-02-12T23:09:03.598Z" }, + { url = "https://files.pythonhosted.org/packages/b3/ea/c8f89d32e7912269d38c58f3649e453ac32c528f93bb7f4219258be2e7ed/ruff-0.15.1-py3-none-macosx_10_12_x86_64.whl", hash = "sha256:939c995e9277e63ea632cc8d3fae17aa758526f49a9a850d2e7e758bfef46602", size = 10798618, upload-time = "2026-02-12T23:09:22.928Z" }, + { url = "https://files.pythonhosted.org/packages/5e/0f/1d0d88bc862624247d82c20c10d4c0f6bb2f346559d8af281674cf327f15/ruff-0.15.1-py3-none-macosx_11_0_arm64.whl", hash = "sha256:1d83466455fdefe60b8d9c8df81d3c1bbb2115cede53549d3b522ce2bc703899", size = 10148518, upload-time = "2026-02-12T23:08:58.339Z" }, + { url = "https://files.pythonhosted.org/packages/f5/c8/291c49cefaa4a9248e986256df2ade7add79388fe179e0691be06fae6f37/ruff-0.15.1-py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a9457e3c3291024866222b96108ab2d8265b477e5b1534c7ddb1810904858d16", size = 10518811, upload-time = "2026-02-12T23:09:31.865Z" }, + { url = "https://files.pythonhosted.org/packages/c3/1a/f5707440e5ae43ffa5365cac8bbb91e9665f4a883f560893829cf16a606b/ruff-0.15.1-py3-none-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:92c92b003e9d4f7fbd33b1867bb15a1b785b1735069108dfc23821ba045b29bc", size = 10196169, upload-time = "2026-02-12T23:09:17.306Z" }, + { url = "https://files.pythonhosted.org/packages/2a/ff/26ddc8c4da04c8fd3ee65a89c9fb99eaa5c30394269d424461467be2271f/ruff-0.15.1-py3-none-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:1fe5c41ab43e3a06778844c586251eb5a510f67125427625f9eb2b9526535779", size = 10990491, upload-time = "2026-02-12T23:09:25.503Z" }, + { url = "https://files.pythonhosted.org/packages/fc/00/50920cb385b89413f7cdb4bb9bc8fc59c1b0f30028d8bccc294189a54955/ruff-0.15.1-py3-none-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:66a6dd6df4d80dc382c6484f8ce1bcceb55c32e9f27a8b94c32f6c7331bf14fb", size = 11843280, upload-time = "2026-02-12T23:09:19.88Z" }, + { url = "https://files.pythonhosted.org/packages/5d/6d/2f5cad8380caf5632a15460c323ae326f1e1a2b5b90a6ee7519017a017ca/ruff-0.15.1-py3-none-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:6a4a42cbb8af0bda9bcd7606b064d7c0bc311a88d141d02f78920be6acb5aa83", size = 11274336, upload-time = "2026-02-12T23:09:14.907Z" }, + { url = "https://files.pythonhosted.org/packages/a3/1d/5f56cae1d6c40b8a318513599b35ea4b075d7dc1cd1d04449578c29d1d75/ruff-0.15.1-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:4ab064052c31dddada35079901592dfba2e05f5b1e43af3954aafcbc1096a5b2", size = 11137288, upload-time = "2026-02-12T23:09:07.475Z" }, + { url = "https://files.pythonhosted.org/packages/cd/20/6f8d7d8f768c93b0382b33b9306b3b999918816da46537d5a61635514635/ruff-0.15.1-py3-none-manylinux_2_31_riscv64.whl", hash = "sha256:5631c940fe9fe91f817a4c2ea4e81f47bee3ca4aa646134a24374f3c19ad9454", size = 11070681, upload-time = "2026-02-12T23:08:55.43Z" }, + { url = "https://files.pythonhosted.org/packages/9a/67/d640ac76069f64cdea59dba02af2e00b1fa30e2103c7f8d049c0cff4cafd/ruff-0.15.1-py3-none-musllinux_1_2_aarch64.whl", hash = "sha256:68138a4ba184b4691ccdc39f7795c66b3c68160c586519e7e8444cf5a53e1b4c", size = 10486401, upload-time = "2026-02-12T23:09:27.927Z" }, + { url = "https://files.pythonhosted.org/packages/65/3d/e1429f64a3ff89297497916b88c32a5cc88eeca7e9c787072d0e7f1d3e1e/ruff-0.15.1-py3-none-musllinux_1_2_armv7l.whl", hash = "sha256:518f9af03bfc33c03bdb4cb63fabc935341bb7f54af500f92ac309ecfbba6330", size = 10197452, upload-time = "2026-02-12T23:09:12.147Z" }, + { url = "https://files.pythonhosted.org/packages/78/83/e2c3bade17dad63bf1e1c2ffaf11490603b760be149e1419b07049b36ef2/ruff-0.15.1-py3-none-musllinux_1_2_i686.whl", hash = "sha256:da79f4d6a826caaea95de0237a67e33b81e6ec2e25fc7e1993a4015dffca7c61", size = 10693900, upload-time = "2026-02-12T23:09:34.418Z" }, + { url = "https://files.pythonhosted.org/packages/a1/27/fdc0e11a813e6338e0706e8b39bb7a1d61ea5b36873b351acee7e524a72a/ruff-0.15.1-py3-none-musllinux_1_2_x86_64.whl", hash = "sha256:3dd86dccb83cd7d4dcfac303ffc277e6048600dfc22e38158afa208e8bf94a1f", size = 11227302, upload-time = "2026-02-12T23:09:36.536Z" }, + { url = "https://files.pythonhosted.org/packages/f6/58/ac864a75067dcbd3b95be5ab4eb2b601d7fbc3d3d736a27e391a4f92a5c1/ruff-0.15.1-py3-none-win32.whl", hash = "sha256:660975d9cb49b5d5278b12b03bb9951d554543a90b74ed5d366b20e2c57c2098", size = 10462555, upload-time = "2026-02-12T23:09:29.899Z" }, + { url = "https://files.pythonhosted.org/packages/e0/5e/d4ccc8a27ecdb78116feac4935dfc39d1304536f4296168f91ed3ec00cd2/ruff-0.15.1-py3-none-win_amd64.whl", hash = "sha256:c820fef9dd5d4172a6570e5721704a96c6679b80cf7be41659ed439653f62336", size = 11599956, upload-time = "2026-02-12T23:09:01.157Z" }, + { url = "https://files.pythonhosted.org/packages/2a/07/5bda6a85b220c64c65686bc85bd0bbb23b29c62b3a9f9433fa55f17cda93/ruff-0.15.1-py3-none-win_arm64.whl", hash = "sha256:5ff7d5f0f88567850f45081fac8f4ec212be8d0b963e385c3f7d0d2eb4899416", size = 10874604, upload-time = "2026-02-12T23:09:05.515Z" }, +] + [[package]] name = "urllib3" version = "2.6.3"