class PythonShellInterface:
def __init__(self):
self._initializeDefaults()
def _initializeDefaults(self):
self.host = DEFAULT_HOST
self.port = DEFAULT_PORT
self.databaseName = DEFAULT_DATABASE_NAME
def setHost(self, host):
self.host = host
def setPort(self, port):
self.port = port
def setDatabaseName(self, databaseName):
self.databaseName = databaseName
def connectToDatabase(self):
self._createShellManagerAndConnection()
self.shell_connection = self._getOrCreateFreeShell()
#print("Connected")
def _getOrCreateFreeShell(self):
while True:
try:
shell = self._getExistingFreeShell()
if not shell:
shell = self._createNewShell()
return shell
except ConnectionRefusedError:
time.sleep(0.1)
def _getExistingFreeShell(self):
self._retrieveShellsFromServer()
if len(self.freeShellsForDatabase) == 0:
return None
return self._connectToShellWithPort(self.freeShellsForDatabase[0][0])
def _retrieveShellsFromServer(self):
self.shells = list(self.shell_manager.list())
prefix = self._getPythonShellPrefix()
self.shellsForDatabase = [(port, name, occupied) for (port, dbName, name, occupied) in self.shells
if name.startswith(prefix)]
self.freeShellsForDatabase = [(port, name)
for (port, name, occupied) in self.shellsForDatabase
if occupied == 'free']
def _connectToShellWithPort(self, port):
connection = OctopusShellConnection(self.host, port)
connection.connect()
return connection
def _createNewShell(self):
shellname = self._generateNameForNewShell()
#Creates a new shell/thread
port = self.shell_manager.create(self.databaseName, shellname)
return self._connectToShellWithPort(port)
def _generateNameForNewShell(self):
prefix = self._getPythonShellPrefix()
number = self._getNumberForNewShell()
return prefix + '_' + str(number)
def _getPythonShellPrefix(self):
return ".python#" + self.databaseName
def _getNumberForNewShell(self):
shellnumbers = [int(name.split('#')[2]) for (port, name, occupied) in self.shellsForDatabase]
if len(shellnumbers) == 0: return 0
shellnumbers.sort()
highestShellNumber = shellnumbers[-1]
return highestShellNumber + 1
def _createShellManagerAndConnection(self):
self.shell_manager = ShellManager(self.host, self.port)
self.shell_connection = OctopusShellConnection(self.host, self.port)
# Close connection
def close(self):
OctopusShellConnection(self.host, self.port).close()
def runGremlinQuery(self, query):
#print("Run query "+query)
while True:
try:
result = self.shell_connection.run_command(query)
return result
except ConnectionResetError:
self.shell_connection = self._getOrCreateFreeShell()
time.sleep(0.1)
"""
Create chunks from a list of ids.
This method is useful when you want to execute many independent
traversals on a large set of start nodes. In that case, you
can retrieve the set of start node ids first, then use 'chunks'
to obtain disjoint subsets that can be passed to idListToNodes.
"""
def chunks(self, idList, chunkSize):
for i in range(0, len(idList), chunkSize):
yield idList[i:i+chunkSize]
def _createShellManagerAndConnection(self):
self.shell_manager = ShellManager(self.host, self.port)
self.shell_connection = OctopusShellConnection(self.host, self.port)
def _connectToShellWithPort(self, port):
connection = OctopusShellConnection(self.host, port)
connection.connect()
return connection
def close(self):
OctopusShellConnection(self.host, self.port).close()
class PythonShellInterface:
def __init__(self):
self._initializeDefaults()
def _initializeDefaults(self):
self.host = DEFAULT_HOST
self.port = DEFAULT_PORT
self.databaseName = DEFAULT_DATABASE_NAME
def setHost(self, host):
self.host = host
def setPort(self, port):
self.port = port
def setDatabaseName(self, databaseName):
self.databaseName = databaseName
def connectToDatabase(self):
self._createShellManagerAndConnection()
self.shell_connection = self._getOrCreateFreeShell()
def _getOrCreateFreeShell(self):
while True:
try:
shell = self._getExistingFreeShell()
if not shell:
shell = self._createNewShell()
return shell
except ConnectionRefusedError:
time.sleep(0.1)
def _getExistingFreeShell(self):
self._retrieveShellsFromServer()
if len(self.freeShellsForDatabase) == 0:
return None
return self._connectToShellWithPort(self.freeShellsForDatabase[0][0])
def _retrieveShellsFromServer(self):
self.shells = list(self.shell_manager.list())
prefix = self._getPythonShellPrefix()
self.shellsForDatabase = [(port, name, occupied) for (port, dbName, name, occupied) in self.shells
if name.startswith(prefix)]
self.freeShellsForDatabase = [(port, name)
for (port, name, occupied) in self.shellsForDatabase
if occupied == 'free']
def _connectToShellWithPort(self, port):
connection = OctopusShellConnection(self.host, port)
connection.connect()
return connection
def _createNewShell(self):
shellname = self._generateNameForNewShell()
port = self.shell_manager.create(self.databaseName, shellname)
return self._connectToShellWithPort(port)
def _generateNameForNewShell(self):
prefix = self._getPythonShellPrefix()
number = self._getNumberForNewShell()
return prefix + '_' + str(number)
def _getPythonShellPrefix(self):
return ".python_" + self.databaseName
def _getNumberForNewShell(self):
shellnumbers = [int(name.split('_')[2]) for (port, name, occupied) in self.shellsForDatabase]
if len(shellnumbers) == 0: return 0
shellnumbers.sort()
highestShellNumber = shellnumbers[-1]
return highestShellNumber + 1
def _createShellManagerAndConnection(self):
self.shell_manager = OrientDBShellManager(self.host, self.port)
self.shell_connection = OctopusShellConnection(self.host, self.port)
def runGremlinQuery(self, query):
while True:
try:
return self.shell_connection.run_command(query)
except ConnectionResetError:
self.shell_connection = self._getOrCreateFreeShell()
time.sleep(0.1)
"""
Create chunks from a list of ids.
This method is useful when you want to execute many independent
traversals on a large set of start nodes. In that case, you
can retrieve the set of start node ids first, then use 'chunks'
to obtain disjoint subsets that can be passed to idListToNodes.
"""
def chunks(self, idList, chunkSize):
for i in range(0, len(idList), chunkSize):
yield idList[i:i+chunkSize]
def _createShellManagerAndConnection(self):
self.shell_manager = OrientDBShellManager(self.host, self.port)
self.shell_connection = OctopusShellConnection(self.host, self.port)
def _connectToShellWithPort(self, port):
connection = OctopusShellConnection(self.host, port)
connection.connect()
return connection