Files
30-seconds-of-code/scripts/pretty.py
Rohit Tanwar 714b2aa553 autopep8
2018-01-16 17:50:40 +05:30

15 lines
520 B
Python

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