import os
import json
from py2neo import Graph,Node

class MedicalGraph:
    def __init__(self):
        cur_dir = '/'.json(os.path.abspath(__file__).split('/')[:-1])
        self.data_path = os.path.join(cur_dir,'data/medical2.json')
        self.g = Graph("http//localhost:7474",username="neo4j",password="airline-edward-volume-album-bagel-2728")
        # 读取文件
            

if __name__ == '__main__':
    handler = MedicalGraph() # 连接数据库
    handler.create_graphnodes() # 创建节点
    handler.create_graphrels() # 连接边

def read_nodes(self):
       # 输入节点
       drugs = [] # 创建节点事例
       
       # 构建节点实体管理
       rels_department = []
       rels_noteat = [] # 疾病-忌吃食物关系
       rels_doeat = [] # 疾病-宜吃食物关系
       
       count = 0
       for data in open(self.data_path):  # 遍历data
           disease_dict = {}
           count += 1
           print(count)
           data_json = json.loads(data) # 将数据转化为json格式
           disease = data_json['name']
           disease_dict['name'] = disease # 根据病的名字构建一个新的字典
           disease.append(disease)
           disease_dict['desc'] = '' # 将这个病的信息都存在这个字典当中
           disease_dict['prevent'] = ''
           
       # 根据出现的字段进行判断，该字段是该业务的属性还是关联关系
       # 把病的名字与病的症状相关联
       if 'symptom' in data_json:
           symptom += data_json['symptom']
           for symptom in data_json['symptom']:
               rels_sympmtom.append([disease,symptom]) # 如：[感冒，头痛]
               
       #疾病与生病原因      
       if 'cause' in data_json:
           for symptom in data_json['cause']:
               rels_cause.append([disease,cause]) # 如：[感冒，着凉]
       
       # 科室之间的关系
       if 'cure_department' in data_json:
           cure_department = data_json['cure_department']
           if len(cure_department) == 1:
               rels_category.append([disease,cure_department[0]])
           if len(cure_department) == 2:
               big = cure_department[0]
               small = cure_department[1]
               rels_department.append([small,big])
               rels_category.append([disease,small])
       # 如何去治疗
       if 'cure_way' in data_json:
           disease_dict['cure_way'] = data_json['cure_way']
           
       return set(drugs),set(foods),rels_noteat,rels_category# 将创建的电和关系返回
   
   #####
   
   

# 最后·创建知识图谱实体节点类型schema
	def create_graphnodes(self):
        Drugs,Foods,Checks,Departments,Producers，rels_acompany,rels_category = self.read_nodes() # 要用的所有节点名称列举
        
        self.create_disease_nodes(disease_infos)
        
        self.create_node('Drug',Drugs)
        print(len(Drugs))
        
        self.create_node('Food',Foods)
        print(len(Foods))
        
        self.create_node('Check',Checks)
        print(len(Checks))
        

# 创建知识图谱中心疾病的节点
   def create_disease_nodes(self,disease_infos):
       count = 0
       for disease_dict in disease_infos:
           node = Node("Disease",name=disease_dict['name'],desc=disease_dict['desc'],
                      prevent=disease_dict['prevent'])
           self.g.create(node) # 等待连接到节点后，创建节点
           count +=1
           print(count)
       return 
           

# 建立节点
def create_node(self,label,nodes):
    count = 0
    for node_name in nodes:
        node = Node(label,name=node_name)
        self.g.create(node)
        count +=1
        print(count,len(nodes))
    return     

# 创建实体边关系
def create_graphrels(self):
    Durgs,Foods,Checks,Departments,Producers,Symptoms,Diseases,disease_infos,rels_check,rels_recommandeat
    self.create_relationship('Disease','Food',rels_recommandeat,'recommand_eat','推荐食谱')

# 创建关联实体边
def create_relationship(self,start_node,end_node,edges,rel_type,rel_name):
    count = 0
    # 去重处理
    set_edge = []
    for edge in edges:
        set_edges.append('###'.join(edge)) # ["dhcv"###"jsdb"]
    all = len(set(set_edge))
    for edge in set(set_edge):
        edge = edge.split('###')
        p = edge[0]
        q = edge[1]
        query = "match(p:%s),(q:%s)wherebp.name='%s'and q.name='%s'create (p)-[rel:%s{name:'%s'}]->(q)"%(start_node,end_node,p,q,rel_type,rel_name)
        try:
            self.g.run(query) # 执行语句
            count += 1
            print(rel_type,count,all)
        except Exception as e:
            print(e)
    return

from question_classifier import *
from question_parser import *
from answer_search import *