Skip to content

OrientDB driver for Python that uses the binary protocol.

License

Notifications You must be signed in to change notification settings

brucetony/pyorientdb

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pyorientdb

State
Build Status Coverage Status

Orientdb driver for python that uses the binary protocol.

Disclaimer

This fork of pyorient by Ostico is an updated version of the fork maintained by orienttechnologies. Efforts were made to have the pyorient PyPI package transferred to orienttechnologies, but after a year there has been no progress (see the GitHub issue). Additionally, orienttechnologies does not appear to be actively maintaining this driver, and I have made some small modifications to get it working with OrientDB 3.0+, but I do not plan to actively update this fork. Rather, I wanted to have a working PyPI package that those using Python with OrientDB can easily download and install, so I am creating a new package called pyorientdb using my updated fork. Any and all contributions and support are welcome.

Installation

pip install pyorientdb

Documentation

OrientDB PyOrient Python Driver

How to run tests

  • Ensure you have ant and nose installed properly
  • Bootstrap orient by running ./ci/ci-start.sh from project directory
    • It will download latest orient and make some change on config and database for the tests*
  • Run with nosetests

Using this library with OrientDB 3.1+

As of OrientDB 3.1+, session tokens are now required for interacting with databases. You can find a brief description of how to use session tokens below for older version, but now they are enabled by default when a client is initialized:

client = pyorientdb.OrientDB("localhost", 2424)
client.set_session_token(True)
client.db_open("GratefulDeadConcerts", "admin", "admin")
client.command("create class my_class if not exists extends V")
client.command(f"insert into my_class (row_id, work, holiday) values (1, 'banker', 'christmas')")
client.query('select from V limit 1')

Note that one can connect to a database and run commands and queries within that database without a session ID. Some methods will require creating a session ID in order to perform (e.g. checking the existence of a database or creating a new one):

client = pyorientdb.OrientDB("localhost", 2424)
client.set_session_token(True)
client.db_exists("GratefulDeadConcerts")
# Results in an error: pyorient.exceptions.PyOrientSecurityException: 
# com.orientechnologies.orient.enterprise.channel.binary.OTokenSecurityException - missing session and token

To create new databases, or perform other restricted actions, you must connect to the client with approved user credentials:

client = pyorientdb.OrientDB("localhost", 2424)
client.set_session_token(True)
client.connect("root", "rootPassword")
client.db_exists("GratefulDeadConcerts")
# True

Init the Client

client = pyorientdb.OrientDB("localhost", 2424)
session_id = client.connect( "admin", "admin" )

Create a DB

client.db_create(db_name, pyorientdb.DB_TYPE_GRAPH, pyorientdb.STORAGE_TYPE_MEMORY)

Check if a DB exists

client.db_exists(db_name, pyorientdb.STORAGE_TYPE_MEMORY)

Open a DB

client.db_open(db_name, "admin", "admin")

Close a DB and destroy the connection ( by OrientDB design )

client.db_close()

Get the the list of databases ( needs to be connected )

client.db_list()

Get the size of a database ( needs a DB opened )

client.db_size()

Get the number of records in a database in the OrientDB Server instance

client.db_count_records()

Send a command

cluster_id = client.command( "create class my_class extends V" )
client.command(
    "insert into my_class ( 'accommodation', 'work', 'holiday' ) values( 'B&B', 'garage', 'mountain' )"
)

Create a record

Warning Some issues are experimented with record_create/record_upload and OrientDB < 2.0. These command are strongly discouraged with these versions

rec = { '@my_class': { 'accommodation': 'house', 'work': 'office', 'holiday': 'sea' } }
rec_position = client.record_create( cluster_id, rec )

Update a record

Warning Some issues are experimented with record_create/record_upload and OrientDB < 2.0. These command are strongly discouraged with these versions

rec3 = { '@my_class': { 'accommodation': 'hotel', 'work': 'home', 'holiday': 'hills' } }
update_success = client.record_update( rec_position._rid, rec_position._rid, rec3, rec_position._version )

Load a record

client.record_load( rec_position._rid )

Load a record with cache

def _my_callback(for_every_record):
    print(for_every_record)

client.record_load( rec_position._rid, "*:-1", _my_callback )

Make a query

result = client.query("select from my_class", 10, '*:0')

Make an Async query

def _my_callback(for_every_record):
    print(for_every_record)

result = client.query_async("select from my_class", 10, '*:0', _my_callback)

Delete a record

client.record_delete( cluster_id, rec_position._rid )

Drop a DB

client.db_drop( db_name )

Create a new cluster

new_cluster_id = client.data_cluster_add(
    'my_cluster_1234567', pyorient.CLUSTER_TYPE_PHYSICAL
)

Reload DB ( refresh clusters info )

client.db_reload()

Get the range of record ids for a cluster

client.data_cluster_data_range( new_cluster_id )

Get the number of records in one or more clusters

client.data_cluster_count( [ 1, 2, 3, 4, 11 ] )

Drop a data cluster

client.data_cluster_drop( new_cluster_id )

Shut down the server. Requires "shutdown" permission to be set in orientdb-server-config.xml file

client.shutdown( "root", "a_super_secret_password" )

Transactions

### use a cluster
cluster_id = 3

### execute real create to get some info
rec = { 'accommodation': 'mountain hut', 'work': 'not!', 'holiday': 'lake' }
rec_position = client.record_create( cluster_id, rec )

tx = client.tx_commit()
tx.begin()

### create a new record
rec1 = { 'accommodation': 'home', 'work': 'some work', 'holiday': 'surf' }
rec_position1 = client.record_create( -1, rec1 )

### prepare for an update
rec2 = { 'accommodation': 'hotel', 'work': 'office', 'holiday': 'mountain' }
update_record = client.record_update( cluster_id, rec_position._rid, rec2, rec_position._version )

tx.attach( rec_position1 )
tx.attach( rec_position1 )
tx.attach( update_record )
res = tx.commit()

assert res["#3:1"].holiday == 'mountain'
assert res["#3:2"].holiday == 'surf'
assert res["#3:3"].holiday == 'surf'

Execute OrientDB SQL Batch

cmd = ("begin;"
    "let a = create vertex set script = true;"
    "let b = select from v limit 1;"
    "let e = create edge from $a to $b;"
    "commit retry 100;")

    edge_result = self.client.batch(cmd)

Persistent Connections - Session Token

Since version 27 is introduced an extension to allow use a token based session. This functionality must be enabled on the server config.

  • In the first negotiation the client can ask for a token based authentication using the client.set_session_token method.
  • The server will reply with a token or with an empty string meaning that it not support token based session and is using an old style session.
  • For each request, the client will send the token and eventually it will get a new one if token lifetime ends.

When using the token based authentication, the connections can be shared between users of the same server.

client = pyorient.OrientDB("localhost", 2424)
client.set_session_token( True )  # set true to enable the token based
authentication
client.db_open( "GratefulDeadConcerts", "admin", "admin" )

### store this token somewhere
sessionToken = client.get_session_token()

### destroy the old client, equals to another user/socket/ip ecc.
del client

### create a new client
client = pyorient.OrientDB("localhost", 2424)

### set the previous obtained token to re-attach to the old session
client.set_session_token( sessionToken )

### now the dbOpen is not needed to perform database operations
record = client.query( 'select from V where @rid = #9:1' )

### set the flag again to true if you want to renew the token
client.set_session_token( True )  # set true
client.db_open( "GratefulDeadConcerts", "admin", "admin" )
new_sessionToken = client.get_session_token()

assert sessionToken != new_sessionToken

A GRAPH Example

The GRAPH representation of animals and its food

import pyorientdb

client = pyorientdb.OrientDB("localhost", 2424)  # host, port

### open a connection (username and password)
client.connect("admin", "admin")

### create a database
client.db_create("animals", pyorientdb.DB_TYPE_GRAPH, pyorientdb.STORAGE_TYPE_MEMORY)

### select to use that database
client.db_open("animals", "admin", "admin")

### Create the Vertex Animal
client.command("create class Animal extends V")

### Insert a new value
client.command("insert into Animal set name = 'rat', specie = 'rodent'")

### query the values
client.query("select * from Animal")
[ < OrientRecord
at
0x7f > ..., ...]

### Create the vertex and insert the food values

client.command('create class Food extends V')
client.command("insert into Food set name = 'pea', color = 'green'")

### Create the edge for the Eat action
client.command('create class Eat extends E')

### Lets the rat likes to eat pea
eat_edges = client.command(
  "create edge Eat from ("
  "select from Animal where name = 'rat'"
  ") to ("
  "select from Food where name = 'pea'"
  ")"
)

### Who eats the peas?
pea_eaters = client.command("select expand( in( Eat )) from Food where name = 'pea'")
for animal in pea_eaters:
  print(animal.name, animal.specie)
'rat rodent'
...

### What each animal eats?
animal_foods = client.command("select expand( out( Eat )) from Animal")
for food in animal_foods:
  animal = client.query(
    "select name from ( select expand( in('Eat') ) from Food where name = 'pea' )"
  )[0]
  print(food.name, food.color, animal.name)
'pea green rat'

Authors

About

OrientDB driver for Python that uses the binary protocol.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 97.4%
  • Shell 2.6%