/
create-flowchart-data.py
73 lines (54 loc) · 4.25 KB
/
create-flowchart-data.py
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
69
70
71
72
73
from neo4jrestclient import GraphDatabase
from models import Flow, Question, Answer
flows = [Flow('morrowind', "Morrowind-style", "A fantasy character creation process like the one at the start of the game Morrowind.")]
def morrowind_flow(start_node):
flow_id = 'morrowind'
question_index = db.nodes.indexes.get('questions')
q1 = Question(id = 'q1', text = 'What surrounded you when you were young?',
answers = [Answer(id = 'q1a1', text = 'The sea', reward_type = 'Background', reward_value = 'Coastal', next='q4'),
Answer(id = 'q1a2', text = 'Fields', reward_type = 'Background', reward_value = 'Rural', next = 'q2'),
Answer(id = 'q1a3', text = 'Buildings', reward_type = 'Background', reward_value = 'Urban', next = 'q2'),
Answer(id = 'q1a4', text = 'Trees', reward_type = 'Background', reward_value = 'Backwoods', next = 'q2'),])
q2 = Question(id = 'q2', text = 'When you were young your family home caught fire. You could hear your mother crying for help inside. How did you save her?',
answers = [Answer(id = 'q2a1', text = "I seized an axe and cut through the wall of the house, creating an escape route.", reward_type = 'Body', reward_value = '+1', next = 'q3'),
Answer(id = 'q2a2', text ='I gathered up a rope and covered my mouth with a damp cloth. I made my way through the smoke and followed the rope back to safety.', reward_type = 'Mind', reward_value = '+1', next = 'q3'),
Answer(id = 'q2a3', text = 'I charged into the house immediately and found her. We made it out together.', reward_type = 'Spirit', reward_value = '+1', next = 'q3')])
q3 = Question(id = 'q3', text = 'You won a prize at your county fair. What was is it that won you this accolade?',
answers = [Answer(id = 'q3a1', text = 'My skills in archery.', reward_type = 'Skill', reward_value = 'Archery'),
Answer(id = 'q3a2', text = 'I was a fearless bull leaper.', reward_type = 'Skill', reward_value = 'Acrobatics'),
Answer(id = 'q3a3', text = 'My horsemanship.', reward_type = 'Skill', reward_value = 'Riding')])
q4 = Question(id = 'q4', text = 'Did you have your own boat?',
answers = [Answer(id = 'q4a1', text ="No, the sea terrified me.", reward_type = "Trait", reward_value = "Cautious", next = 'q2'),
Answer(id = 'q4a2', text = "No, I helped crew my father's boat.", reward_type = "Trait", reward_value = "Salty dog", next = 'q2'),
Answer(id = 'q4a3', text ="Yes, I would sail her whenever I could.", reward_type = "Skill", reward_value = "Sailor", next = 'q2')])
questions = [q1, q2, q3, q4,]
current_questions = start_node.relationships.outgoing(['Question'])
for question in questions:
if not question.id in [q.end.properties['id'] for q in current_questions]:
new_question = db.nodes.create(id = question.id, text = question.text)
start_node.relationships.create('Question', new_question)
question_index[flow_id][question.id] = new_question
for question in questions:
question_node = question_index[flow_id][question.id][0]
for answer in question.answers:
answer_node = db.nodes.create(id = answer.id, text = answer.text)
reward_node = db.nodes.create(type = answer.reward_type, value = answer.reward_value)
answer_node.relationships.create('Reward', reward_node)
question_node.relationships.create('Answer', answer_node)
if answer.next and question_index[flow_id][answer.next]:
answer_node.relationships.create('Next', question_index[flow_id][answer.next][0])
if len(start_node.relationships.outgoing(['First'])) == 0:
first_question = [question.end for question in start_node.relationships.outgoing(['Question']) if question.end.properties['id'] == 'q1'][0]
start_node.relationships.create('First', first_question)
db = GraphDatabase("http://localhost:7474/db/data/")
for index_name in ['flows', 'questions']:
if not index_name in db.nodes.indexes.keys():
db.nodes.indexes.create(index_name)
print 'Created index %s' % index_name
for flow in flows:
flow_index = db.nodes.indexes.get("flows")
if not flow_index['flow']['start']:
new_flow_node = db.nodes.create(id = flow.id, title = flow.title, description = flow.description)
flow_index['flow']['start'] = new_flow_node
print "Created flow %s" % flow.title
morrowind_flow([node for node in db.nodes.indexes.get("flows")['flow']['start'] if node['id'] == "morrowind"][0])