wccn.py

#!/usr/bin/env python3
# This file is kept for backward compatibility
# The functionality has been moved to the chinese_wordcount package

from chinese_wordcount.counter import count_file, walk

if __name__ == "__main__":
    import sys
    
    if len(sys.argv) != 4:
        print("usage: python chinese_wc.py [dir] [depth] [appendix]\r\nlike: python chinese_wc.py ~/Documents/org 3 \'.org\'")
    else:
        dir = sys.argv[1]  # directory
        depth = int(sys.argv[2])
        appendix = sys.argv[3]

        print("the directory is: ", dir)

        files = walk(dir, depth)
        for f in files:
            for e in f[2]:
                if e.endswith(appendix):
                    print("{count}\t{filename}".format(filename=f[0]+'/'+e, count=count_file(f[0]+'/'+e)))