FilesExpand file tree

 master

/

solution.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

52 lines (37 loc) · 1.33 KB

 master

/

solution.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

52 lines (37 loc) · 1.33 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

#!/usr/bin/env python3

"""Emulate wc (word count)"""

import argparse

import sys

# --------------------------------------------------

def get_args():

"""Get command-line arguments"""

parser = argparse.ArgumentParser(

description='Emulate wc (word count)',

formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument('file',

metavar='FILE',

nargs='*',

default=[sys.stdin],

type=argparse.FileType('rt'),

help='Input file(s)')

return parser.parse_args()

# --------------------------------------------------

def main():

"""Make a jazz noise here"""

args = get_args()

total_lines, total_bytes, total_words = 0, 0, 0

for fh in args.file:

num_lines, num_words, num_bytes = 0, 0, 0

for line in fh:

num_lines += 1

num_bytes += len(line)

num_words += len(line.split())

total_lines += num_lines

total_bytes += num_bytes

total_words += num_words

print(f'{num_lines:8}{num_words:8}{num_bytes:8} {fh.name}')

if len(args.file) > 1:

print(f'{total_lines:8}{total_words:8}{total_bytes:8} total')

# --------------------------------------------------

if __name__ == '__main__':

main()