# -*- coding: utf-8 -*-
###########################################################################
# Copyright (c), The AiiDA team. All rights reserved. #
# This file is part of the AiiDA code. #
# #
# The code is hosted on GitHub at https://github.com/aiidateam/aiida_core #
# For further information on the license, see the LICENSE.txt file #
# For further information please visit http://www.aiida.net #
###########################################################################
from __future__ import absolute_import
import copy
from sqlalchemy import literal
from sqlalchemy.exc import SQLAlchemyError, ProgrammingError
from sqlalchemy.orm.exc import NoResultFound
from sqlalchemy.orm.attributes import flag_modified
from aiida.backends.utils import get_automatic_user
from aiida.backends.sqlalchemy.models.node import DbNode, DbLink
from aiida.backends.sqlalchemy.models.comment import DbComment
from aiida.backends.sqlalchemy.models.user import DbUser
from aiida.backends.sqlalchemy.models.computer import DbComputer
from aiida.common.utils import get_new_uuid
from aiida.common.folders import RepositoryFolder
from aiida.common.exceptions import (InternalError, ModificationNotAllowed,
NotExistent, UniquenessError)
from aiida.common.links import LinkType
from aiida.orm.implementation.general.node import AbstractNode, _NO_DEFAULT, _HASH_EXTRA_KEY
from aiida.orm.implementation.sqlalchemy.computer import Computer
from aiida.orm.implementation.sqlalchemy.group import Group
from aiida.orm.implementation.sqlalchemy.utils import django_filter, \
get_attr
from aiida.orm.implementation.general.utils import get_db_columns
from aiida.orm.mixins import Sealable
import aiida.backends.sqlalchemy
import aiida.orm.autogroup
[docs]class Node(AbstractNode):
[docs] def __init__(self, **kwargs):
super(Node, self).__init__()
self._temp_folder = None
dbnode = kwargs.pop('dbnode', None)
# Set the internal parameters
# Can be redefined in the subclasses
self._init_internal_params()
if dbnode is not None:
if not isinstance(dbnode, DbNode):
raise TypeError("dbnode is not a DbNode instance")
if dbnode.id is None:
raise ValueError("If cannot load an aiida.orm.Node instance "
"from an unsaved DbNode object.")
if kwargs:
raise ValueError("If you pass a dbnode, you cannot pass any "
"further parameter")
# If I am loading, I cannot modify it
self._to_be_stored = False
self._dbnode = dbnode
# If this is changed, fix also the importer
self._repo_folder = RepositoryFolder(section=self._section_name,
uuid=self.uuid)
else:
# TODO: allow to get the user from the parameters
user = get_automatic_user()
self._dbnode = DbNode(user=user,
uuid=get_new_uuid(),
type=self._plugin_type_string)
self._to_be_stored = True
# As creating the temp folder may require some time on slow
# filesystems, we defer its creation
self._temp_folder = None
# Used only before the first save
self._attrs_cache = {}
# If this is changed, fix also the importer
self._repo_folder = RepositoryFolder(section=self._section_name,
uuid=self.uuid)
# Automatically set all *other* attributes, if possible, otherwise
# stop
self._set_with_defaults(**kwargs)
[docs] @classmethod
def get_subclass_from_uuid(cls, uuid):
from aiida.orm.querybuilder import QueryBuilder
from sqlalchemy.exc import DatabaseError
try:
qb = QueryBuilder()
qb.append(cls, filters={'uuid': {'==': str(uuid)}})
if qb.count() == 0:
raise NotExistent("No entry with UUID={} found".format(uuid))
node = qb.first()[0]
if not isinstance(node, cls):
raise NotExistent("UUID={} is not an instance of {}".format(
uuid, cls.__name__))
return node
except DatabaseError as de:
raise ValueError(de.message)
[docs] @classmethod
def get_subclass_from_pk(cls, pk):
from aiida.orm.querybuilder import QueryBuilder
from sqlalchemy.exc import DatabaseError
# If it is not an int make a final attempt
# to convert to an integer. If you fail,
# raise an exception.
try:
pk = int(pk)
except:
raise ValueError("Incorrect type for int")
try:
qb = QueryBuilder()
qb.append(cls, filters={'id': {'==': pk}})
if qb.count() == 0:
raise NotExistent("No entry with pk= {} found".format(pk))
node = qb.first()[0]
if not isinstance(node, cls):
raise NotExistent("pk= {} is not an instance of {}".format(
pk, cls.__name__))
return node
except DatabaseError as de:
raise ValueError(de.message)
[docs] @classmethod
def query(cls, *args, **kwargs):
raise NotImplementedError("The node query method is not supported in "
"SQLAlchemy. Please use QueryBuilder.")
@property
def type(self):
# Type is immutable so no need to ensure the model is up to date
return self._dbnode.type
@property
def ctime(self):
"""
Return the creation time of the node.
"""
self._ensure_model_uptodate(attribute_names=['ctime'])
return self._dbnode.ctime
@property
def mtime(self):
"""
Return the modification time of the node.
"""
self._ensure_model_uptodate(attribute_names=['mtime'])
return self._dbnode.mtime
[docs] def get_user(self):
"""
Get the user.
:return: a Django DbUser model object
"""
self._ensure_model_uptodate(attribute_names=['user'])
return self._dbnode.user
[docs] def get_computer(self):
"""
Get the computer associated to the node.
:return: the Computer object or None.
"""
self._ensure_model_uptodate(attribute_names=['dbcomputer'])
if self._dbnode.dbcomputer is None:
return None
else:
return Computer(dbcomputer=self._dbnode.dbcomputer)
[docs] def _get_db_label_field(self):
"""
Get the label of the node.
:return: a string.
"""
self._ensure_model_uptodate(attribute_names=['label'])
return self._dbnode.label
[docs] def _update_db_label_field(self, field_value):
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
self._dbnode.label = field_value
if not self._to_be_stored:
session.add(self._dbnode)
self._increment_version_number_db()
[docs] def _get_db_description_field(self):
"""
Get the description of the node.
:return: a string
:rtype: str
"""
self._ensure_model_uptodate(attribute_names=['description'])
return self._dbnode.description
[docs] def _update_db_description_field(self, field_value):
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
self._dbnode.description = field_value
if not self._to_be_stored:
session.add(self._dbnode)
self._increment_version_number_db()
[docs] def _replace_dblink_from(self, src, label, link_type):
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
try:
self._add_dblink_from(src, label)
except UniquenessError:
# I have to replace the link; I do it within a transaction
try:
self._remove_dblink_from(label)
self._add_dblink_from(src, label, link_type)
session.commit()
except:
session.rollback()
raise
[docs] def _remove_dblink_from(self, label):
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
link = DbLink.query.filter_by(label=label).first()
if link is not None:
session.delete(link)
[docs] def _add_dblink_from(self, src, label=None, link_type=LinkType.UNSPECIFIED):
from aiida.backends.sqlalchemy import get_scoped_session
from aiida.orm.querybuilder import QueryBuilder
session = get_scoped_session()
if not isinstance(src, Node):
raise ValueError("src must be a Node instance")
if self.uuid == src.uuid:
raise ValueError("Cannot link to itself")
if self._to_be_stored:
raise ModificationNotAllowed(
"Cannot call the internal _add_dblink_from if the "
"destination node is not stored")
if src._to_be_stored:
raise ModificationNotAllowed(
"Cannot call the internal _add_dblink_from if the "
"source node is not stored")
# Check for cycles. This works if the transitive closure is enabled; if
# it isn't, this test will never fail, but then having a circular link
# is not meaningful but does not pose a huge threat
#
# I am linking src->self; a loop would be created if a DbPath exists
# already in the TC table from self to src
if link_type is LinkType.CREATE or link_type is LinkType.INPUT:
if QueryBuilder().append(
Node, filters={'id': self.pk}, tag='parent').append(
Node, filters={'id': src.pk}, tag='child', descendant_of='parent').count() > 0:
raise ValueError(
"The link you are attempting to create would generate a loop")
if label is None:
autolabel_idx = 1
existing_from_autolabels = session.query(DbLink.label).filter(
DbLink.output_id == self._dbnode.id,
DbLink.label.like("link%")
)
while "link_{}".format(autolabel_idx) in existing_from_autolabels:
autolabel_idx += 1
safety_counter = 0
while True:
safety_counter += 1
if safety_counter > 100:
# Well, if you have more than 100 concurrent addings
# to the same node, you are clearly doing something wrong...
raise InternalError("Hey! We found more than 100 concurrent"
" adds of links "
"to the same nodes! Are you really doing that??")
try:
self._do_create_link(src, "link_{}".format(autolabel_idx), link_type)
break
except UniquenessError:
# Retry loop until you find a new loop
autolabel_idx += 1
else:
self._do_create_link(src, label, link_type)
[docs] def _do_create_link(self, src, label, link_type):
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
try:
with session.begin_nested():
link = DbLink(input_id=src.id, output_id=self.id,
label=label, type=link_type.value)
session.add(link)
except SQLAlchemyError as e:
raise UniquenessError("There is already a link with the same "
"name (raw message was {})"
"".format(e))
[docs] def _get_db_output_links(self, link_type):
link_filter = {'input': self._dbnode}
if link_type is not None:
link_filter['type'] = link_type.value
return ((i.label, i.output.get_aiida_class()) for i in
DbLink.query.filter_by(**link_filter).distinct().all())
[docs] def _set_db_computer(self, computer):
self._dbnode.dbcomputer = DbComputer.get_dbcomputer(computer)
[docs] def _set_db_attr(self, key, value):
"""
Set the value directly in the DB, without checking if it is stored, or
using the cache.
DO NOT USE DIRECTLY.
:param str key: key name
:param value: its value
"""
try:
self._dbnode.set_attr(key, value)
self._increment_version_number_db()
except:
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
session.rollback()
raise
[docs] def _del_db_attr(self, key):
try:
self._dbnode.del_attr(key)
self._increment_version_number_db()
except:
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
session.rollback()
raise
[docs] def _get_db_attr(self, key):
try:
return get_attr(self._attributes(), key)
except (KeyError, IndexError):
raise AttributeError("Attribute '{}' does not exist".format(key))
[docs] def _db_iterattrs(self):
for k, v in self._attributes().iteritems():
yield (k, v)
[docs] def _db_attrs(self):
for k in self._attributes().iterkeys():
yield k
[docs] def _increment_version_number_db(self):
self._dbnode.nodeversion = self.nodeversion + 1
try:
self._dbnode.save()
except:
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
session.rollback()
raise
[docs] def copy(self, **kwargs):
# Make sure we have the latest version from the database
self._ensure_model_uptodate()
newobject = self.__class__()
newobject._dbnode.type = self._dbnode.type # Inherit type
newobject._dbnode.label = self._dbnode.label # Inherit label
# TODO: add to the description the fact that this was a copy?
newobject._dbnode.description = self._dbnode.description # Inherit description
newobject._dbnode.dbcomputer = self._dbnode.dbcomputer # Inherit computer
for k, v in self.iterattrs():
if k != Sealable.SEALED_KEY:
newobject._set_attr(k, v)
for path in self.get_folder_list():
newobject.add_path(self.get_abs_path(path), path)
return newobject
@property
def id(self):
return self._dbnode.id
@property
def dbnode(self):
self._ensure_model_uptodate()
return self._dbnode
@property
def nodeversion(self):
self._ensure_model_uptodate(attribute_names=['nodeversion'])
return self._dbnode.nodeversion
@property
def public(self):
self._ensure_model_uptodate(attribute_names=['public'])
return self._dbnode.public
[docs] def _db_store_all(self, with_transaction=True, use_cache=None):
"""
Store the node, together with all input links, if cached, and also the
linked nodes, if they were not stored yet.
:parameter with_transaction: if False, no transaction is used. This
is meant to be used ONLY if the outer calling function has already
a transaction open!
"""
self._store_input_nodes()
self.store(with_transaction=False, use_cache=use_cache)
self._store_cached_input_links(with_transaction=False)
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
if with_transaction:
try:
session.commit()
except SQLAlchemyError as e:
session.rollback()
raise
return self
[docs] def _db_store(self, with_transaction=True):
"""
Store a new node in the DB, also saving its repository directory
and attributes.
After being called attributes cannot be
changed anymore! Instead, extras can be changed only AFTER calling
this store() function.
:note: After successful storage, those links that are in the cache, and
for which also the parent node is already stored, will be
automatically stored. The others will remain unstored.
:parameter with_transaction: if False, no transaction is used. This
is meant to be used ONLY if the outer calling function has already
a transaction open!
:param bool use_cache: Whether I attempt to find an equal node in the DB.
"""
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
# TODO: This needs to be generalized, allowing for flexible methods
# for storing data and its attributes.
# I save the corresponding django entry
# I set the folder
# NOTE: I first store the files, then only if this is successful,
# I store the DB entry. In this way,
# I assume that if a node exists in the DB, its folder is in place.
# On the other hand, periodically the user might need to run some
# bookkeeping utility to check for lone folders.
self._repository_folder.replace_with_folder(
self._get_temp_folder().abspath, move=True, overwrite=True)
import aiida.backends.sqlalchemy
try:
# aiida.backends.sqlalchemy.get_scoped_session().add(self._dbnode)
session.add(self._dbnode)
# Save its attributes 'manually' without incrementing
# the version for each add.
self._dbnode.attributes = self._attrs_cache
flag_modified(self._dbnode, "attributes")
# This should not be used anymore: I delete it to
# possibly free memory
del self._attrs_cache
self._temp_folder = None
self._to_be_stored = False
# Here, I store those links that were in the cache and
# that are between stored nodes.
self._store_cached_input_links(with_transaction=False)
if with_transaction:
try:
# aiida.backends.sqlalchemy.get_scoped_session().commit()
session.commit()
except SQLAlchemyError as e:
# print "Cannot store the node. Original exception: {" \
# "}".format(e)
session.rollback()
raise
# This is one of the few cases where it is ok to do a 'global'
# except, also because I am re-raising the exception
except:
# I put back the files in the sandbox folder since the
# transaction did not succeed
self._get_temp_folder().replace_with_folder(
self._repository_folder.abspath, move=True, overwrite=True)
raise
self._dbnode.set_extra(_HASH_EXTRA_KEY, self.get_hash())
return self
@property
def uuid(self):
return unicode(self._dbnode.uuid)
[docs] def _attributes(self):
self._ensure_model_uptodate(['attributes'])
return self._dbnode.attributes
[docs] def _ensure_model_uptodate(self, attribute_names=None):
if self.is_stored:
self._dbnode.session.expire(self._dbnode, attribute_names=attribute_names)