else/Toolbox/fileprefix.py

'''
When you run this file from the commandline given a single argument, all
of the files in the current working directory will be renamed in the format
{argument}_{count} where argument is your cmd input and count is a zero-padded
integer that counts each file in the folder.
'''

import argparse
import os
import random
import string
import re
import sys

from voussoirkit import pathclass
from voussoirkit import safeprint

IGNORE_EXTENSIONS = ['py', 'lnk', 'ini']


def natural_sorter(x):
    '''
    http://stackoverflow.com/a/11150413
    '''
    convert = lambda text: int(text) if text.isdigit() else text.lower() 
    alphanum_key = lambda key: [convert(c) for c in re.split('([0-9]+)', key)] 
    return alphanum_key(x)

def fileprefix(
        prefix='',
        sep=' ',
        ctime=False,
        dry=False,
    ):
    current_directory = pathclass.Path('.')

    prefix = prefix.strip()
    if prefix == ':':
        prefix = current_directory.basename + ' - '
    elif prefix != '':
        prefix += sep

    filepaths = current_directory.listdir()
    filepaths = [f for f in filepaths if f.is_file]
    filepaths = [f for f in filepaths if f.extension.lower() not in IGNORE_EXTENSIONS]

    try:
        pyfile = pathclass.Path(__file__)
        filepaths.remove(pyfile)
    except ValueError:
        pass

    # trust me on this.
    zeropadding = len(str(len(filepaths)))
    zeropadding = max(2, zeropadding)
    zeropadding = str(zeropadding)

    format = '{{prefix}}{{index:0{pad}d}}{{extension}}'.format(pad=zeropadding)

    if ctime:
        print('Sorting by time')
        filepaths.sort(key=lambda x: x.stat.st_ctime)
    else:
        print('Sorting by name')
        filepaths.sort(key=lambda x: natural_sorter(x.basename))

    for (index, filepath) in enumerate(filepaths):
        extension = filepath.extension
        if extension != '':
            extension = '.' + extension

        newname = format.format(prefix=prefix, index=index, extension=extension)
        if filepath.basename != newname:
            message = filepath.basename + ' -> ' + newname
            safeprint.safeprint(message)
            if not dry:
                os.rename(filepath.absolute_path, newname)
                pass
    if dry:
        print('Dry. No files renamed.')



def fileprefix_argparse(args):
    return fileprefix(
        prefix=args.prefix,
        sep=args.sep,
        ctime=args.ctime,
        dry=args.dry,
    )

def main(argv):
    parser = argparse.ArgumentParser()

    parser.add_argument('prefix', nargs='?', default='')
    parser.add_argument('--sep', dest='sep', default=' ')
    parser.add_argument('--ctime', dest='ctime', action='store_true')
    parser.add_argument('--dry', dest='dry', action='store_true')
    parser.set_defaults(func=fileprefix_argparse)

    args = parser.parse_args(argv)
    args.func(args)

if __name__ == '__main__':
    main(sys.argv[1:])