AppFlow/FITK_Kernel/FITKPython/PythonAgent.cpp

#include "PythonAgent.h"
#include "FITKCore/FITKDirFileTools.h"
#include <QFile>
#include <QFileInfo>
#include <QDir>
#include <QMutex>
#include <QMutexLocker>
#include <QWidget>
#include <QVBoxLayout>
#include <QTextStream>
#include <PythonQt.h>
#include <PythonQt_QtAll.h>
#include <PythonQtScriptingConsole.h>

namespace Python
{
    QMutex PythonAgent::m_mutex;

    PythonAgent::PythonAgent()
    {
        //PythonQt初始化
        PythonQt::init(PythonQt::IgnoreSiteModule | PythonQt::RedirectStdOut);
        PythonQt_QtAll::init();

        //PythonQt中Main模块获取
        m_mainModule = PythonQt::self()->getMainModule();

        //添加Mian模块至可视化界面中
        m_Console = new PythonQtScriptingConsole(nullptr, m_mainModule);
    }

    PythonAgent::~PythonAgent()
    {
        if (m_Console != nullptr){
            delete m_Console;
            m_Console = nullptr;
        }
    }

    void PythonAgent::addPyQtDecorator(QObject * obj)
    {
        if (obj == nullptr)return;
        PythonQt::self()->addDecorators(obj);
    }

    void PythonAgent::registerPyQtCPPClass(const char * typeName, const char * parentTypeName, const char * package)
    {
        //添加互斥锁
        QMutexLocker mutexLoc(&m_mutex);

        if (classIsExist(typeName)){
            qDebug() << "Failed to register " << typeName << " to PythonQt. The typename already exists!";
            return;
        }
        PythonQt::self()->registerCPPClass(typeName, parentTypeName, package);
        m_regClassNames << typeName;
    }

    void PythonAgent::submitScriptFile(const QString &ScriptFile)
    {
        QFile file(ScriptFile);
        if (!file.open(QIODevice::ReadOnly | QIODevice::Text))return;
        QFileInfo fileInfo(file);

        //获取脚本名称与脚本路径
        QString filePath = fileInfo.absolutePath();
        QString fileName = fileInfo.fileName();

        //python系统模块导入，添加添加脚本目录
        submitScript("import sys");
        submitScript(QString("sys.path.append(\"%1\")").arg(filePath));
        submitScript(QString("import %1").arg(fileName.remove(".py")));
        file.close();
    }

    void PythonAgent::submitScript(const QString &Script)
    {
        //添加互斥锁
        QMutexLocker mutexLoc(&m_mutex);

        m_Console->insertCompletion(Script);
        m_Console->executeLine(false);
    }

    void PythonAgent::submitScript(const QStringList &Scripts)
    {
        //添加互斥锁
        QMutexLocker mutexLoc(&m_mutex);

        for (auto scr : Scripts){
            m_Console->insertCompletion(scr);
            m_Console->executeLine(false);
        }
    }

    bool PythonAgent::saveScriptHistory(const QString &filePath)
    {
        //针对目标文件及其目录不存在逻辑处理
        QFile file(filePath);
        if (!file.exists()){
            QFileInfo fileinfo(file);
            QDir dir;
            if (!dir.exists(fileinfo.absolutePath())) Core::CreateDir(fileinfo.absolutePath());
        }

        if (!file.open(QIODevice::WriteOnly))return false;

        //书写历史脚本至目标文件
        QTextStream fstream(&file);
        for (auto script : m_Console->history()){
            fstream << script << endl;
        }
        file.close();
        return true;
    }

    QWidget * PythonAgent::getPyWidget()
    {
        return m_Console;
    }

    bool PythonAgent::classIsExist(const char * typeName)
    {
        if (m_regClassNames.contains(typeName))return true;
        return false;
    }
}