#!/usr/bin/env python
# -*- coding:utf-8 -*-
import sys
sys.path.append("/home/core_adm/wei_chen")
from graphframes import GraphFrame
from pyspark import SparkContext, SparkConf
from pyspark.sql import SparkSession, HiveContext
from py2neo import *
graph = Graph("bolt://10.2.5.20:7687",user="neo4j",password="6bc405faee9d7206196cc67fdbb555f6")
def FindDepthNOde(spark,NodeSearch,Depth):
sql="MATCH (n)-[r]->(m) WHERE id(m)=index RETURN id(n) as src,id(m) as dst, type(r) as relation "
edges=graph.data(sql.replace("index",str(NodeSearch)))
return edges
def Graph_ini(spark,NodeSearch,Depth):
sql="MATCH (n)-[r]->(m) WHERE id(m)=index RETURN id(n) as src,id(m) as dst, type(r) as relation "
edges=graph.data(sql.replace("index",str(NodeSearch)))
print