Python2017/labs05/task04.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
Implementacja narzedzia ``wc`` z linuksa (word counter).
Zwraca liczbę słów, znaków i linii.

** zad. 4 (Domowe) **
Plik ``task04.py`` zawiera kod prorgamu, który działa jak popularne narzędzie unixowe ``wc`` (Word Counter): zlicza liczbę linii, wyrazów i znaków.
Aktualnie program potrafi działać wyłącznie na wejściu podanym z klawiatury. Dodaj do niego opcje programu:
 * domyślnie program ma zliczać na wejściu z klawiatury (stdin) i wyświetlać wszystkie 3 liczby.
 * Jeżeli został podany przełącznik `-l`, to to ma zostać zwrócona tylko liczba linii.
 * Jeżeli został podany przełącznik `-w`, to to ma zostać zwrócona tylko liczba słów.
 * Jeżeli został podany przełącznik `-c`, to to ma zostać zwrócona tylko liczba znaków.
 * Jeżeli został podany inny argument, to należy założyć że jest to nazwa pliku i potraktować ten plik jako wejście do programu.
"""

import sys
import argparse


def count_lines(text):
    """ return number of lines. """
    return len(text.strip().split('\n'))

def count_words(text):
    """ return number of words. """
    return sum([len([1 for word in line.split(' ') if len(word)])
                for line in text.split('\n')])

def count_chars(text):
    """ return number of characters. """
    return len(text)

def wc(text):
    """ proper wc """
    lines = count_lines(text)
    words = count_words(text)
    chars = count_chars(text)
    return lines, words, chars


def main():
    """ main """
    parser = argparse.ArgumentParser()
    parser.add_argument('-l', help="number of lines", action="store_true")
    parser.add_argument('-w', help="number of words", action="store_true")
    parser.add_argument('-c', help="number of chars", action="store_true")
    parser.add_argument('-filename', type=argparse.FileType('r'), default='-', help='filename to read from')
    args = parser.parse_args()

    if args.filename:
        lines, words, chars = wc(args.filename.read())
    elif not sys.stdin.isatty():
        lines, words, chars = wc(args.stdin.read())
    else:
        pass

    if(args.l):
        print(lines)
    elif(args.w):
        print(words)
    elif(args.c):
            print(chars)
    else:
        print(lines, words, chars)


if __name__ == "__main__":
    main()