From d64b99f05ec39926c0fe938fb8b90a1b6a2d7cd2 Mon Sep 17 00:00:00 2001 From: edpibu Date: Tue, 17 Feb 2026 11:34:03 +0000 Subject: [PATCH] Add json export --- config.toml | 3 ++- pyedf/__init__.py | 3 +-- pyedf/edfpdf.py | 7 +++++-- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/config.toml b/config.toml index 761f10d..32a60bd 100644 --- a/config.toml +++ b/config.toml @@ -1 +1,2 @@ -pdf_url = "https://particulier.edf.fr/content/dam/2-Actifs/Documents/Offres/Grille_prix_Tarif_Bleu.pdf" \ No newline at end of file +pdf_url = "https://particulier.edf.fr/content/dam/2-Actifs/Documents/Offres/Grille_prix_Tarif_Bleu.pdf" +json_location = "edf_bleu.json" \ No newline at end of file diff --git a/pyedf/__init__.py b/pyedf/__init__.py index db7291a..bc56170 100644 --- a/pyedf/__init__.py +++ b/pyedf/__init__.py @@ -7,5 +7,4 @@ from .edfpdf import PDFReader def main(): config = tomllib.load(open("config.toml", "rb")) pdf_reader = PDFReader(config["pdf_url"]) - - pp(pdf_reader.tarifs) + pdf_reader.gen_json(config["json_location"]) diff --git a/pyedf/edfpdf.py b/pyedf/edfpdf.py index 22b5ae8..01c8fe8 100644 --- a/pyedf/edfpdf.py +++ b/pyedf/edfpdf.py @@ -1,10 +1,9 @@ import io +import json import pypdf import re import requests -from pprint import pprint as pp - class PDFReader: def __init__(self, url): @@ -98,3 +97,7 @@ class PDFReader: self.tarifs["tempo"] = tarifs else: raise Exception("PDF text not available. Please call read_pdf() first.") + + def gen_json(self, location): + with open(location, "w") as f: + json.dump(self.tarifs, f) \ No newline at end of file