DlgProjectUtility.cpp

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (c) 2011 Werner Mayer <wmayer[at]users.sourceforge.net>     *
 *                                                                         *
 *   This file is part of the FreeCAD CAx development system.              *
 *                                                                         *
 *   This library is free software; you can redistribute it and/or         *
 *   modify it under the terms of the GNU Library General Public           *
 *   License as published by the Free Software Foundation; either          *
 *   version 2 of the License, or (at your option) any later version.      *
 *                                                                         *
 *   This library  is distributed in the hope that it will be useful,      *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU Library General Public License for more details.                  *
 *                                                                         *
 *   You should have received a copy of the GNU Library General Public     *
 *   License along with this library; see the file COPYING.LIB. If not,    *
 *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
 *   Suite 330, Boston, MA  02111-1307, USA                                *
 *                                                                         *
 ***************************************************************************/


#include "PreCompiled.h"
#include <sstream>
#include <QDir>
#include <QMessageBox>

#include <App/Document.h>
#include <App/PropertyStandard.h>

#include "DlgProjectUtility.h"
#include "Application.h"
#include "ui_DlgProjectUtility.h"


using namespace Gui::Dialog;

// taken from the script doctools.py
const char* doctools = 
"import os,sys,string\n"
"import xml.sax\n"
"import xml.sax.handler\n"
"import xml.sax.xmlreader\n"
"import zipfile\n"
"\n"
"# SAX handler to parse the Document.xml\n"
"class DocumentHandler(xml.sax.handler.ContentHandler):\n"
"       def __init__(self, dirname):\n"
"               self.files = []\n"
"               self.dirname = dirname\n"
"\n"
"       def startElement(self, name, attributes):\n"
"               item=attributes.get(\"file\")\n"
"               if item != None:\n"
"                       self.files.append(os.path.join(self.dirname,str(item)))\n"
"\n"
"       def characters(self, data):\n"
"               return\n"
"\n"
"       def endElement(self, name):\n"
"               return\n"
"\n"
"def extractDocument(filename, outpath):\n"
"       zfile=zipfile.ZipFile(filename)\n"
"       files=zfile.namelist()\n"
"\n"
"       for i in files:\n"
"               data=zfile.read(i)\n"
"               dirs=i.split(\"/\")\n"
"               if len(dirs) > 1:\n"
"                       dirs.pop()\n"
"                       curpath=outpath\n"
"                       for j in dirs:\n"
"                               curpath=curpath+\"/\"+j\n"
"                               os.mkdir(curpath)\n"
"               output=open(outpath+\"/\"+i,\'wb\')\n"
"               output.write(data)\n"
"               output.close()\n"
"\n"
"def createDocument(filename, outpath):\n"
"       files=getFilesList(filename)\n"
"       compress=zipfile.ZipFile(outpath,\'w\',zipfile.ZIP_DEFLATED)\n"
"       for i in files:\n"
"               dirs=os.path.split(i)\n"
"               #print i, dirs[-1]\n"
"               compress.write(i,dirs[-1],zipfile.ZIP_DEFLATED)\n"
"       compress.close()\n"
"\n"
"def getFilesList(filename):\n"
"       dirname=os.path.dirname(filename)\n"
"       handler=DocumentHandler(dirname)\n"
"       parser=xml.sax.make_parser()\n"
"       parser.setContentHandler(handler)\n"
"       parser.parse(filename)\n"
"\n"
"       files=[]\n"
"       files.append(filename)\n"
"       files.extend(iter(handler.files))\n"
"       dirname=os.path.join(dirname,\"GuiDocument.xml\")\n"
"       if os.path.exists(dirname):\n"
"               files.append(dirname)\n"
"       return files\n"
;



/* TRANSLATOR Gui::Dialog::DlgProjectUtility */

DlgProjectUtility::DlgProjectUtility(QWidget* parent, Qt::WFlags fl)
  : QDialog(parent, fl), ui(new Ui_DlgProjectUtility)
{
    ui->setupUi(this);
}

DlgProjectUtility::~DlgProjectUtility()
{
  // no need to delete child widgets, Qt does it all for us
}

void DlgProjectUtility::on_extractButton_clicked()
{
    QString source = ui->extractSource->fileName();
    QString dest = ui->extractDest->fileName();
    if (source.isEmpty()) {
        QMessageBox::critical(this, tr("Empty source"), tr("No source is defined."));
        return;
    }
    if (dest.isEmpty()) {
        QMessageBox::critical(this, tr("Empty destination"), tr("No destination is defined."));
        return;
    }

    std::stringstream str;
    str << doctools << "\n";
    str << "extractDocument(\"" << (const char*)source.toUtf8()
        << "\", \"" << (const char*)dest.toUtf8() << "\")";
    Application::Instance->runPythonCode(str.str().c_str());
}

void DlgProjectUtility::on_createButton_clicked()
{
    QString source = ui->createSource->fileName();
    QString dest = ui->createDest->fileName();
    if (source.isEmpty()) {
        QMessageBox::critical(this, tr("Empty source"), tr("No source is defined."));
        return;
    }
    if (dest.isEmpty()) {
        QMessageBox::critical(this, tr("Empty destination"), tr("No destination is defined."));
        return;
    }

    dest = QDir(dest).absoluteFilePath(QString::fromUtf8("project.fcstd"));

    std::stringstream str;
    str << doctools << "\n";
    str << "createDocument(\"" << (const char*)source.toUtf8()
        << "\", \"" << (const char*)dest.toUtf8() << "\")";
    Application::Instance->runPythonCode(str.str().c_str());

    if (ui->checkLoadProject->isChecked())
        Application::Instance->open((const char*)dest.toUtf8(),"FreeCAD");
}

#include "moc_DlgProjectUtility.cpp"