put trace in a result file
This commit is contained in:
parent
618213572f
commit
e042263871
3 changed files with 47 additions and 43 deletions
59
main.py
59
main.py
|
@ -5,6 +5,8 @@ from pathlib import Path
|
|||
import requests
|
||||
from dotenv import load_dotenv
|
||||
from pdfminer.high_level import extract_text
|
||||
from pdfminer.pdfdocument import PDFDocument
|
||||
from pdfminer.pdfparser import PDFParser
|
||||
|
||||
from config import (
|
||||
MENU_CRECHE_PDF_URL,
|
||||
|
@ -16,51 +18,16 @@ from config import (
|
|||
from menus import Menus, MenuMessageFormatter
|
||||
from message_sender import SignalMessager
|
||||
from pdf_downloader import download
|
||||
from result_file import ResultFile
|
||||
|
||||
load_dotenv()
|
||||
|
||||
def send_signal_message(
|
||||
days: list[str], menu: dict[str, dict[str, str]], menu_pdf_file: Path
|
||||
) -> None:
|
||||
headers = {"Content-Type": "application/json"}
|
||||
message = f"""
|
||||
**Menu de la crèche** (message automatique)
|
||||
|
||||
Menu {MENU_TYPE}:
|
||||
|
||||
{days[0].upper()}:
|
||||
*Midi*: {menu[days[0]]["midi"]}
|
||||
*Goûter*: {menu[days[0]]["gouter"]}
|
||||
|
||||
{days[1].upper()}:
|
||||
*Midi*: {menu[days[1]]["midi"]}
|
||||
*Goûter*: {menu[days[1]]["gouter"]}
|
||||
|
||||
{days[2].upper()}:
|
||||
*Midi*: {menu[days[2]]["midi"]}
|
||||
*Goûter*: {menu[days[2]]["gouter"]}
|
||||
|
||||
{days[3].upper()}:
|
||||
*Midi*: {menu[days[3]]["midi"]}
|
||||
*Goûter*: {menu[days[3]]["gouter"]}
|
||||
|
||||
{days[4].upper()}:
|
||||
*Midi*: {menu[days[4]]["midi"]}
|
||||
*Goûter*: {menu[days[4]]["gouter"]}
|
||||
"""
|
||||
text_mode = "styled"
|
||||
attachment = base64.b64encode(menu_pdf_file.read_bytes())
|
||||
body = {
|
||||
"message": message,
|
||||
"text_mode": text_mode,
|
||||
"number": SIGNAL_SENDER,
|
||||
"recipients": SIGNAL_RECIPIENTS,
|
||||
"base64_attachments": [
|
||||
f"data:application/pdf;filename=menu.pdf;base64,{attachment.decode()}"
|
||||
],
|
||||
}
|
||||
|
||||
requests.post(SIGNAL_API_URL, headers=headers, json=body)
|
||||
def read_pdf_metadata(pdf_file: Path):
|
||||
fp = open(pdf_file, "rb")
|
||||
parser = PDFParser(fp)
|
||||
doc = PDFDocument(parser)
|
||||
return doc.info
|
||||
|
||||
|
||||
def main():
|
||||
|
@ -68,12 +35,22 @@ def main():
|
|||
|
||||
menus = Menus(menu_pdf_file)
|
||||
|
||||
pdf_metadata = read_pdf_metadata(menu_pdf_file)
|
||||
creation_date = pdf_metadata[0]["CreationDate"].decode("utf-8").removeprefix("D:")
|
||||
modification_date = pdf_metadata[0]["ModDate"].decode("utf-8").removeprefix("D:")
|
||||
|
||||
result_file = ResultFile(Path.home() / ".cache" / "menu_creche.txt")
|
||||
if result_file.was_already_sent(creation_date, modification_date, MENU_TYPE):
|
||||
return
|
||||
|
||||
message_formatter = MenuMessageFormatter()
|
||||
message = message_formatter.create_message(menus, MENU_TYPE)
|
||||
|
||||
signal_messager = SignalMessager(SIGNAL_API_URL, SIGNAL_SENDER)
|
||||
signal_messager.send_message(message, menu_pdf_file, SIGNAL_RECIPIENTS)
|
||||
|
||||
result_file.write(creation_date, modification_date, MENU_TYPE)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue