#-*- coding: UTF-8 -*-
#连接数据库，生成查询语句，返回结果

import re
from py2neo import Graph

class Query():
    def __init__(self):
        #这里暂时使用的是我的服务器数据库，你也可以搭建自己的本地数据库
        self.graph=Graph("http://104.224.145.185:7474", username="neo4j",password="123456")
    # 运行cql语句
    def run(self,cql):
        find_rela = self.graph.run(cql)
        return find_rela


class QuestionTemplate():
    def __init__(self):
        self.q_template_dict={
            0:self.get_movie_rating,
            1:self.get_movie_releasedate,
            2:self.get_movie_type,
            3:self.get_movie_introduction,
            4:self.get_movie_actor_list,
            5:self.get_actor_info,
            6:self.get_actor_act_type_movie,
            7:self.get_actor_act_movie_list,
            8:self.get_movie_rating_bigger,
            9:self.get_movie_rating_smaller,
            10:self.get_actor_movie_type,
            11:self.get_cooperation_movie_list,
            12:self.get_actor_movie_num,
            13:self.get_actor_birthday
        }

        # 连接数据库
        self.graph = Query()

    def get_question_answer(self,question,template):
        # 如果问题模板的格式不正确则结束
        assert len(str(template).strip().split("\t"))==2
        template_id,template_str=int(str(template).strip().split("\t")[0]),str(template).strip().split("\t")[1]
        self.template_id=template_id
        self.template_str2list=str(template_str).split()

        # 预处理问题
        question_word,question_flag=[],[]
        for one in question:
            word, flag = one.split("/")
            question_word.append(str(word).strip())
            question_flag.append(str(flag).strip())
        assert len(question_flag)==len(question_word)
        self.question_word=question_word
        self.question_flag=question_flag
        self.raw_question=question
        # 根据问题模板来做对应的处理，获取答案
        answer=self.q_template_dict[template_id]()
        return answer

    # 获取电影名字
    def get_movie_name(self):
        ## 获取nm在原问题中的下标
        tag_index = self.question_flag.index("nm")
        ## 获取电影名称
        movie_name = self.question_word[tag_index]
        return movie_name
    def get_name(self,type_str):
        name_count=self.question_flag.count(type_str)
        if name_count==1:
            ## 获取nm在原问题中的下标
            tag_index = self.question_flag.index(type_str)
            ## 获取电影名称
            name = self.question_word[tag_index]
            return name
        else:
            result_list=[]
            for i,flag in enumerate(self.question_flag):
                if flag==str(type_str):
                    result_list.append(self.question_word[i])
            return result_list

    def get_num_x(self):
        x = re.sub(r'\D', "", "".join(self.question_word))
        return x
    # 0:nm 评分
    def get_movie_rating(self):
        # 获取电影名称，这个是在原问题中抽取的
        movie_name=self.get_movie_name()
        cql = f"match (m:Movie)-[]->() where m.title='{movie_name}' return m.rating"
        print(cql)
        answer = self.graph.run(cql)
        answer = round(list(answer)[0][0],2)
        final_answer=movie_name+"电影评分为"+str(answer)+"分！"
        return final_answer
    # 1:nm 上映时间
    def get_movie_releasedate(self):
        movie_name = self.get_movie_name()
        cql = f"match(m:Movie)-[]->() where m.title='{movie_name}' return m.releasedate"
        print(cql)
        answer = self.graph.run(cql)
        answer = list(answer)[0][0]
        final_answer = movie_name + "的上映时间是" + str(answer) + "！"
        return final_answer
    # 2:nm 类型
    def get_movie_type(self):
        movie_name = self.get_movie_name()
        cql = f"match(m:Movie)-[r:is]->(b) where m.title='{movie_name}' return b.name"
        print(cql)
        answer = self.graph.run(cql)
        answer_list=list(answer)
        answers = []
        for ans in answer_list:
            answers.append(str(ans[0]))
        answers="、".join(answers)
        final_answer = movie_name + "是" + answers + "等类型的电影！"
        return final_answer
    # 3:nm 简介
    def get_movie_introduction(self):
        movie_name = self.get_movie_name()
        cql = f"match(m:Movie)-[]->() where m.title='{movie_name}' return m.introduction"
        print(cql)
        answer = self.graph.run(cql)
        final_answer = movie_name + "主要讲述了" + str(list(answer)[0][0]) + "！"
        return final_answer
    # 4:nm 演员列表
    def get_movie_actor_list(self):
        movie_name=self.get_movie_name()
        cql = f"match(n:Person)-[r:actedin]->(m:Movie) where m.title='{movie_name}' return n.name"
        print(cql)
        answer = self.graph.run(cql)
        answer_list = list(answer)
        answers = []
        for ans in answer_list:
            answers.append(str(ans[0]))
        answers = "、".join(answers)
        final_answer = movie_name + "由" + answers + "等演员主演！"
        return final_answer
    # 5:nnt 介绍
    def get_actor_info(self):
        actor_name = self.get_name('nr')
        cql = f"match(n:Person)-[]->() where n.name='{actor_name}' return n.biography"
        print(cql)
        answer = self.graph.run(cql)
        final_answer = list(answer)[0][0]
        return final_answer
    # 6:nnt ng 电影作品
    def get_actor_act_type_movie(self):
        actor_name = self.get_name("nr")
        type=self.get_name("ng")
        # 查询电影名称
        cql = f"match(n:Person)-[]->(m:Movie) where n.name='{actor_name}' return m.title"
        print(cql)
        movie_name_list = list(self.graph.run(cql))
        #print(movie_name_list)
        # 查询类型
        result = []
        for movie_name in movie_name_list:
            movie_name = movie_name[0].strip()
            try:
                cql = f"match(m:Movie)-[r:is]->(t) where m.title='{movie_name}' return t.name"
                #print(cql)
                temp_type = []
                temp = list(self.graph.run(cql))
                for t in temp:
                    temp_type.append(t[0])
                #print(temp_type)
                if len(temp_type) == 0:
                    continue
                if type in temp_type:
                    result.append(movie_name)
            except:
                continue
        #print(result)
        answers = "、".join(result)
        final_answer = actor_name + "演过的" + type + "电影有:\n" + answers + "。"
        return final_answer


    # 7:nnt 电影作品
    def get_actor_act_movie_list(self):
        actor_name = self.get_name("nr")
        answers=self.get_actorname_movie_list(actor_name)
        answer_list = "、".join(answers)
        final_answer = actor_name + "演过" + answer_list + "等电影！"
        return final_answer
    def get_actorname_movie_list(self,actorname):
        # 查询电影名称
        cql = f"match(n:Person)-[]->(m:Movie) where n.name='{actorname}' return m.title"
        print(cql)
        answer = self.graph.run(cql)
        answer_list = list(answer)
        answers = []
        for ans in answer_list:
            answers.append(str(ans[0]))
        return answers

    # 8:nnt 参演评分 大于 x
    def get_movie_rating_bigger(self):
        actor_name=self.get_name('nr')
        x=self.get_num_x()
        cql = f"match(n:Person)-[r:actedin]->(m:Movie) where n.name='{actor_name}' and m.rating>={x} return m.title"
        print(cql)
        answer = self.graph.run(cql)
        answer_list = list(answer)
        answers = []
        for ans in answer_list:
            answers.append(str(ans[0]))
        answer_list = "、".join(answers)
        final_answer=actor_name+"演的电影评分大于"+x+"分的有"+answer_list+"等！"
        return final_answer

    # 9:nnt 参演评分 小于 x
    def get_movie_rating_smaller(self):
        actor_name = self.get_name('nr')
        x = self.get_num_x()
        cql = f"match(n:Person)-[r:actedin]->(m:Movie) where n.name='{actor_name}' and m.rating<{x} return m.title"
        print(cql)
        answer = self.graph.run(cql)
        answer_list = list(answer)
        answers = []
        for ans in answer_list:
            answers.append(str(ans[0]))
        answer_list = "、".join(answers)
        final_answer = actor_name + "演的电影评分小于" + x + "分的有" + answer_list + "等！"
        return final_answer

    # 10:nnt 出演过哪些类型的电影
    def get_actor_movie_type(self):
        actor_name = self.get_name("nr")
        # 查询电影名称
        cql = f"match(n:Person)-[]->(m:Movie) where n.name='{actor_name}' return m.title"
        print(cql)
        movie_name_list = list(self.graph.run(cql))
        # 查询类型
        #print(movie_name_list)
        result = set('')
        for movie_name in movie_name_list:
            movie_name = movie_name[0].strip()
            try:
                cql = f"match(m:Movie)-[r:is]->(t) where m.title='{movie_name}' return t.name"
                # print(cql)
                temp_type = []
                temp = list(self.graph.run(cql))
                for t in temp:
                    result.add(t[0])
                if len(temp_type) == 0:
                    continue
                #result.add(temp_type)
            except:
                continue
        answers = "、".join(result)
        final_answer = actor_name + "演过的电影有" + answers + "等类型。"
        return final_answer


    # 11: 演员A和演员B合作了哪些电影
    def get_cooperation_movie_list(self):
        # 获取演员名字
        actor_name_list=self.get_name('nr')
        movie_list={}
        for i,actor_name in enumerate(actor_name_list):
            answer_list=self.get_actorname_movie_list(actor_name)
            movie_list[i]=answer_list
        result_list=list(set(movie_list[0]).intersection(set(movie_list[1])))
        #print(result_list)
        answer="、".join(result_list)
        final_answer=actor_name_list[0]+"和"+actor_name_list[1]+"一起演过的电影主要有"+answer+"!"
        return final_answer

    # 12: nnt 一共演过多少部电影
    def get_actor_movie_num(self):
        actor_name=self.get_name("nr")
        answer_list=self.get_actorname_movie_list(actor_name)
        movie_num=len(set(answer_list))
        answer=movie_num
        final_answer=actor_name+"演过"+str(answer)+"部电影!"
        return final_answer

    # 13: nnt 出生日期
    def get_actor_birthday(self):
        actor_name = self.get_name('nr')
        cql = f"match(n:Person)-[]->() where n.name='{actor_name}' return n.birth"
        print(cql)
        answer = self.graph.run(cql)
        #print(list(answer)[0][0])
        final_answer = actor_name+"的生日是"+list(answer)[0][0]+"。"
        return final_answer