diff --git a/src/caosdb/common/models.py b/src/caosdb/common/models.py index d4c8968f2dd0bce312b12b0d92b4d158d7286088..9c420c8cae75171e0b233f2fd23f87dc2fa61367 100644 --- a/src/caosdb/common/models.py +++ b/src/caosdb/common/models.py @@ -37,6 +37,8 @@ from random import randint from sys import hexversion from tempfile import NamedTemporaryFile +from lxml import etree + from caosdb.common.utils import uuid, xml2str from caosdb.configuration import get_config from caosdb.connection.connection import get_connection @@ -47,7 +49,6 @@ from caosdb.exceptions import (AmbiguityException, AuthorizationException, EntityHasNoDatatypeError, TransactionError, UniqueNamesError, UnqualifiedParentsError, UnqualifiedPropertiesError, URITooLongException) -from lxml import etree _ENTITY_URI_SEGMENT = "Entity" @@ -3478,6 +3479,20 @@ class Info(): m = _parse_single_xml_element(e) self.messages.append(m) + def __str__(self): + if "counts" not in self.messages: + return "caosdb.Info" + + if int(self.messages["counts"]["records"]) > 0: + return "Connection to CaosDB with {} Records." .format( + self.messages["counts"]["records"] + ) + else: + return "Connection to CaosDB without Records." + + def __repr__(self): + return self.__str__() + class Permission():