Files
30-seconds-of-code/scripts/lint.py
Rohit Tanwar 55f39b14bd new snippets
2018-02-01 13:49:59 +05:30

17 lines
616 B
Python

import autopep8
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)
#print(re.split(codeRe,fileData)[0])
formatedCode = '\n'+autopep8.fix_code(re.split(codeRe,fileData)[1]).strip()+'\n'
fileToSave = fileData.replace(originalCode,('```python'+formatedCode+'```'))
someFile = open("snippets/"+file,'w')
someFile.write(fileToSave)
someFile.close()