FilesExpand file tree

 master

/

solution6_map.py

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

executable file

·

68 lines (47 loc) · 1.55 KB

 master

/

solution6_map.py

Top

File metadata and controls

• Code
• Blame

executable file

·

68 lines (47 loc) · 1.55 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

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

#!/usr/bin/env python3

"""Ransom note"""

import argparse

import os

import random

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

def get_args():

"""Get command-line arguments"""

parser = argparse.ArgumentParser(

description='Ransom Note',

formatter_class=argparse.ArgumentDefaultsHelpFormatter)

parser.add_argument('text', metavar='text', help='Input text or file')

parser.add_argument('-s',

'--seed',

help='Random seed',

metavar='int',

type=int,

default=None)

args = parser.parse_args()

if os.path.isfile(args.text):

args.text = open(args.text).read().rstrip()

return args

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

def main():

"""Make a jazz noise here"""

args = get_args()

random.seed(args.seed)

# Method 6: map

ransom = map(choose, args.text)

print(''.join(ransom))

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

def choose(char):

"""Randomly choose an upper or lowercase letter to return"""

return char.upper() if random.choice([0, 1]) else char.lower()

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

def test_choose():

"""Test choose"""

state = random.getstate()

random.seed(1)

assert choose('a') == 'a'

assert choose('b') == 'b'

assert choose('c') == 'C'

assert choose('d') == 'd'

random.setstate(state)

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

if __name__ == '__main__':

main()