2025-06-04 13:16:17 +08:00
|
|
|
#
|
|
|
|
|
# Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
|
|
|
|
|
#
|
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
# you may not use this file except in compliance with the License.
|
|
|
|
|
# You may obtain a copy of the License at
|
|
|
|
|
#
|
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
#
|
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
# See the License for the specific language governing permissions and
|
|
|
|
|
# limitations under the License.
|
|
|
|
|
#
|
|
|
|
|
|
|
|
|
|
import json
|
|
|
|
|
|
|
|
|
|
from docx import Document # pip install python-docx
|
|
|
|
|
from openpyxl import Workbook # pip install openpyxl
|
|
|
|
|
from PIL import Image, ImageDraw # pip install Pillow
|
|
|
|
|
from pptx import Presentation # pip install python-pptx
|
|
|
|
|
from reportlab.pdfgen import canvas # pip install reportlab
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_docx_file(path):
|
|
|
|
|
doc = Document()
|
2025-06-13 17:30:29 +08:00
|
|
|
doc.add_paragraph("This is a test DOCX file.")
|
2025-06-04 13:16:17 +08:00
|
|
|
doc.save(path)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_excel_file(path):
|
|
|
|
|
wb = Workbook()
|
|
|
|
|
ws = wb.active
|
2025-06-13 17:30:29 +08:00
|
|
|
ws["A1"] = "Test Excel File"
|
2025-06-04 13:16:17 +08:00
|
|
|
wb.save(path)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_ppt_file(path):
|
|
|
|
|
prs = Presentation()
|
|
|
|
|
slide = prs.slides.add_slide(prs.slide_layouts[0])
|
2025-06-13 17:30:29 +08:00
|
|
|
slide.shapes.title.text = "Test PPT File"
|
2025-06-04 13:16:17 +08:00
|
|
|
prs.save(path)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_image_file(path):
|
|
|
|
|
img = Image.new("RGB", (100, 100), color="blue")
|
|
|
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
|
draw.text((10, 40), "Test", fill="white")
|
|
|
|
|
img.save(path)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_pdf_file(path):
|
|
|
|
|
if not isinstance(path, str):
|
|
|
|
|
path = str(path)
|
|
|
|
|
c = canvas.Canvas(path)
|
2025-06-13 17:30:29 +08:00
|
|
|
c.drawString(100, 750, "Test PDF File")
|
2025-06-04 13:16:17 +08:00
|
|
|
c.save()
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_txt_file(path):
|
|
|
|
|
with open(path, "w", encoding="utf-8") as f:
|
2025-06-13 17:30:29 +08:00
|
|
|
f.write("This is the content of a test TXT file.")
|
2025-06-04 13:16:17 +08:00
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_md_file(path):
|
2025-06-13 17:30:29 +08:00
|
|
|
md_content = "# Test MD File\n\nThis is a test Markdown file."
|
2025-06-04 13:16:17 +08:00
|
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
|
|
|
f.write(md_content)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_json_file(path):
|
2025-06-13 17:30:29 +08:00
|
|
|
data = {"message": "This is a test JSON file", "value": 123}
|
2025-06-04 13:16:17 +08:00
|
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
|
|
|
json.dump(data, f, indent=2)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_eml_file(path):
|
|
|
|
|
eml_content = (
|
|
|
|
|
"From: sender@example.com\n"
|
|
|
|
|
"To: receiver@example.com\n"
|
2025-06-13 17:30:29 +08:00
|
|
|
"Subject: Test EML File\n\n"
|
|
|
|
|
"This is a test email content.\n"
|
2025-06-04 13:16:17 +08:00
|
|
|
)
|
|
|
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
|
|
|
f.write(eml_content)
|
|
|
|
|
return path
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_html_file(path):
|
|
|
|
|
html_content = (
|
|
|
|
|
"<html>\n"
|
2025-06-13 17:30:29 +08:00
|
|
|
"<head><title>Test HTML File</title></head>\n"
|
|
|
|
|
"<body><h1>This is a test HTML file</h1></body>\n"
|
2025-06-04 13:16:17 +08:00
|
|
|
"</html>"
|
|
|
|
|
)
|
|
|
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
|
|
|
f.write(html_content)
|
|
|
|
|
return path
|
