low2upper.py

#!/usr/bin/env python3
# coding=utf-8
"""Rename filename, lower to upper.

:author: Jon Jiang
:email: jiangyingming@live.com
"""
from textwrap import shorten
import argparse
import glob
import itertools
import os
import shutil
import sys


def low2upper(src_file, out_dir, keep_src):
    """Rename filename, lower to upper:
    1. If out_dir is None, rename original file;
    2. If out_dir is not None and keep_src is true, rename using copy;
    3. If out_dir is not None and keep_src is false, rename using move.
    """
    src_dir, filename = os.path.split(src_file)
    if out_dir is None:
        dst_file = os.path.join(src_dir, filename.upper())
    else:
        dst_file = os.path.join(out_dir, filename.upper())

    if keep_src:
        shutil.copy2(src_file, dst_file)
    else:
        shutil.move(src_file, dst_file)


def init_args():
    """Initilize function, parse user input."""
    # initilize a argument parser
    parser = argparse.ArgumentParser(
        description="Rename filename, lower to upper."
    )
    # add arguments
    parser.add_argument('-v', '--version', action='version',
                        version='%(prog)s 0.2.1')
    parser.add_argument('-k', '--keep', action='store_true',
                        help='keep original file')
    parser.add_argument('-r', '--recursive', action='store_true',
                        help='search file recursively')
    parser.add_argument('-out', metavar='<directory>', default=None,
                        help='output directory, [default: original folder]')
    parser.add_argument('files', metavar='<file>', nargs='+',
                        help='file will be processed')

    return parser.parse_args()


def main():
    """Main function."""
    args = init_args()
    globstrs, out_dir = args.files, args.out
    keep_src, recursive = args.keep, args.recursive
    # if the out_dir is None and --keep is setted, process as an error
    if keep_src and out_dir is None:
        err_message = 'Error! Blank output directory is conflict with --keep.'
        print(err_message, file=sys.stderr)
        return 1
    # start process
    if out_dir is not None:
        os.makedirs(out_dir, exist_ok=True)
    # collect input globstrs into a glob list
    print('Start processing: {}'.format(shorten(', '.join(globstrs), 62)))
    globs = [glob.iglob(globstr, recursive=recursive) for globstr in globstrs]
    count = 0
    for src_file in itertools.chain(*globs):
        low2upper(src_file, out_dir, keep_src)
        count += 1

    print('{} files have been processed.'.format(count))

    return 0


if __name__ == '__main__':
    main()