Files
30-seconds-of-code/scripts/pretty.py
2018-01-09 03:19:50 +05:30

15 lines
542 B
Python

from yapf.yapflib.yapf_api import FormatCode
import re
import os
files = os.listdir('snippets')
codeRe = "```\s*python([\s\S]*?)```"
for file in files:
someFile = open("snippets/" + file)
fileData = someFile.read()
someFile.close()
originalCode = re.search(codeRe,fileData).group(0)
formatedCode = FormatCode(re.split(codeRe,fileData)[1])
fileToSave = fileData.replace(originalCode,('```python \n'+formatedCode[0]+'\n```'))
someFile = open("snippets/"+file,'w')
someFile.write(fileToSave)
someFile.close()