kindred/create
1
1
Fork 0
Code Issues 37 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

Import: Apply clang format

Browse Source
This commit is contained in:
wmayer
2023-09-10 16:40:51 +02:00
committed by wwmayer
parent c9b33ea8eb
commit ed46125a38
6 changed files with 133 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

125
src/Mod/Import/stepZ.py
View File

@@ -1,11 +1,11 @@
# -*- coding: utf-8 -*-
#****************************************************************************
#* Copyright (c) 2018 Maurice <easyw@katamail.com> *
#* *
#* StepZ Import Export compressed STEP files for FreeCAD *
#* License: LGPLv2+ *
#* *
#****************************************************************************
# ****************************************************************************
# * Copyright (c) 2018 Maurice <easyw@katamail.com> *
# * *
# * StepZ Import Export compressed STEP files for FreeCAD *
# * License: LGPLv2+ *
# * *
# ****************************************************************************
# workaround for unicode in gzipping filename
# OCC7 doesn't support non-ASCII characters at the moment
@@ -34,130 +34,149 @@ import builtins
import importlib
import zipfile as zf
import zipfile as zf
# import stepZ; import importlib; importlib.reload(stepZ); stepZ.open(u"C:/Temp/brick.stpz")
def mkz_string(input):
if isinstance(input, str):
return input
else:
input = input.encode('utf-8')
input = input.encode("utf-8")
return input
####
def mkz_unicode(input):
if isinstance(input, str):
return input
else:
input = input.decode('utf-8')
input = input.decode("utf-8")
return input
####
def sayz(msg):
FreeCAD.Console.PrintMessage(msg)
FreeCAD.Console.PrintMessage('\n')
FreeCAD.Console.PrintMessage("\n")
####
def sayzw(msg):
FreeCAD.Console.PrintWarning(msg)
FreeCAD.Console.PrintWarning('\n')
FreeCAD.Console.PrintWarning("\n")
####
def sayzerr(msg):
FreeCAD.Console.PrintError(msg)
FreeCAD.Console.PrintWarning('\n')
FreeCAD.Console.PrintWarning("\n")
####
def import_stpz(fn,fc,doc):
def import_stpz(fn, fc, doc):
# sayz(fn)
ext = os.path.splitext(os.path.basename(fn))[1]
fname=os.path.splitext(os.path.basename(fn))[0]
basepath=os.path.split(fn)[0]
filepath = os.path.join(basepath,fname + u'.stp')
fname = os.path.splitext(os.path.basename(fn))[0]
basepath = os.path.split(fn)[0]
filepath = os.path.join(basepath, fname + ".stp")
tempdir = tempfile.gettempdir() # get the current temporary directory
tempfilepath = os.path.join(tempdir,fname + u'.stp')
tempdir = tempfile.gettempdir() # get the current temporary directory
tempfilepath = os.path.join(tempdir, fname + ".stp")
with builtins.open(tempfilepath, 'wb') as f: #py3
with builtins.open(tempfilepath, "wb") as f: # py3
f.write(fc)
#ImportGui.insert(filepath)
# ImportGui.insert(filepath)
if doc is None:
ImportGui.open(tempfilepath)
else:
ImportGui.open(tempfilepath,doc.Name)
ImportGui.open(tempfilepath, doc.Name)
FreeCADGui.SendMsgToActiveView("ViewFit")
try:
os.remove(tempfilepath)
except OSError:
sayzerr("error on removing "+tempfilepath+" file")
sayzerr("error on removing " + tempfilepath + " file")
###
def open(filename,doc=None):
def open(filename, doc=None):
if zf.is_zipfile(filename):
with zf.ZipFile(filename, 'r') as fz:
with zf.ZipFile(filename, "r") as fz:
file_names = fz.namelist()
for fn in file_names:
sayz(fn)
with fz.open(fn) as zfile:
file_content = zfile.read()
import_stpz(filename,file_content,doc)
import_stpz(filename, file_content, doc)
else:
with gz.open(filename, 'rb') as f:
fnm=os.path.splitext(os.path.basename(filename))[0]
with gz.open(filename, "rb") as f:
fnm = os.path.splitext(os.path.basename(filename))[0]
sayz(fnm)
file_content = f.read()
import_stpz(filename,file_content,doc)
import_stpz(filename, file_content, doc)
####
def insert(filename,doc):
def insert(filename, doc):
doc = FreeCAD.ActiveDocument
open(filename, doc)
####
def export(objs,filename):
def export(objs, filename):
"""exporting to file folder"""
#sayz(filename)
sayz("stpZ version "+___stpZversion___)
# sayz(filename)
sayz("stpZ version " + ___stpZversion___)
ext = os.path.splitext(os.path.basename(filename))[1]
fname=os.path.splitext(os.path.basename(filename))[0]
basepath=os.path.split(filename)[0]
tempdir = tempfile.gettempdir() # get the current temporary directory
fname = os.path.splitext(os.path.basename(filename))[0]
basepath = os.path.split(filename)[0]
tempdir = tempfile.gettempdir() # get the current temporary directory
filepath = os.path.join(basepath,fname) + u'.stp'
filepath_base = os.path.join(basepath,fname)
filepath = os.path.join(basepath, fname) + ".stp"
filepath_base = os.path.join(basepath, fname)
namefpath = os.path.join(basepath,fname)
namefpath = os.path.join(basepath, fname)
outfpath = os.path.join(basepath,fname)+u'.stpZ'
outfpathT = os.path.join(tempdir,fname)+u'.stpZ'
outfpath_stp = os.path.join(basepath,fname)+u'.stp'
outfpathT_stp = os.path.join(tempdir,fname)+u'.stp'
outfpath = os.path.join(basepath, fname) + ".stpZ"
outfpathT = os.path.join(tempdir, fname) + ".stpZ"
outfpath_stp = os.path.join(basepath, fname) + ".stp"
outfpathT_stp = os.path.join(tempdir, fname) + ".stp"
outfpath_base = basepath
#outfpath_str = mkz_string(os.path.join(basepath,fname))
outfpath_str = os.path.join(basepath,fname)+u'.stp'
outfpathT_str = os.path.join(tempdir,fname)+u'.stp'
# outfpath_str = mkz_string(os.path.join(basepath,fname))
outfpath_str = os.path.join(basepath, fname) + ".stp"
outfpathT_str = os.path.join(tempdir, fname) + ".stp"
if os.path.exists(outfpathT_stp):
os.remove(outfpathT_stp)
sayzw("Old temp file with the same name removed '"+ outfpathT_stp +"'")
ImportGui.export(objs,outfpathT_stp)
with builtins.open(outfpathT_stp, 'rb') as f_in:
sayzw("Old temp file with the same name removed '" + outfpathT_stp + "'")
ImportGui.export(objs, outfpathT_stp)
with builtins.open(outfpathT_stp, "rb") as f_in:
file_content = f_in.read()
new_f_content = file_content
f_in.close()
with gz.open(outfpathT_str, 'wb') as f_out:
with gz.open(outfpathT_str, "wb") as f_out:
f_out.write(new_f_content)
f_out.close()
if os.path.exists(outfpath):
shutil.move(outfpathT_str, outfpath)
#os.remove(outfpathT_stp)
# os.remove(outfpathT_stp)
else:
shutil.move(outfpathT_str, outfpath)
#os.remove(outfpathT_stp)
# os.remove(outfpathT_stp)
####