copyspecial.py

#!/usr/bin/python
# Copyright 2010 Google Inc.
# Licensed under the Apache License, Version 2.0
# http://www.apache.org/licenses/LICENSE-2.0

# Google's Python Class
# http://code.google.com/edu/languages/google-python-class/

import sys
import re
import os
import shutil

"""Copy Special exercise
"""


# +++your code here+++
# Write functions and modify main() to call them


def get_special_paths(dir):
    """Given a dirname, returns a list of all its special files."""
    result = []
    paths = os.listdir(dir)  # list of paths in that dir
    for fname in paths:
        match = re.search(r'__(\w+)__', fname)
        if match:
            result.append(os.path.abspath(os.path.join(dir, fname)))
    return result


def copy_to(paths, to_dir):
    """Copy all of the given files to the given dir, creating it if necessary."""
    if not os.path.exists(to_dir):
        os.mkdir(to_dir)
    for path in paths:
        fname = os.path.basename(path)
        shutil.copy(path, os.path.join(to_dir, fname))
        # could error out if already exists os.path.exists():


def zip_to(paths, zipfile):
    """Zip up all of the given files into a new zip file with the given name."""
    cmd = 'zip -j ' + zipfile + ' ' + ' '.join(paths)
    print("Command I'm going to do:" + cmd)
    os.system(cmd)


def main():
    # This basic command line argument parsing code is provided.
    # Add code to call your functions below.

    # Make a list of command line arguments, omitting the [0] element
    # which is the script itself.
    args = sys.argv[1:]
    if not args:
        print("usage: [--todir dir][--tozip zipfile] dir [dir ...]")
        sys.exit(1)

    # todir and tozip are either set from command line
    # or left as the empty string.
    # The args array is left just containing the dirs.
    todir = ''
    if args[0] == '--todir':
        todir = args[1]
        del args[0:2]

    tozip = ''
    if args[0] == '--tozip':
        tozip = args[1]
        del args[0:2]

    if len(args) == 0:
        print("error: must specify one or more dirs")
        sys.exit(1)

    # +++your code here+++

    # Call your functions
    paths = []
    for dirname in args:
        paths.extend(get_special_paths(dirname))

    if todir:
        copy_to(paths, todir)
    elif tozip:
        zip_to(paths, tozip)
    else:
        print('\n'.join(paths))
    # LAB(end solution)


if __name__ == "__main__":
    main()