FilesExpand file tree

 master

/

caves.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (43 loc) · 1.83 KB

 master

/

caves.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (43 loc) · 1.83 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

from random import choice

class Caves(object):

def __init__(self, number_of_caves):

self.number_of_caves = number_of_caves

self.cave_list = range(number_of_caves)

self.unvisited = range(number_of_caves)[1:]

self.visited = [0]

self.caves = []

self.setup_caves(number_of_caves)

self.link_caves()

def setup_caves(self, cave_numbers):

""" Create the starting list of caves """

for cave in range(cave_numbers):

self.caves.append([])

def link_caves(self):

"""Mark sure all of the caves are connected with two-way tunnels"""

while self.unvisited != []:

this_cave = self.choose_cave(self.visited)

next_cave = self.choose_cave(self.unvisited)

self.create_tunnel(this_cave, next_cave)

self.visit_cave(next_cave)

def create_tunnel(self, cave_from, cave_to):

""" Create a tunnel between cave_from and cave_to"""

self.caves[cave_from].append(cave_to)

self.caves[cave_to].append(cave_from)

def visit_cave(self, cave_number):

"""Mark a cave as visited"""

self.visited.append(cave_number)

self.unvisited.remove(cave_number)

def choose_cave(self, cave_list):

"""Pick a cave from a list, provided that the cave has less than 3 tunnels."""

cave_number = choice(cave_list)

while len(self.caves[cave_number]) >= 3:

cave_number = choice(cave_list)

return cave_number

def print_caves(self):

"""Print out the current cave structure"""

for number in self.cave_list:

print number, ":", self.caves[number]

print '---------------------------------------------------'

if __name__ == '__main__':

caves = Caves(20)

caves.print_caves()