update
This commit is contained in:
156
Scripts/python/import_pdf_from_o365.py
Normal file
156
Scripts/python/import_pdf_from_o365.py
Normal file
@@ -0,0 +1,156 @@
|
||||
import os
|
||||
import requests
|
||||
import msal
|
||||
import base64
|
||||
|
||||
# ==============================================================================
|
||||
# KONFIGURÁCIÓ
|
||||
# ==============================================================================
|
||||
# Az Azure App Registration-ból kapott adatok
|
||||
TENANT_ID = "TENANT_ID_MASOLD_BE_IDE"
|
||||
CLIENT_ID = "CLIENT_ID_MASOLD_BE_IDE"
|
||||
CLIENT_SECRET = "CLIENT_SECRET_MASOLD_BE_IDE" # A "Value", nem a "Secret ID"
|
||||
|
||||
# A postafiók, amit figyelni kell
|
||||
USER_EMAIL = "Bestellung-Fax-Eingang@aps-hh.de"
|
||||
|
||||
# Hova mentsük a letöltött PDF-eket
|
||||
DOWNLOAD_DIR = r"\\aps-nb090\test"
|
||||
|
||||
# A mappa neve, ahova a feldolgozott leveleket helyezzük
|
||||
PROCESSED_FOLDER_NAME = "erledigt"
|
||||
# ==============================================================================
|
||||
|
||||
# Microsoft Graph API végpontok
|
||||
GRAPH_API_ENDPOINT = "https://graph.microsoft.com/v1.0"
|
||||
AUTHORITY_URL = f"https://login.microsoftonline.com/{TENANT_ID}"
|
||||
SCOPES = ["https://graph.microsoft.com/.default"]
|
||||
|
||||
|
||||
def get_graph_api_token():
|
||||
"""Beszerzi a hozzáférési tokent a Microsoft Graph API-hoz."""
|
||||
app = msal.ConfidentialClientApplication(
|
||||
client_id=CLIENT_ID,
|
||||
authority=AUTHORITY_URL,
|
||||
client_credential=CLIENT_SECRET
|
||||
)
|
||||
result = app.acquire_token_silent(scopes=SCOPES, account=None)
|
||||
if not result:
|
||||
result = app.acquire_token_for_client(scopes=SCOPES)
|
||||
|
||||
if "access_token" in result:
|
||||
return result["access_token"]
|
||||
else:
|
||||
print("Hiba a token beszerzése során!")
|
||||
print(result.get("error"))
|
||||
print(result.get("error_description"))
|
||||
return None
|
||||
|
||||
def get_folder_id(access_token, folder_name):
|
||||
"""Megkeresi egy mappa ID-ját a neve alapján."""
|
||||
headers = {"Authorization": f"Bearer {access_token}"}
|
||||
url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/mailFolders"
|
||||
response = requests.get(url, headers=headers)
|
||||
response.raise_for_status()
|
||||
folders = response.json().get("value", [])
|
||||
|
||||
for folder in folders:
|
||||
if folder["displayName"].lower() == folder_name.lower():
|
||||
return folder["id"]
|
||||
|
||||
# Kezelhetnénk azt az esetet is, ha a mappa nem létezik, és létrehoznánk.
|
||||
# Most egyszerűen hibát dobunk.
|
||||
raise ValueError(f"A '{folder_name}' mappa nem található.")
|
||||
|
||||
|
||||
def main():
|
||||
"""Fő feldolgozó funkció."""
|
||||
print("Graph API token beszerzése...")
|
||||
access_token = get_graph_api_token()
|
||||
if not access_token:
|
||||
return
|
||||
|
||||
headers = {"Authorization": f"Bearer {access_token}"}
|
||||
|
||||
try:
|
||||
print(f"'{PROCESSED_FOLDER_NAME}' mappa ID-jának keresése...")
|
||||
processed_folder_id = get_folder_id(access_token, PROCESSED_FOLDER_NAME)
|
||||
print("Mappa ID sikeresen lekérve.")
|
||||
except (requests.HTTPError, ValueError) as e:
|
||||
print(f"Hiba a mappa ID lekérése közben: {e}")
|
||||
return
|
||||
|
||||
# Csak az olvasatlan, csatolmánnyal rendelkező levelek lekérdezése
|
||||
# $select=id,subject -> Csak a szükséges mezőket kérjük le a hatékonyságért
|
||||
messages_url = (
|
||||
f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/mailFolders/inbox/messages?"
|
||||
"$filter=isRead eq false and hasAttachments eq true&"
|
||||
"$select=id,subject"
|
||||
)
|
||||
|
||||
response = requests.get(messages_url, headers=headers)
|
||||
response.raise_for_status()
|
||||
messages = response.json().get("value", [])
|
||||
|
||||
if not messages:
|
||||
print("Nincsenek új, feldolgozandó e-mailek.")
|
||||
return
|
||||
|
||||
print(f"{len(messages)} új e-mail található csatolmánnyal.")
|
||||
|
||||
for message in messages:
|
||||
msg_id = message["id"]
|
||||
subject = message.get("subject", "N/A")
|
||||
print(f"\n--- Feldolgozás alatt: '{subject}' (ID: {msg_id}) ---")
|
||||
|
||||
attachments_url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/messages/{msg_id}/attachments"
|
||||
response = requests.get(attachments_url, headers=headers)
|
||||
if response.status_code != 200:
|
||||
print(f" Hiba a csatolmányok lekérésekor: {response.json()}")
|
||||
continue
|
||||
|
||||
attachments = response.json().get("value", [])
|
||||
pdf_found = False
|
||||
|
||||
for att in attachments:
|
||||
filename = att.get("name", "unknown")
|
||||
content_type = att.get("contentType", "")
|
||||
|
||||
if filename.lower().endswith(".pdf") or content_type == "application/pdf":
|
||||
print(f" PDF csatolmány található: {filename}")
|
||||
|
||||
# A csatolmány tartalma a 'contentBytes' mezőben van, Base64 kódolással
|
||||
file_content = base64.b64decode(att["contentBytes"])
|
||||
|
||||
# Fájlnév "tisztítása"
|
||||
safe_filename = "".join(c if c.isalnum() or c in (" ", ".", "_", "-") else "_" for c in filename)
|
||||
filepath = os.path.join(DOWNLOAD_DIR, safe_filename)
|
||||
|
||||
try:
|
||||
with open(filepath, "wb") as f:
|
||||
f.write(file_content)
|
||||
print(f" PDF sikeresen mentve: {filepath}")
|
||||
pdf_found = True
|
||||
except Exception as e:
|
||||
print(f" Hiba a fájl mentése közben: {e}")
|
||||
|
||||
if pdf_found:
|
||||
print(f" E-mail áthelyezése a '{PROCESSED_FOLDER_NAME}' mappába...")
|
||||
move_url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/messages/{msg_id}/move"
|
||||
move_payload = {"destinationId": processed_folder_id}
|
||||
|
||||
response = requests.post(move_url, headers=headers, json=move_payload)
|
||||
if response.status_code == 201:
|
||||
print(" E-mail sikeresen áthelyezve.")
|
||||
else:
|
||||
# Az áthelyezéshez Mail.ReadWrite jogosultság kell!
|
||||
print(f" Hiba az e-mail áthelyezésekor: {response.status_code} - {response.text}")
|
||||
|
||||
print("\nFeldolgozás befejezve.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
if not os.path.exists(DOWNLOAD_DIR):
|
||||
print(f"Hiba: A letöltési mappa nem létezik: {DOWNLOAD_DIR}")
|
||||
else:
|
||||
main()
|
||||
Reference in New Issue
Block a user