edpibu/pyedf
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Format with Ruff

Browse source
This commit is contained in:
Edgar P. Burkhart 2026-02-17 11:20:26 +00:00
parent 57550c3282
commit e50eba313f
4 changed files with 61 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

1
pyedf/__init__.py
View file

@ -3,6 +3,7 @@ from pprint import pprint as pp
from .edfpdf import PDFReader from .edfpdf import PDFReader
def main(): def main():
config = tomllib.load(open("config.toml", "rb")) config = tomllib.load(open("config.toml", "rb"))
pdf_reader = PDFReader(config["pdf_url"]) pdf_reader = PDFReader(config["pdf_url"])

26
pyedf/edfpdf.py
View file

@ -4,6 +4,8 @@ import re
import requests import requests
from pprint import pprint as pp from pprint import pprint as pp
class PDFReader: class PDFReader:
def __init__(self, url): def __init__(self, url):
self.url = url self.url = url
@ -20,10 +22,12 @@ class PDFReader:
if response.status_code == 200: if response.status_code == 200:
self.pdf_content = response.content self.pdf_content = response.content
else: 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): def read_pdf(self):
if hasattr(self, 'pdf_content'): if hasattr(self, "pdf_content"):
reader = pypdf.PdfReader(io.BytesIO(self.pdf_content)) reader = pypdf.PdfReader(io.BytesIO(self.pdf_content))
text = reader.pages[0].extract_text() text = reader.pages[0].extract_text()
self.pdf_text = text self.pdf_text = text
@ -31,11 +35,13 @@ class PDFReader:
raise Exception("PDF content not available. Please call get_pdf() first.") raise Exception("PDF content not available. Please call get_pdf() first.")
def extract_base(self): def extract_base(self):
if hasattr(self, 'pdf_text'): if hasattr(self, "pdf_text"):
tarifs = dict() tarifs = dict()
split_text = self.pdf_text.split("Option Base", 1)[1] split_text = self.pdf_text.split("Option Base", 1)[1]
for power in [3, 6, 9, 12, 15, 18, 24, 30, 36]: 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] = { tarifs[power] = {
"abonnement": extraction.group(1), "abonnement": extraction.group(1),
"kWh": extraction.group(2), "kWh": extraction.group(2),
@ -45,11 +51,13 @@ class PDFReader:
raise Exception("PDF text not available. Please call read_pdf() first.") raise Exception("PDF text not available. Please call read_pdf() first.")
def extract_hc(self): def extract_hc(self):
if hasattr(self, 'pdf_text'): if hasattr(self, "pdf_text"):
tarifs = dict() tarifs = dict()
split_text = self.pdf_text.split("Option Heures Creuses", 1)[1] split_text = self.pdf_text.split("Option Heures Creuses", 1)[1]
for power in [6, 9, 12, 15, 18, 24, 30, 36]: 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] = { tarifs[power] = {
"abonnement": extraction.group(1), "abonnement": extraction.group(1),
"kWh": { "kWh": {
@ -62,11 +70,13 @@ class PDFReader:
raise Exception("PDF text not available. Please call read_pdf() first.") raise Exception("PDF text not available. Please call read_pdf() first.")
def extract_tempo(self): def extract_tempo(self):
if hasattr(self, 'pdf_text'): if hasattr(self, "pdf_text"):
tarifs = dict() tarifs = dict()
split_text = self.pdf_text.split("Option Tempo", 1)[1] split_text = self.pdf_text.split("Option Tempo", 1)[1]
for power in [6, 9, 12, 15, 18, 30, 36]: 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: if extraction:
self.tarifs.setdefault("tempo", dict())[power] = { self.tarifs.setdefault("tempo", dict())[power] = {
"abonnement": extraction.group(1), "abonnement": extraction.group(1),

5
pyproject.toml
View file

@ -8,3 +8,8 @@ dependencies = [
"pypdf>=6.7.0", "pypdf>=6.7.0",
"requests>=2.32.5", "requests>=2.32.5",
] ]
[dependency-groups]
dev = [
"ruff>=0.15.1",
]

33
uv.lock generated
View file

@ -86,12 +86,20 @@ dependencies = [
{ name = "requests" }, { name = "requests" },
] ]
[package.dev-dependencies]
dev = [
{ name = "ruff" },
]
[package.metadata] [package.metadata]
requires-dist = [ requires-dist = [
{ name = "pypdf", specifier = ">=6.7.0" }, { name = "pypdf", specifier = ">=6.7.0" },
{ name = "requests", specifier = ">=2.32.5" }, { name = "requests", specifier = ">=2.32.5" },
] ]
[package.metadata.requires-dev]
dev = [{ name = "ruff", specifier = ">=0.15.1" }]
[[package]] [[package]]
name = "pypdf" name = "pypdf"
version = "6.7.0" 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" }, { 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]] [[package]]
name = "urllib3" name = "urllib3"
version = "2.6.3" version = "2.6.3"