mirror of
https://github.com/vivisect/vivisect
synced 2026-06-08 18:04:23 +00:00
invisigoth
44 lines
896 B
Python
44 lines
896 B
Python
'''
|
|
A package for a basic command interpreter for the graph.
|
|
(and graph db...)
|
|
'''
|
|
|
|
import sys
|
|
import cmd
|
|
|
|
import visgraph.dbcore as vg_dbcore
|
|
import visgraph.pathcore as vg_pcore
|
|
import visgraph.graphcore as vg_gcore
|
|
|
|
class GraphCli(cmd.Cmd):
|
|
|
|
def __init__(self, graph=None):
|
|
cmd.Cmd.__init__(self)
|
|
if graph == None:
|
|
graph = vg_gcore.Graph()
|
|
self.graph = graph
|
|
|
|
def do_addnode(self, line):
|
|
'''
|
|
Add a node with the given key=value properties on the CLI.
|
|
(the property nid is special and MUST be an integer)
|
|
'''
|
|
nid = self.graph.addNode()
|
|
print 'Node %d added!' % nid
|
|
|
|
def do_quit(self, line):
|
|
'''
|
|
Exit the visgraph cli....
|
|
'''
|
|
raise SystemExit()
|
|
|
|
#class GraphDBCli(
|
|
|
|
def main():
|
|
cli = GraphCli()
|
|
cli.cmdloop()
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|
|
|