o hP @s|ddlZddlmZmZmZmZddlmZddlmZddl m Z e e Z ee Zdedefdd Zejd d gd d efddZejdd gd ddZejdd gd ddZedddZedddZe dkrzeWneyZze eeedWYdZ[ndZ[wwe de de d e d!e d"eZej d#iedSdS)$N)Flaskjsonifyrequest make_response)config)export_notion_page_to_markdown)convert_markdown_to_pdftitlereturncCs~|}tdd|}tdd|}tdd|}tdd|}|d}t|dkr/|dd}|r;|dks;|dds=d }|S) a  Sanitize a string to make it safe for use as a filename. Only allows ASCII characters to avoid HTTP header encoding issues. Args: title: The original title string Returns: str: A sanitized filename-safe string with only ASCII characters z [<>:"/\\|?*]_z[^a-zA-Z0-9\s\-_.]z\s+z_{2,}dNUntitled)resubstriplenreplace)r sanitizedr /var/www/html/notion2html/app.pysanitize_filename s    rz/page/GET)methodspage_idc Cslztd|tjdddk}|s$d|vs"d|vr$d}td||d d}d |vr=|d d }d |vrH|d d }td ||rXt|dkrat ddidfWSt |\}}td|d|d|rz%t ||\}}td|t |}d|j d<d|d|j d<|WWSty}ztdt|t ddt|idfWYd}~WSd}~wwt|} | d} td | d!d| dd"} |d#| fWSty } ztd$t| t dd%idfWYd} ~ Sd} ~ wty5} z td&|d't| t dd(t| idfWYd} ~ Sd} ~ ww))aY Export a Notion page to Markdown format or PDF format. Args: page_id: The ID of the Notion page to export Query Parameters: pdf: If set to 'true', returns PDF instead of Markdown Returns: The page content in Markdown or PDF format as a file download with page name as filename z!Received request to export page: pdfr truez &pdf=truez ?pdf=trueTzPDF output requested: -&r?zCleaned page_id: errorz>Invalid page ID format. Page ID should be a 32-character UUID.izSuccessfully exported page 'z' (z ) to MarkdownzSuccessfully converted to PDF: zapplication/pdf Content-Typezattachment; filename=""Content-DispositionzPDF conversion failed: zFailed to convert to PDF: Nz.mdzSuggesting filename: ztext/markdown; charset=utf-8)r"r$zConfiguration error: zBServer configuration error. Please check that NOTION_TOKEN is set.zError exporting page z: zFailed to export page: )loggerinforargsgetlowerrsplitrrrrrheaders Exceptionr!strr ValueError) r pdf_requested clean_page_id page_titlemarkdown_content pdf_content pdf_filenameresponse pdf_error safe_filenamefilenamer-errr export_page2sz      r</healthcCtddddfS)zHealth check endpoint.healthyNotion to Markdown Exporter)statusservicer&rrrrr health_check rD/cCstddddddiddd fS) z&Home endpoint with usage instructions.r@zNSend GET request to /page/ to export a Notion page to Markdown or PDFz&/page/897e5a76ae524b489fdfe71f5945d1afz//page/897e5a76ae524b489fdfe71f5945d1af?pdf=truerz=Optional. Set to "true" to get PDF output instead of Markdownr=)rBusageexample pdf_example parametersrDr&rCrrrrhomes rKcCr>)zHandle 404 errors.zEndpoint not foundzGSend GET request to /page/ to export a Notion page to Markdown)r!rGrLrCr!rrr not_foundrErNr%cCs$tdt|tddidfS)zHandle 500 errors.zInternal server error: r!zInternal server errorr%)r'r!r/rrMrrrinternal_errors rO__main__z.Starting Notion to Markdown Exporter server...zAvailable endpoints:z6 GET /page/ - Export Notion page to Markdownz GET /health - Health checkz GET / - Service informationr)!rflaskrrrrr notion_to_mdr md_to_pdfr get_logger__name__r'appr/rrouter<rDrK errorhandlerrNrOvalidate_required_configr0r;r!exitr(get_flask_config flask_configrunrrrrsD    %[