SASDocumentation/SASParser.py at master · InsolublePancake/SASDocumentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import sys
import os
import re

from SASObjects.SASProgram import SASProgram
from SASAnalysis.SASFlowChart import SASFlowChart

class SASParser(object):

	def __init__(self, codepath, docpath):

		self.codepath = codepath
		self.docpath = docpath

		self.outDir = os.path.join(docpath,'source','code')

		if not os.path.exists(self.outDir):
			os.makedirs(self.outDir)

		with open(os.path.join(self.outDir,'code.rst'),'w+') as codeRST:
			codeRST.write('Code\n====\n\n.. toctree::\n   :maxdepth: 2\n   :glob:\n\n   *')

		for root, dirs, files in os.walk(self.codepath):
			for name in files:
				if len(re.findall('\.sas$',name,flags=re.IGNORECASE))>0:
					SASfile = os.path.join(root,name)
					print("Parsing {}\r".format(SASfile))

					parsedSASFile = SASProgram(SASfile)
					flowChart = SASFlowChart(parsedSASFile)
					flowChartImg = os.path.join(self.outDir,re.sub('\s','',re.sub('.sas$','.png',flowChart.SASProgram.fileName,flags=re.IGNORECASE)))
					flowChart.saveFig(flowChartImg)

					self.writeMD(parsedSASFile,self.outDir,flowChartImg=flowChartImg)


	def writeMD(self,SASProgram,outpath,flowChartImg=None):

		mdPath = os.path.join(outpath,re.sub('.sas$','.md',SASProgram.fileName,flags=re.IGNORECASE))
		with open(mdPath, 'w+') as out:
			out.write('# {}\n\n'.format(SASProgram.name))
			if SASProgram.about is not None:
				out.write('## About\n')
				out.write('{}\n\n'.format(SASProgram.about))
			if flowChartImg is not None:
				out.write('## Program Struture\n\n')
				out.write('![Program Structure]({})\n\n'.format(os.path.basename(flowChartImg)))
			if len(SASProgram.libnames) > 0:
				out.write('## Libname(s)\n\n')
				out.write('| Name | Location |\n')
				out.write('| --- | --- |\n')
				for libname in SASProgram.libnames:
					out.write('| {} | [{}]({}) |\n'.format(libname.name,libname.path,libname.posixPath))
				out.write('\n\n')
			if len(SASProgram.includes) > 0:
				out.write('## Include(s)\n\n')
				out.write('| Path |\n')
				out.write('| --- |\n')
				for include in SASProgram.includes:
					out.write('| [{}]({}) |\n'.format(include.path,include.posixPath))
				out.write('\n\n')
			if len(SASProgram.macros)> 0:
				out.write('## Macro(s)\n')
				for macro in SASProgram.macros:
					out.write('### {}\n'.format(macro.name))
					out.write('#### About\n')
					out.write('{}\n\n'.format(macro.docString))
					if len(macro.help)>0:
						out.write('#### Help\n')
						out.write('{}\n\n'.format(macro.help))
					if len(macro.arguments)>0:
						out.write('#### Argument(s)\n\n')
						out.write('| Name | Type | Default Value | About |\n')
						out.write('| --- | --- | --- | --- |\n')
						for arg in macro.arguments:
							out.write('| {} | {} | {} | {} |\n'.format(arg.name,arg.type,arg.defaultValue,arg.docString))
					out.write('\n\n')
			if len(SASProgram.uniqueDataItems) > 0:
				out.write('## Datasets(s)\n\n')
				out.write('| Library | Name |\n')
				out.write('| --- | --- |\n')
				for dataItem in SASProgram.uniqueDataItems:
					out.write('| {} | {} |\n'.format(dataItem[0],dataItem[1]))
				out.write('\n\n')

			out.write('## Full code:\n\n<details><summary>Show/Hide</summary>\n\n')
			out.write('~~~~sas\n\n')
			out.write(SASProgram.rawProgram)
			out.write('\n\n~~~~\n\n')
			out.write('</details>\n\n')
			out.write('## Properties\n\n')
			out.write('| Meta | Property |\n| --- | --- |\n')
			out.write('| **Author:** | |\n')
			out.write('| **Path:** | *{}* |\n'.format(SASProgram.filePath))
			out.write('| **Last updated:** | *{}* |\n'.format(SASProgram.LastUpdated))


if __name__ == "__main__":
	if len(sys.argv)>2:
		path = sys.argv[1]
		out = sys.argv[2]
		parser = SASParser(path,out)