macromanager.py

# -*- coding: utf-8 -*-
#-------------------------------------------------
#-- macro manager
#--
#-- microelly 2015
#--
#-- GNU Lesser General Public License (LGPL)
#-------------------------------------------------


__Vers__ = '0.1b'
__Status__ = 'alpha'

import FreeCAD,PySide,os,FreeCADGui,time
from PySide import QtCore, QtGui, QtSvg
from PySide.QtGui import * 
import os,re, ast, _ast 

from configmanager import ConfigManager
cm=ConfigManager("MacroManager")
dir_name=cm.get('macrodir',FreeCAD.ConfigGet('UserAppData')+"/Mod/plugins/FreeCAD-macros")
dir_name='/home/thomas/.FreeCAD/Mod/plugins'

docstrings=['__Author__','__Version__','__Comment__','__Wiki__','__Icon__','__Help__','__Web__']



global sayexc
import sys,traceback

def sayexc(mess=''):
	exc_type, exc_value, exc_traceback = sys.exc_info()
	ttt=repr(traceback.format_exception(exc_type, exc_value,exc_traceback))
	lls=eval(ttt)
	l=len(lls)
	l2=lls[(l-3):]
	FreeCAD.Console.PrintError(mess + "\n" +"-->  ".join(l2))




def getplugin(name=''):
#	''' read the file filenam and create the parsetree '''

	# test data
	print "getplugin " + name 
	if False:
		content='''
		
__Comment__ = 'My macro is a super macro and can be used whenever other macros fail '
__Web__ = "http://forum.freecadweb.org/viewtopic.php?f=8&t=11302"
__Wiki__ = "http://www.freecadweb.org/wiki/index.php?title=Macro_FreeCAD_to_Kerkythea"
__Icon__  = "Part_Common.svg"
__Help__ = "This is the help text of this macro"
__Author__ = "Freek Ad"
__Version__ = 0.1
__Status__ = 'alpha'
__Requires__ = ''

'''
	else:
		content=open(name).read()
	try:	
		tree = ast.parse(content)
		try:
			# astpp found here
			# http://alexleone.blogspot.de/2010/01/python-ast-pretty-printer.html
			import astpp
			print astpp.dump(tree)
		except:
			pass
		print "done"
	except:
		tree=None
	return tree

def getplugindata(files):
	plugindata={}
	for plin in files:
		#print plin
		tree=getplugin(plin)
		if tree:
			doc={}
			for id in docstrings:
				doc[id]=""
			for k in tree.body:
				try:
					for t in k.targets:
						if t.id in docstrings:
							print t.lineno
							print t.id
							if k.value.__class__ == _ast.Str:
								v=k.value.s
							elif k.value.__class__ == _ast.Num:
								v=k.value.n
							elif k.value.__class__ == _ast.Name:
								v=k.value.id
							doc[t.id]=v
				except:
					pass
			plugindata[plin]=doc
	return plugindata


def findfiles(dir_name):
	macrofiles=[]
	for root, dirs, files in os.walk(dir_name, topdown=False):
		print('Found directory: %s' % root)
		for fname in files:
			print('\t%s' % fname)		
			if re.search('\.FCMacro$', fname):  
				s=root+'/'+fname
				macrofiles.append(s)
			if re.search('\.py$', fname):  
				s=root+'/'+fname
				macrofiles.append(s)

	print macrofiles
	return macrofiles


def myclick(ak,n):
	FreeCAD.Console.PrintMessage( str(n) + ' '+ str(ak.obj) + "\n")

###############
def sayexc(mess=''):
	exc_type, exc_value, exc_traceback = sys.exc_info()
	ttt=repr(traceback.format_exception(exc_type, exc_value,exc_traceback))
	lls=eval(ttt)
	l=len(lls)
	l2=lls[(l-3):]
	FreeCAD.Console.PrintError(mess + "\n" +"-->  ".join(l2))

class MyAction2():
	def __init__(self,method):
		self.cmd=method


	def run(self):
			FreeCAD.Console.PrintMessage("!"+self.cmd+"!")
			import re 
			pat=r"FreeCADGui.activateWorkbench\(\"(.*)\"\)"
			m = re.match(pat, self.cmd)
			if m:
				pre=m.group(1)
				if not pre in FreeCADGui.listWorkbenches():
					myDialoge("The Workbench \n\n*** " + pre + " ***\n\nis not available \nplease \n\n1. install it and \n2. restart FreeCAD!") 
					return
			try:
				exec(self.cmd)
			except:
				sayexc(self.cmd)


###############




def layout(widget,files,plugindata):	
	zz=[]
	n=0
	print "generat layout for Macrolist"
	print files
	for fe in files:
		print fe
		n += 1
		try:
			author=plugindata[fe]['__Author__']
			comment=plugindata[fe]['__Comment__']
			ic=plugindata[fe]['__Icon__']
			print fe,plugindata[fe]
		except:
			author="UNKNOW"
			comment="NO COMM"
			ic="No icon"
			
		base=os.path.basename(fe)
		st=base.upper() + '  ------  '
		if author:
			st += ' ---------  AUTHOR: ' +author
		if comment:
			st += ' COMMENTS:' + comment
		st +=  ' PATH:'  + fe
		
		ak0= QWidget()
		
		hlay=  QHBoxLayout()
		hlay.setSpacing(0)
		
		ak0.setLayout(hlay)
		# Hilfs button
		ak1 = QtGui.QPushButton(QtGui.QIcon('icons:Tree_Annotation.svg'), ""  )
		ak1.setMaximumWidth(20)
		hlay.addWidget(ak1)
		
		try:
			cmdh='import WebGui; WebGui.openBrowser( "' +plugindata[fe]['__Wiki__']+ '")'
		except:
			cmdh="NONE"
		
		FreeCAD.Console.PrintMessage(cmdh+"!-----------------------------------\n")
		yh=MyAction2(cmdh)
		ak1.yh=yh
		try:
			plugindata[fe]['__Wiki__']
			ak1.clicked.connect(yh.run)
			ak1.setToolTip('See WebSite Documentation')
		except:
			ak1.setIcon(QtGui.QIcon('icons:Tree_Dimension.svg'))
		
		
		# Hilfs button
		ak2 = QtGui.QPushButton(QtGui.QIcon('icons:Document.svg'), ""  )
		ak2.setMaximumWidth(20)
		hlay.addWidget(ak2)
		try:
			
			cmdh='import WebGui; WebGui.openBrowser( "' +plugindata[fe]['__Web__']+ '")'
			FreeCAD.Console.PrintMessage(cmdh+"!-----------------------------------\n")
			FreeCAD.Console.PrintMessage(cmdh+"!-----------------------------------\n")
			yh=MyAction2(cmdh)
			ak2.yh=yh
			if plugindata[fe]['__Web__'] <>'':
				ak2.clicked.connect(yh.run) 
				ak2.setToolTip('See Forum Thread')
			else:
				ak2.setIcon(QtGui.QIcon('icons:Tree_Dimension.svg'))
		except:
			sayexc()
			pass
		
		ak = QtGui.QPushButton(QtGui.QIcon(ic), st  )
		ak.obj=fe
		ak.clicked.connect(lambda ak=ak, arg=n: myclick(ak,arg))
		zz.append(ak)
		
		hlay.addWidget(ak)
		widget.lay.addWidget(ak0)
		
	return zz



class Widget(QWidget):

	def __init__(self, parent= None):
		super(Widget, self).__init__()
		self.setStyleSheet("QPushButton { text-align:left;background-color: lightblue;text-align:left;}\
		 {margin:0px;padding:0px;}")
		self.setWindowTitle("Macro Manager " + __Vers__)
		self.setObjectName("MacroManager")
		self.setWindowFlags(QtCore.Qt.WindowStaysOnTopHint)
		
		#Container Widget       
		widget = QWidget()
		
		#Layout of Container Widget
		layout = QVBoxLayout(self)
		layout.setSpacing(0)
		###layout.setAlignment(QtCore.Qt.AlignTop)
		self.lay=layout
		widget.setLayout(layout)
		scroll = QScrollArea()
		#scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
		#scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
		scroll.setWidgetResizable(True)
		scroll.setWidget(widget)
		 
		#Scroll Area Layer add
		vLayout = QVBoxLayout(self)
		vLayout.setSpacing(40)
		vLayout.setAlignment(QtCore.Qt.AlignTop)
		vLayout.addWidget(scroll)
		self.setLayout(vLayout)

def macroManager():
		files=findfiles(dir_name)
		print "files 1---"
		print files
		print "plugindaat"
		plugindata=getplugindata(files)
		print plugindata
		ww= Widget()
		print "files ---"
		print files
		buttonlist=layout(ww,files,plugindata)
		ww.show()
		return ww

t=macroManager()