pygraph_redis

Simple python library to manipulate directed graphs in redis

License

pygraph_redis is released under the MIT Public License

Description

pygraph_redis is a simple library to manipulate directed graphs inside a redis database.

In this library, a graph is a bunch of nodes, each node knows its predecessors and its successors. A node can store some attributs (strings or sets of strings).

Installation

to install:

python setup.py install

How to use

First you need a redis database, it's up to you to install it.

The library itself is quite simple:

Cheat Sheet

#              initialization
#       arg1      |    arg2    |     arg3
#--------------------------------------------
# redis connexion | graph_name |    logger
#    redis obj    |  unicode   |  logger obj 

mygraph1 = Directed_graph(r_server, u'mygraph1', logger)

#                    create or add elements to a node
#    arg1   |     arg2     |     arg3     |             arg4
#---------------------------------------------------------------------------
# node name |  successors  | predecessors |           attributs
#  unicode  | unicode list | unicode list |      dictionnary of unicode 
#           |              |              | or set of unicode (key: unicode)

mygraph1.write_on_node(u'm1', 
                       [u's2'],
                       [u'p1'], 
                       {u'a3': set([u'69']), u'a2': u'42'}
)

#             delete elements from a node
#    arg1   |     arg2     |     arg3     |      arg4
#----------------------------------------------------------
# node name |  successors  | predecessors | attributs names
#  unicode  | unicode list | unicode list | list of unicode 

mygraph1.write_off_node(u'm1', [u's2'], [u'p1'], [u'attr3', u'attr2']

# delete a node
#     arg1 
#--------------
#  node name 
#   unicode   

mygraph1.remove_node(u'm1')

# get attributs list
#     arg1 
#--------------
#  node name 
#   unicode   

mygraph1.get_attributs_list(u'm1')

# get an attribut
#     arg1     |     arg2
#--------------|--------------
#  node name   | attribut name
#   unicode    |    unicode

mygraph1.get_attributs(u'm1', u'a2')

# get successors
#     arg1 
#--------------
#  node name 
#   unicode   

mygraph1.get_successors(u'm1')

# get predecessors
#     arg1 
#--------------
#  node name 
#   unicode   

mygraph1.get_predecessors(u'm1')

Initialization

Create an instance of "Directed_graph":

#importing directed_graph
from pygraph_redis.directed_graph import Directed_graph
import redis

#creating a basic logger
import logging
logging.basicConfig(format = u'%(message)s')
logger = logging.getLogger(u'redis')
logger.parent.setLevel(logging.DEBUG)

#creating the redis connexion
r_server = redis.Redis("localhost")

#creating the graph object
mygraph1 = Directed_graph(r_server, u'mygraph1', logger)

#creating the graph object with a different separator
mygraph2 = Directed_graph(r_server, u'mygraph2', logger, separator = u'mysep')

#creating the graph object with a "root" (improper name, I know)
mygraph2 = Directed_graph(r_server, u'mygraph2', logger, has_root = True)
#"has\_root = True" ensures that every node has a predecessor
#if enabled, a node has at least root as a predecessor, 
#but if it has any other predecessor it doesn't have root as predecessor

Node manipulation

Node creation:

#add node 'm1' to 'mygraph1' with:
#successors: 's1' and 's2'
#predecessors: 'p1' and 'p2'
#attributs:
#   * 'attr1': set([u'51',u'69'])
#   * 'attr2': '42'    

mygraph1.write_on_node(u'm1',
    [u's1', u's2'],
    [u'p1', u'p2'],
    {u'attr1': set([u'51', u'69']), u'attr2': u'42'}
)

Node edition:

#add new elements or edit existing elements of a node
#it's exactly the same function as before
mygraph1.write_on_node(u'm1', 
    [u's4'], 
    [], 
    {u'attr3': set([u'16', u'32', u'64']), u'attr2': u'5150'}
)

#remove some elements of a node (successors, predecessors, attributs)
mygraph1.write_off_node(u"m1", [u"s1"], [u"p2"],[u'attr2'])

#completely delete a node
mygraph1.remove_node(u'm1')

Node attributs manipulation

To manipulate the attributs of a node:

#create the node 'm2'
mygraph1.write_on_node(u'm2',
    [u's1', u's2'],
    [u'p1', u'p2'],
    {u'attr1': set([u'51', u'69']), u'attr2': u'42'}
)

#get the set of attribut names
set_of_attributs = mygraph1.get_attributs_list(u'm2')
print set_of_attributs

#get a specific attribut
attr2 = mygraph1.get_attribut(u'm2', u'attr2')
print attr2

Graph navigation

To navigate inside the graph, you have two functions:

#get the predecessors of 'm2'
predecessors = mygraph1.get_predecessors(u'm2')
print predecessors

#get the successors of 'm2'
successors = mygraph1.get_successors(u'm2')

if you have the has_root flag enable:

#get the "root" name
root = mygraph1.get_root_name()

print root

#get the successors of 'root'
successors = mygraph1.get_successors(root)
print successors

About the redis keys

Redis key format:

  <graph name><sep><node_name><sep><variable_name>[<sep><other>]*
  
  <graph name>: name of the graph
  <sep>: the key fields separator 
       (this string should not be in node_name or variable_name,
        otherwise, there is a redis key collision possibility)
  <node_name>: name of the node
  <variable_name>: name of the variable
  [<sep><other>]: optional extension

To avoid key collision, you must carefully choose the key separator, it must not be included in any node name or node attribut name (possible redis key collision).

About the logs

This library provides a lot of logs, most of them are debug, except possible redis key collision which is warning.