如何为Python L-system Turtle绘图项目添加--export导出eps功能

你可以通过以下步骤修改现有代码，实现命令行指定--export参数导出EPS文件的需求：

步骤1：新增命令行参数解析依赖 #

在代码头部的导入区域新增argparse模块引入：

import json
import turtle as tur
from datetime import datetime
import argparse # 新增这一行

步骤2：修改main函数逻辑，加入参数解析和导出功能 #

将原有main函数替换为如下代码：

def main():
    # 解析命令行参数
    parser = argparse.ArgumentParser()
    parser.add_argument('--export', type=str, help='导出EPS文件的保存路径')
    args = parser.parse_args()

    file = getFile()

    # Get input on how many iterations the user wants to preform
    iter = int(input("How many iterations of the given l-system do you want to preform?: "))

    # Used_functions is a list of the possible translations that refer to a function
    # When you add more possible functions, add the translations of the function in the used_functions below
    used_functions = ("draw", "angle", "forward", "nop", "push", "pop", "color")

    data = getData(file)
    if checkFile(data) == False:
        print("The given input file is not contain the needed variabes.")
        return

    axiom, rules, variables, constants, trans = getVariables(data)
    alph = makeAlph(variables, constants)

    if checkData(axiom, rules, alph, trans, used_functions) == False:
        print("The given variables in the input file are not correct.")
        return

    lstring = lSystem(axiom, rules, iter)
    addHistory(axiom, rules, alph, trans, iter, lstring, variables, constants)
    print(lstring)
    draw(lstring, trans)
    
    # 新增导出逻辑
    if args.export:
        screen = tur.getscreen()
        # 导出画布内容为EPS文件
        screen.getcanvas().postscript(file=args.export)
        # 导出完成直接退出，不需要等待点击
        tur.bye()
    else:
        # 原有交互逻辑：点击窗口才退出
        tur.Screen().exitonclick()

步骤3：运行测试 #

执行你要求的命令即可完成导出：

python3 main.py --export my_image.eps

按照提示输入JSON配置文件路径和迭代次数，运行结束后当前目录就会生成目标EPS文件。

无图形界面Linux环境适配 #

如果你使用的是没有桌面环境的Linux服务器，需要安装虚拟显示组件运行：

1. 安装依赖：sudo apt install xvfb
2. 运行命令：xvfb-run python3 main.py --export my_image.eps

内容的提问来源于stack exchange，提问作者mdaneels