批量修改名字的脚本

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)