from pprint import pprint
import requests
import json

headers = {
	"token": '369ca613c7a74a58b0f95be2cfd59257',
	"Content-Type": "application/json",
}

json_obj = {"query": "山东师范", # 最终检索的query，可能是点击sug触发 
# "inputQuery": "山东师范",# ⽤户⼿动输⼊的query // 筛选条件 
#"channelId": 1, # 频道id，int，可选，没有是不过滤 
#"nodeId": 1, # 知识树节点id，int，可选，没有是不过滤，当前只⽀持叶⼦节点过滤 
"timeFilter": {}, # 可选，时间范围过滤，为空表示不进⾏时间 过滤，begin和end为unix时间戳。 
"tagFilter": ['教育经历'], # 可选，标签过滤，为空表示不进⾏标签过滤 
#"docTypeFilter": "word", # 可选，⽂档类型筛选项，没有不进⾏⽂档类型过滤，当前枚举值： word|excel|pdf
"sort": "relevance",# 排序⽅式默认相关性，其他排序时间，趋势热度
"pn": 1, # ⻚码，从1开始编码
"ps": 20, # pagesize，每⻚⻚⾯条数，默认不填20
"resultFields": [],
"searchType":"advanced",
"logId": "8966036931719178386",# string类型，除了翻⻚，其他检索不要加logId
"needCorrect": True, # 是否需要纠错，不需要则按照输⼊的query进⾏检索，需要则根据da返回的 纠错query以及置信度进⾏检索（显示看接⼝响应的escoreType&rewriteQuery）
"saveHistory": True, # 是否保存历史记录
"classesFilter":[] # 类别过滤（新增）
}
base_url = 'http://180.76.188.39:8085/mpks/api/search'
r = requests.post(base_url, headers=headers, json=json_obj)
rst = json.loads(r.text)
pprint(rst)



# url = 'http://180.76.188.39:8085/mpks/api/extra/tree/getAllParent'
# params = {
# 	'nodeId':1
# }
# response = requests.get(url=url, headers=headers, params=params)
# pprint(json.loads(response.text))



url = 'http://180.76.188.39:8085/mpks/api/extra/gremlin'
json_obj = {
	# 通过id查询实体
	# "gremlin": "g.key('b67029eb1341457162598cf84198d7e2').with('*').graph"
	# 2 查询实体并包括关联实体的数据
	# "gremlin" : "g.key('b67029eb1341457162598cf84198d7e2').out.with('*').graph" 
	# 4 图谱边关系显示范围控制
	# "gremlin": "g.key('b67029eb1341457162598cf84198d7e2').out.out.out.with('*').graph" 
	# 查询name为测试数据的实体以及出边
	# "gremlin" : "g.has('name.@value', MATCH, '测试数据').out.with('*').graph"
	"gremlin" : "g.has('type', MATCH, '人才特征demo').out.with('*').graph"
}
r = requests.post(url, headers=headers, json=json_obj)
pprint(json.loads(r.text))


# #  获取全部知识树信息
# url = 'http://180.76.188.39:8085/mpks/api/extra/getNodes'
# response = requests.get(url=url, headers=headers)
# pprint(json.loads(response.text))

# # 通过知识树节点获取所有相关⽂档
# url = 'http://180.76.188.39:8085/mpks/api/extra/source/data/list'
# json_obj = {
# 	"contentType": "struct",
# 	"inputTypes": "api_push",
# }


# 根据类⽬查询实体
# url = 'http://180.76.188.39:8085/api/datamanage/graph/view'
# json_obj = {
# 	"type": "学校demo",
# 	"wd": "山东师范",
# 	"fuzzy": 1,
# 	"mode": 0,
# 	"pageNumbe": 1,
# 	"pageSize": 10,
# }
# r = requests.post(url, headers=headers, json=json_obj)
# pprint(json.loads(r.text))