Skip to content
Snippets Groups Projects
Select Git revision
2 results Searching

postBuild

Blame
    • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    print_graph_test.py 3.27 KiB
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68
    69
    70
    71
    72
    73
    74
    75
    76
    77
    78
    79
    80
    81
    82
    83
    84
    85
    86
    87
    88
    89
    90
    91
    92
    93
    94
    95
    96
    97
    98
    99
    100
    101
    102
    103
    104
    105
    106
    107
    108 
    

    

    

    # -*- coding: utf-8 -*-
"""
Functions to test and print the nodes and edges sets

"""

__authors__ = "Donna Löding, Alina Molkentin, Xinyi Tang, Judith Große, Malte Schokolowski"
__email__ = "cis-project2021@zbh.uni-hamburg.de"
__status__ = "Production"
#__copyright__ = ""
#__credits__ = ["", "", "", ""]
#__license__ = ""
#__version__ = ""
#__maintainer__ = ""


import sys

#sys.path.insert(1, 'C:\Users\Malte\Git\CiS-Projekt\ci-s-projekt-verarbeitung\input')
sys.path.append("../../")
from verarbeitung.construct_new_graph.initialize_graph import init_graph_construction
from verarbeitung.update_graph.import_from_json import input_from_json
from verarbeitung.update_graph.update_graph import update_graph

# a function to print nodes and edges from a graph
def print_graph(nodes, edges):
    print("Knoten:\n")
    for node in nodes:
        print(node.title, "\n")
    print("\nKanten:\n")
    for edge in edges:
        print(edge,"\n")
    print(len(nodes))
    print(len(edges))
    print(" ")

def print_extended_graph(nodes, edges):
    print("Knoten:\n")
    for node in nodes:
        print(node.title, "\n")
        print(node.doi_url)
        for reference in node.references:
            print(reference.doi_url)
        for citation in node.citations:
            print(citation.doi_url)
    print("\nKanten:\n")
    for edge in edges:
        print(edge,"\n")
    print(len(nodes))
    print(len(edges))
    print(" ")

def print_simple(nodes, edges):
    # for node in nodes:
    #     print(node)
    # for edge in edges:
    #     print(edge)
    print(len(nodes))
    print(len(edges))
    print(" ")

# program test with some random dois
def try_known_publications():
    doi_list = []
    doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.9b00249')
    #doi_list.append('https://doi.org/10.1021/acs.jcim.9b00249')
    doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.1c00203')
    #arr.append('https://pubs.acs.org/doi/10.1021/acs.jcim.9b00249')
    doi_list.append('https://doi.org/10.1021/acs.jmedchem.0c01332')
    #arr.append('https://doi.org/10.1021/acs.jcim.0c00741')

    #arr.append('https://doi.org/10.1021/ci700007b')
    #doi_list.append('https://doi.org/10.1021/acs.jcim.5b00292')
    
    #doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.0c00675')
    #url = sys.argv[1]
    #arr.append[url]


    nodes, edges = init_graph_construction(doi_list,2,2)

    print_graph(nodes, edges) 

    return(nodes, edges)

def try_delete_nodes():
    doi_list = []
    doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.9b00249')
    #doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.1c00203')
    nodes, edges = init_graph_construction(doi_list,1,1)
    #print_simple(nodes, edges)

    # list_of_nodes_py, list_of_edges_py = input_from_json('json_text.json')
    # doi_list = []
    # doi_list.append('https://pubs.acs.org/doi/10.1021/acs.jcim.9b00249')
    # valid_nodes, valid_edges = update_graph(doi_list, list_of_nodes_py, list_of_edges_py)
    # print_simple(valid_nodes, valid_edges)

def try_import():
    nodes, edges = input_from_json('json_text.json')
    print_extended_graph(nodes,edges)

#nodes, edges = try_known_publications()
#nodes_new, edges_new = input_from_json("json_text.json")
#print_graph(nodes_new, edges_new)
try_delete_nodes()

#try_import()