Skip to content

Commit a20739a

Browse files
committed
Imported the valid animal list from cowsay library
1 parent ddae9b9 commit a20739a

1 file changed

Lines changed: 8 additions & 7 deletions

File tree

implement-cowsay/cow.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,16 @@
33

44
parser = argparse.ArgumentParser()
55

6-
parser.add_argument("--animal", default="cow")
6+
parser.add_argument("--animal", default="cow", choices=cowsay.char_names)
77
parser.add_argument("message", nargs="+")
88

99
args = parser.parse_args()
1010
message = " ".join(args.message)
1111

12-
if args.animal == "turtle":
13-
cowsay.turtle(message)
14-
elif args.animal == "dragon":
15-
cowsay.dragon(message)
16-
else:
17-
cowsay.cow(message)
12+
# getting list of animals names from the cowsay library, using (char_names);
13+
animals = cowsay.char_names
14+
15+
for animal in animals:
16+
if args.animal == animal:
17+
print(cowsay.get_output_string(animal, message))
18+
break

0 commit comments

Comments
 (0)