bab49f88ef
Directly calling subprocess.Popen without checking the presence of the program is the optimal solution because: 1) Avoids race condition, where the program disappears between the check and the Popen call. 2) Popen will raise exception if the program is absent anyway, no race condition here.
281 lines
8.8 KiB
Python
281 lines
8.8 KiB
Python
# -*- coding: utf8 -*-
|
|
## @package importDWG
|
|
# \ingroup DRAFT
|
|
# \brief DWG file importer & exporter
|
|
'''
|
|
@package importDWG
|
|
ingroup DRAFT
|
|
\brief DWG file importer & exporter
|
|
|
|
This module provides support for importing and exporting Autodesk DWG files.
|
|
This module is only a thin layer that uses the ODA (formerly Teigha) File
|
|
Converter application to convert to/from DXF. Then the real work is done by
|
|
importDXF
|
|
|
|
The converter may be called
|
|
/usr/bin/TeighaFileConverter
|
|
/usr/bin/ODAFileConverter
|
|
|
|
Test files
|
|
https://knowledge.autodesk.com/support/autocad/downloads/
|
|
caas/downloads/content/autocad-sample-files.html
|
|
'''
|
|
# Check code quality with
|
|
# flake8 --ignore=E226,E266,E401,W503
|
|
|
|
# ***************************************************************************
|
|
# * Copyright (c) 2009 Yorik van Havre <yorik@uncreated.net> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program 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 program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
# TODO: use subprocess.popen() instead of subprocess.call()
|
|
|
|
import six
|
|
import FreeCAD
|
|
from FreeCAD import Console as FCC
|
|
|
|
if FreeCAD.GuiUp:
|
|
from DraftTools import translate
|
|
else:
|
|
def translate(context, txt):
|
|
return txt
|
|
|
|
# Save the native open function to avoid collisions
|
|
if open.__module__ == '__builtin__':
|
|
pythonopen = open
|
|
|
|
|
|
def open(filename):
|
|
"""Open filename and parse using importDXF.open().
|
|
|
|
Parameters
|
|
----------
|
|
filename : str
|
|
The path to the filename to be opened.
|
|
|
|
Returns
|
|
-------
|
|
App::Document
|
|
The new FreeCAD document object created, with the parsed information.
|
|
"""
|
|
dxf = convertToDxf(filename)
|
|
if dxf:
|
|
import importDXF
|
|
doc = importDXF.open(dxf)
|
|
return doc
|
|
return
|
|
|
|
|
|
def insert(filename, docname):
|
|
"""Imports a file using importDXF.insert().
|
|
|
|
If no document exist, it is created.
|
|
|
|
Parameters
|
|
----------
|
|
filename : str
|
|
The path to the filename to be opened.
|
|
docname : str
|
|
The name of the active App::Document if one exists, or
|
|
of the new one created.
|
|
|
|
Returns
|
|
-------
|
|
App::Document
|
|
The active FreeCAD document, or the document created if none exists,
|
|
with the parsed information.
|
|
"""
|
|
dxf = convertToDxf(filename)
|
|
if dxf:
|
|
import importDXF
|
|
# Warning: function doesn't return?
|
|
doc = importDXF.insert(dxf, docname)
|
|
return doc
|
|
return
|
|
|
|
|
|
def export(objectslist, filename):
|
|
"""Export the DWG file with a given list of objects.
|
|
|
|
The objects are exported with importDXF.export().
|
|
Then the result is converted to DWG.
|
|
|
|
Parameters
|
|
----------
|
|
exportList : list
|
|
List of document objects to export.
|
|
filename : str
|
|
Path to the new file.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The same `filename` input.
|
|
"""
|
|
import importDXF, os, tempfile
|
|
outdir = tempfile.mkdtemp()
|
|
_basename = os.path.splitext(os.path.basename(filename))[0]
|
|
dxf = outdir + os.sep + _basename + ".dxf"
|
|
importDXF.export(objectslist, dxf)
|
|
convertToDwg(dxf, filename)
|
|
return filename
|
|
|
|
|
|
def getTeighaConverter():
|
|
"""Find the ODA (formerly Teigha) executable.
|
|
|
|
It searches the FreeCAD parameters database, then searches for common
|
|
paths in Linux and Windows systems.
|
|
|
|
Parameters
|
|
----------
|
|
None
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The full path of the converter executable
|
|
'/usr/bin/TeighaFileConverter'
|
|
"""
|
|
import FreeCAD, os, platform
|
|
p = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Mod/Draft")
|
|
p = p.GetString("TeighaFileConverter")
|
|
if p:
|
|
# path set manually
|
|
teigha = p
|
|
else:
|
|
# try to find teigha
|
|
teigha = None
|
|
if platform.system() == "Linux":
|
|
teigha = "/usr/bin/TeighaFileConverter"
|
|
if not os.path.exists(teigha):
|
|
teigha = "/usr/bin/ODAFileConverter"
|
|
elif platform.system() == "Windows":
|
|
odadir = os.path.expandvars("%ProgramFiles%\ODA")
|
|
if os.path.exists(odadir):
|
|
subdirs = os.walk(odadir).next()[1]
|
|
for sub in subdirs:
|
|
t = (odadir + os.sep + sub + os.sep
|
|
+ "TeighaFileConverter.exe")
|
|
t = os.path.join(odadir, sub, "TeighaFileConverter.exe")
|
|
if os.path.exists(t):
|
|
teigha = t
|
|
if teigha:
|
|
if os.path.exists(teigha):
|
|
return teigha
|
|
from DraftTools import translate
|
|
_msg = ("ODA (formerly Teigha) File Converter not found, "
|
|
"DWG support is disabled")
|
|
FCC.PrintMessage(translate("draft", _msg) + "\n")
|
|
return None
|
|
|
|
|
|
def convertToDxf(dwgfilename):
|
|
"""Convert a DWG file to a DXF file.
|
|
|
|
If the converter is found it is used, otherwise the conversion fails.
|
|
|
|
Parameters
|
|
----------
|
|
dwgfilename : str
|
|
The input filename.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The new file produced.
|
|
"""
|
|
import os, tempfile, subprocess, sys
|
|
|
|
try:
|
|
outdir = tempfile.mkdtemp()
|
|
basename = os.path.basename(dwgfilename)
|
|
result = outdir + os.sep + os.path.splitext(basename)[0] + ".dxf"
|
|
proc = subprocess.Popen(("dwg2dxf", dwgfilename, "-o", result))
|
|
proc.communicate()
|
|
return result
|
|
except:
|
|
pass
|
|
|
|
teigha = getTeighaConverter()
|
|
if teigha:
|
|
indir = os.path.dirname(dwgfilename)
|
|
outdir = tempfile.mkdtemp()
|
|
basename = os.path.basename(dwgfilename)
|
|
cmdline = ('"%s" "%s" "%s" "ACAD2000" "DXF" "0" "1" "%s"'
|
|
% (teigha, indir, outdir, basename))
|
|
FCC.PrintMessage(translate("ImportDWG", "Converting:")
|
|
+ " " + cmdline + "\n")
|
|
if six.PY2:
|
|
if isinstance(cmdline, six.text_type):
|
|
encoding = sys.getfilesystemencoding()
|
|
cmdline = cmdline.encode(encoding)
|
|
subprocess.call(cmdline, shell=True) # os.system(cmdline)
|
|
result = outdir + os.sep + os.path.splitext(basename)[0] + ".dxf"
|
|
if os.path.exists(result):
|
|
FCC.PrintMessage(translate("ImportDWG",
|
|
"Conversion successful") + "\n")
|
|
return result
|
|
else:
|
|
_msg = ("Error during DWG to DXF conversion. "
|
|
"Try moving the DWG file to a directory path\n"
|
|
"without spaces and non-english characters, "
|
|
"or try saving to a lower DWG version.")
|
|
FCC.PrintMessage(translate("ImportDWG", _msg) + "\n")
|
|
return None
|
|
|
|
|
|
def convertToDwg(dxffilename, dwgfilename):
|
|
"""Convert a DXF file to a DWG file.
|
|
|
|
If the converter is found it is used, otherwise the conversion fails.
|
|
|
|
Parameters
|
|
----------
|
|
dxffilename : str
|
|
The input DXF file
|
|
dwgfilename : str
|
|
The output DWG file
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The same `dwgfilename` file path.
|
|
"""
|
|
import os, subprocess
|
|
|
|
import shutil
|
|
if shutil.which("dxf2dwg"):
|
|
proc = subprocess.Popen(("dxf2dwg", dxffilename, "-y", "-o", dwgfilename))
|
|
proc.communicate()
|
|
return dwgfilename
|
|
|
|
teigha = getTeighaConverter()
|
|
if teigha:
|
|
indir = os.path.dirname(dxffilename)
|
|
outdir = os.path.dirname(dwgfilename)
|
|
basename = os.path.basename(dxffilename)
|
|
cmdline = ('"%s" "%s" "%s" "ACAD2000" "DWG" "0" "1" "%s"'
|
|
% (teigha, indir, outdir, basename))
|
|
FCC.PrintMessage(translate("ImportDWG", "Converting:")
|
|
+ " " + cmdline + "\n")
|
|
subprocess.call(cmdline, shell=True) # os.system(cmdline)
|
|
return dwgfilename
|
|
return None
|