save_plugin.py

来自「这是一个支持向量机的代码」· Python 代码 · 共 25 行

from util import GetFilename
from common_interface import File
import shutil

plugin_name  = "Save to File"

def create(callback):
    filename = GetFilename()
    if not filename: return None
    return SaveToFile(filename)

class SaveToFile:
    input_description = [('input', File)]
    output_description = [('output', None)]
    def __init__(self,filename):
        self.filename = filename
    def configure(self):
        filename = GetFilename(self.filename)
        if filename: 
            self.filename = filename
            return True
    def run(self,input):
        shutil.copyfile(input.pathname, self.filename)
        return {'output': input}