python 知识体系的原型

Posted 2021-05-09

import random
import json

# Testing for obj-action-obj node model of knowledge 
# Contains multi-node tree system

OBJECT_TYPE = 'object'
ACTION_TYPE = 'action'

class KnowledgeNode(object):

	def __init__(self, type_info, value, *children):
		self.type_info = type_info
		self.value = value
		self.children = list(children)

	def __repr__(self):
		return json.dumps(self.to_dict(), indent=4)

	def to_dict(self):
		return {
		   'type_info':self.type_info,
		   'value':self.value,
		   'children':[child.to_dict() for child in self.children]
		}

	def create_child(self, type_info, value, *children):
		# Constructs and appends a new knowledge node.
		new_child = KnowledgeNode(type_info, value, *children)
		self.children.append(new_child)

	def add_child(self, child):
		self.children.append(child)

	def __getitem__(self, key):
		return self.children[key]

	def __contains__(self, key):
		for child in self.children:
			if child.value == key:
				return True
		return False


# Special child class which creates and manipulates an object node.
class ObjectNode(KnowledgeNode):
	"""
	>>> h = ObjectNode('Apples')
	>>> h
	{"value": "Apples", "type_info": "object", "children": []}
	"""

	def __init__(self, value, *children):
		super(ObjectNode, self).__init__(OBJECT_TYPE, value, *children)


class ActionNode(KnowledgeNode):
	# Convenience class to create action node specifically

	def __init__(self, value, *children):
		super(ObjectNode, self).__init__(ACTION_TYPE, value, *children)