Files
30-seconds-of-code/scripts/pretty.py
Rohit Tanwar 109ff7e912 travis_ci
2018-01-09 02:59:40 +05:30

16 lines
556 B
Python

from yapf.yapflib.yapf_api import FormatCode
import re
import pprint
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()