| |
| |
| import argparse |
| import os |
| from pathlib import Path |
| import shutil |
| import tempfile |
| import uuid |
|
|
| from project_settings import project_path |
| from toolbox.to_markdown.base_to_markdown import BaseToMarkdown |
| from toolbox.os.command import Command |
|
|
|
|
| def get_args(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument( |
| "--filename", |
| default=(project_path / "data/files/ppt/AI能力分享.pptx").as_posix(), |
| |
| type=str |
| ) |
| args = parser.parse_args() |
| return args |
|
|
|
|
| @BaseToMarkdown.register("pptx2md") |
| class Pptx2md(BaseToMarkdown): |
| """ |
| https://github.com/ptsefton/pptx_to_md |
| """ |
| def __init__(self, filename: str): |
| super().__init__(filename) |
|
|
| def command(self, filename: str, output_file: str, image_dir: str = "media"): |
| cmd = f'python -m pptx2md -o "{output_file}" -i "{image_dir}" "{filename}"' |
| Command.popen(cmd) |
| return cmd |
|
|
| def save_to_zip(self, output_dir: str): |
| basename = str(uuid.uuid4()) |
|
|
| temp_dir = Path(tempfile.gettempdir()) / basename |
| temp_dir.mkdir(parents=True, exist_ok=False) |
|
|
| md_file = temp_dir / f"{basename}.md" |
|
|
| self.command( |
| self.filename, |
| output_file=md_file.as_posix(), |
| image_dir=(temp_dir / "media").as_posix() |
| ) |
|
|
| |
| output_zip_file = os.path.join(output_dir, f"{basename}.zip") |
| self.zip_directory(temp_dir, output_zip_file) |
| shutil.rmtree(temp_dir) |
| return output_zip_file |
|
|
|
|
| def main(): |
| args = get_args() |
|
|
| p2m = Pptx2md(args.filename) |
|
|
| output_zip_file = p2m.save_to_zip(output_dir=".") |
| print(output_zip_file) |
| return |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
