import os

# 获取目录下的子目录
def subdirs(path):
    dl = [];
    for i in os.walk(path, False):
        for d in i[1]:
            dl.append(os.path.join(path, d))
    return dl

# 获取目录下的子文件
def subfiles(path):
    fl = [];
    for i in os.walk(path, False):
        for f in i[2]:
            fl.append(os.path.join(path, f))
    return fl

# 根据文件创建时间对文件进行排序
def fsort(files):
    files.sort(key = lambda s: os.path.getctime(s))
    return files

# 递增数列生成器
class gen:
    def __init__(self, prefix = '', suffix = '', seed = 1, digit = 3):
        self.prefix = prefix
        self.suffix = suffix
        self.seed = seed
        self.digit = digit
        self.it = seed - 1
    
    def gennext(self):
        self.it += 1
        return '{0}{1:0{3}d}{2}'.format(self.prefix, self.it, self.suffix, self.digit)

    def copy(self):
        return gen(self.prefix, self.suffix, self.seed, self.digit)

class renamer:
    def __init__(self, files, ngen = None, delsrc = False, output = ''):
        self.it = 0
        self.total = 0
        self.files = files
        self.delsrc = delsrc
        self.target = output
        if(ngen is None): self.namegen = gen()
        elif(isinstance(ngen, gen)): self.namegen = ngen
        else: raise(TypeError, 'ngen参数只接受gen类型的对象')

    def rename(src, des, delsrc):
        os.rename(src, des)
        if(delsrc == True):
            os.remove(src)

    def run(self):
        self.it = 0
        self.total = len(self.files)
        for f in self.files:
            dn = os.path.dirname(f)
            ex = os.path.splitext(f)
            nn = self.namegen.gennext() + ex[1]
            np = ''
            sd = self.target == ''
            if(sd):
                np = os.path.join(dn, nn)
            else:
                np = os.path.join(self.target, nn)
            print('rename: %s' %(np))
            renamer.rename(f, np, sd and self.delsrc)
            self.it += 1
        print('%d work(s) done' %(self.it))
        print()

# 重命名给定列表中的所有文件
def fs_rename(files, resort = False, ngen = gen(), delsrc = False, output = ''):
    if(resort): fsort(files)
    rn = renamer(files, ngen.copy(), delsrc, output)
    rn.run()

# 重命名给定目录中的所有文件
def ds_rename(dirs, resort = False, ngen = gen(), delsrc = False, output = ''):
    for d in dirs:
        fs = subfiles(d)
        if(resort): fsort(fs)
        print('%d file(s) in %s' %(len(fs), d))
        if(output == '') ngen = ngen.copy()
        rn = renamer(fs, ngen, delsrc, output)
        rn.run()