Google python copyspecial.py(python 2.7)

wdx1992828

#!/usr/bin/python

import sys
import re
import os
import shutil
import commands
import zipfile

"""Copy Special exercise
"""

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

def get_specials(dir):
  com='ls '+ dir
  status,output=commands.getstatusoutput(com)
  specials=[]
  for special in output.split():
    if re.findall(r'__(\w+)__',special):
      specials.append(special)
  return specials

def to_dir(dir,srcs):
  if not os.path.exists(dir):
    os.mkdir(dir)
  for src in srcs:
    shutil.copy(src,dir)

#s1:
def to_zip(zf,srcs):
  f=zipfile.ZipFile(zf,'w')
  for src in srcs:
    f.write(src)
  f.close()

#s2
def to_zip(zf,srcs):
  
  com='zip '+ zf +' '+' '.join(srcs)
  print com
  status,output=commands.getstatusoutput(com)
  if not status:
    print 'zip successed!'

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)
  
  specials=get_specials(args[0])
  if todir:
    to_dir(todir,specials)
  if tozip:
    to_zip(tozip,specials)
  
if __name__ == "__main__":
  main()