批量修改名字的脚本

因为leetcode写到了三位数的题目,和之前两位数的题目的顺序就发生了冲突,所以想在前面补上0,让最前面的数字是4个:

import os
"""
用途:在题目开头补上若干个0使其开头数字为4个
"""
def countStartnum(s):
    count = 0
    for i in s:
        if i in "1234567890":
            count += 1
        else:
            break
    return count
leetcode_name = os.listdir('./leetcode')
for filename in leetcode_name:
    count = countStartnum(filename)
    new_name = '0'*(4-count) + filename
 
    os.rename('./leetcode/'+filename,'./leetcode/'+new_name)

 还有点细节上的问题,比如遇到开头四个以上数字的文件会报错,大家看着改改好了