Exemplo n.º 1
0
def rename_snapshot_files():
    filenames = glob('*_snap.*')
    print(len(filenames))

    maxlen = 0

    for oldfilename in filenames:
        newlen = len(oldfilename.split('_')[0])
        if newlen > maxlen:
            maxlen = newlen

    newfilenames = []

    for oldfilename in filenames:
        ext = os.path.splitext(oldfilename)[1][1:]
        splitname = oldfilename.split('_')
        filenumlen = len(splitname[0])
        if filenumlen < maxlen:
            splitname[0] = pad_left_zeros(splitname[0], maxlen)
        newfilename = 'drop_{}.{}'.format(splitname[0], ext)
        os.rename(oldfilename, newfilename)
        print(oldfilename + ' -> ' + newfilename)
        newfilenames.append(newfilename)

    return newfilenames
Exemplo n.º 2
0
def rename_new_droplet_files():
    """Rename all the original droplet files from 'clusterXXX.xyz' to
    'drop_XXX.xyz', where the number is left-padded with zeros.
    """

    filenames = glob('cluster*.xyz')
    print(filenames)

    # Store the maximum length of the internal number.
    maxlen = 0

    # Unfortunately, two full traversals need to be performed.
    # 1. Find the maximum length of the internal number and store it.
    for oldfilename in filenames:
        numstr = oldfilename[7:-4]
        newlen = len(numstr)
        if newlen > maxlen:
            maxlen = newlen

    newfilenames = []

    # 2. Go through each file again and rename.
    for oldfilename in filenames:
        numstr = oldfilename[7:-4]
        filenumlen = len(numstr)
        if filenumlen < maxlen:
            numstr = pad_left_zeros(numstr, maxlen)
        newfilename = 'drop_{}.xyz'.format(numstr)
        os.rename(oldfilename, newfilename)
        print(oldfilename + ' -> ' + newfilename)
        newfilenames.append(newfilename)

    return newfilenames