cj2001

class Neo4jConnection:
    
    def __init__(self, uri, user, pwd):
        self.__uri = uri
        self.__user = user
        self.__pwd = pwd
        self.__driver = None
        try:
            self.__driver = GraphDatabase.driver(self.__uri, auth=(self.__user, self.__pwd))
        except Exception as e:

cj2001

def get_author_list(line):
    # Cleans author dataframe column, creating a list of authors in the row.
    return [e[1] + ' ' + e[0] for e in line]


def get_category_list(line):
    # Cleans category dataframe column, creating a list of categories in the row.
    return list(line.split(" "))

cj2001

file = "./arxiv-metadata-oai-snapshot.json"

metadata  = []

lines = 100000    # 100k for testing

with open(file, 'r') as f:
    
    for line in tqdm(f):
        metadata.append(json.loads(line))

cj2001

WITH "file:///got-s1-nodes.csv" AS uri
LOAD CSV WITH HEADERS FROM uri AS row
MERGE (c:Character {id:row.Id})
SET c.name = row.Label

cj2001

docker run -p 7474:7474 -p 7687:7687 \
  --volume=$HOME/graph_data/data:/data \
  --volume=$HOME/graph_data/gameofthrones/data:/var/lib/neo4j/import \
  --volume=$HOME/graph_data/plugins:/var/lib/neo4j/plugins \
  --volume=$home/neo4j/logs:/var/lib/neo4j/logs \
  --env NEO4J_dbms_security_procedures_unrestricted=gds.\\\*,apoc.\\\* \
  --env apoc.import.file.enabled=true \
  --env NEO4J_AUTH=neo4j/1234 \
  neo4j:latest

cj2001

LOAD CSV WITH HEADERS FROM "file:///heros_edge_list.csv" AS row
MATCH (h1:hero {hero: row.hero})
MATCH (c1:comic {comic: row.comic})
CALL apoc.create.relationship(h1, c1) YIELD rel
REMOVE rel.noOp;

cj2001

import json
import argparse
import yaml
from typing import Any, Dict

import networkx as nx
import pandas as pd
from tqdm import tqdm
import pygraphviz
import graphviz