started a base for the get3Dmodels script
This commit is contained in:
easyw
199
action_menu_get3D_models.py
Normal file
199
action_menu_get3D_models.py
Normal file
@ -0,0 +1,199 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# licence GPL 2
|
||||
# copyright easyw
|
||||
|
||||
__version__='1.2.2'
|
||||
|
||||
|
||||
import codecs
|
||||
import re,os, sys
|
||||
|
||||
import urllib2
|
||||
import ssl
|
||||
|
||||
ctx = ssl.create_default_context()
|
||||
ctx.check_hostname = False
|
||||
ctx.verify_mode = ssl.CERT_NONE
|
||||
|
||||
|
||||
KISYS3DMOD=ur"C:\KiCad-stable\share\kicad\modules\packages3d"
|
||||
KISYS3DMOD=ur"C:\Program Files (x86)\KiCad\share\kicad\modules\packages3d1"
|
||||
KISYS3DMOD=KISYS3DMOD.replace('\\','/')
|
||||
|
||||
global fileOut
|
||||
|
||||
fileOut=ur"log-3Dmodels.txt"
|
||||
fileOut=fileOut.replace('\\','/')
|
||||
|
||||
filename=ur"C:\Temp\blinky-dev.kicad_pcb"
|
||||
filename=filename.replace('\\','/')
|
||||
|
||||
|
||||
import argparse
|
||||
|
||||
args=sys.argv
|
||||
#for a in args:
|
||||
# print a
|
||||
|
||||
|
||||
def collect_models(fn, kv, k3d, fupd):
|
||||
"""collect models from GH for the present pcb"""
|
||||
|
||||
global fileOut
|
||||
|
||||
overwrite=False
|
||||
if fupd=='/u':
|
||||
overwrite=True
|
||||
filename=fn
|
||||
KISYS3DMOD=k3d
|
||||
kicad_version=kv
|
||||
|
||||
txtFile = codecs.open(filename, mode='rb', encoding='utf-8', errors='replace', buffering=1) #test maui utf-8
|
||||
content = txtFile.readlines()
|
||||
content.append(" ")
|
||||
txtFile.close()
|
||||
data=''.join(content)
|
||||
|
||||
model_list = re.findall(r'\(model\s+(.+?)\.wrl',data)
|
||||
#model_list = re.findall(r'\(model\s+(.+?)\.wrl',data, re.MULTILINE|re.DOTALL)
|
||||
model_list2 = re.findall(r'\(model\s+(.+?)\.WRL',data)
|
||||
model_list=model_list+model_list2
|
||||
model_list2 = re.findall(r'\(model\s+(.+?)\.stp',data)
|
||||
model_list=model_list+model_list2
|
||||
model_list2 = re.findall(r'\(model\s+(.+?)\.step',data)
|
||||
model_list=model_list+model_list2
|
||||
model_list2 = re.findall(r'\(model\s+(.+?)\.STP',data)
|
||||
model_list=model_list+model_list2
|
||||
model_list2 = re.findall(r'\(model\s+(.+?)\.STEP',data)
|
||||
model_list=model_list+model_list2
|
||||
|
||||
#print model_list
|
||||
|
||||
fOut=codecs.open(fileOut, mode='wb', encoding='utf-8', errors='replace', buffering=1)
|
||||
|
||||
parsed=[]
|
||||
fOut.write(u'parsed '+str(len(model_list))+u' model(s)'+ os.linesep)
|
||||
for d in model_list:
|
||||
found=False
|
||||
d1=d.replace('\\','/')
|
||||
#d1=d1.rstrip('/')
|
||||
if 'KISYS3DMOD' in d1:
|
||||
d1=d1.lstrip('${KISYS3DMOD}/')
|
||||
d1=d1.lstrip('/')
|
||||
d=d1
|
||||
local=False
|
||||
#print d1
|
||||
if 'KIPRJMOD' in d1:
|
||||
#print 'LOCAL'
|
||||
#stop
|
||||
local=True
|
||||
d1=d1.lstrip('${KIPRJMOD}')
|
||||
d1=d1.lstrip('/')
|
||||
print d1
|
||||
d=d1
|
||||
if d not in parsed:
|
||||
if local:
|
||||
fOut.write(d+u'.wrl'+ os.linesep)
|
||||
directory=d1
|
||||
prefix=""
|
||||
else:
|
||||
fOut.write(KISYS3DMOD+os.sep+d+u'.wrl'+ os.linesep)
|
||||
directory=KISYS3DMOD+os.sep+d1
|
||||
prefix=KISYS3DMOD+os.sep
|
||||
#print d1
|
||||
##stop
|
||||
#print d1.rfind('/')
|
||||
#stop
|
||||
if d1.rfind('/')!=-1:
|
||||
d1=d1[0:d1.rindex('/')]
|
||||
#else:
|
||||
|
||||
#print 'd, d1, dir'
|
||||
#print d
|
||||
#print d1
|
||||
#print directory
|
||||
#stop
|
||||
if not os.path.exists(directory):
|
||||
try:
|
||||
os.makedirs(directory)
|
||||
except:
|
||||
print 'you need Administrative rights'
|
||||
continue
|
||||
if os.path.exists(prefix+d+u'.wrl'):
|
||||
found=True
|
||||
print u'exists '+d+u'.wrl'
|
||||
#fOut.write(d)
|
||||
#fOut.write('\n')
|
||||
if not found or overwrite:
|
||||
fileDest=prefix+d+u'.wrl'
|
||||
d1=d.replace('\\','/')
|
||||
try:
|
||||
response = urllib2.urlopen(github_src+d1+u'.wrl', context=ctx)
|
||||
myfile = response.read()
|
||||
f_dest=codecs.open(fileDest, mode='wb', encoding='utf-8', errors='replace', buffering=1)
|
||||
f_dest.write(myfile)
|
||||
print fileDest + ur' written'
|
||||
except:
|
||||
print 'error file not found'
|
||||
print github_src+d1+u'.wrl'
|
||||
print 'or you may need Administrative rights'
|
||||
#print myfile
|
||||
|
||||
found=False
|
||||
if os.path.exists(prefix+d+u'.step') or os.path.exists(prefix+d+u'.STEP') or \
|
||||
os.path.exists(prefix+d+u'.stp') or os.path.exists(prefix+d+u'.STP'):
|
||||
print u'exists '+d+u'.step'
|
||||
#fOut.write(d)
|
||||
#fOut.write('\n')
|
||||
found=True
|
||||
if not found or overwrite:
|
||||
fileDest=prefix+d+u'.step'
|
||||
d1=d.replace('\\','/')
|
||||
#print github_src+d1+u'.step'
|
||||
try:
|
||||
response = urllib2.urlopen(github_src+d1+u'.step', context=ctx)
|
||||
myfile = response.read()
|
||||
f_dest=codecs.open(fileDest, mode='wb', encoding='utf-8', errors='replace', buffering=1)
|
||||
f_dest.write(myfile)
|
||||
print fileDest + ur' written'
|
||||
except:
|
||||
print 'error file not found'
|
||||
print github_src+d1+u'.step'
|
||||
print 'or you may need Administrative rights'
|
||||
#print myfile
|
||||
#fOut.write('\n')
|
||||
parsed.append(d)
|
||||
|
||||
fOut.close()
|
||||
#print model_list2
|
||||
##
|
||||
|
||||
if len(args) >= 4:
|
||||
KISYS3DMOD=args[1]
|
||||
kicad_ver=args[2]
|
||||
filename=args[3]
|
||||
force_upd=None
|
||||
if len(args) == 5:
|
||||
force_upd=args[4]
|
||||
print args[1],args[2],args[3]
|
||||
if force_upd is not None:
|
||||
print
|
||||
print 'forcing download and override all 3D models'
|
||||
else:
|
||||
print
|
||||
if kicad_ver=='4':
|
||||
github_src=ur'https://github.com/KiCad/kicad-library/raw/master/modules/packages3d/'
|
||||
gh_base=ur'https://github.com/KiCad/kicad-library/modules/packages3d/'
|
||||
else:
|
||||
github_src=ur'https://github.com/KiCad/kicad-packages3D/raw/master/'
|
||||
gh_base=ur'https://github.com/KiCad/kicad-packages3D/'
|
||||
print
|
||||
print 'collecting packages from:'
|
||||
print gh_base
|
||||
print
|
||||
|
||||
collect_models(filename, kicad_ver, KISYS3DMOD, force_upd)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user