👌 IMPROVE: Add AuthInfo joins to QueryBuilder (#5195)

Allow for `with_user` and `with_computer` when querying `AuthInfo`
aiidateam · Oct 24, 2021 · 6e272a3 · 6e272a3
1 parent a80812c
commit 6e272a3
Show file tree

Hide file tree

Showing 4 changed files with 225 additions and 178 deletions.
diff --git a/aiida/orm/computers.py b/aiida/orm/computers.py
@@ -84,7 +84,7 @@ def objects(cls) -> ComputerCollection:  # pylint: disable=no-self-argument
     def __init__(  # pylint: disable=too-many-arguments
         self,
         label: str = None,
-        hostname: str = None,
+        hostname: str = '',
         description: str = '',
         transport_type: str = '',
         scheduler_type: str = '',
@@ -137,7 +137,7 @@ def _hostname_validator(cls, hostname: str) -> None:
         """
         Validates the hostname.
         """
-        if not hostname.strip():
+        if not (hostname or hostname.strip()):
             raise exceptions.ValidationError('No hostname specified')
 
     @classmethod

diff --git a/aiida/orm/implementation/querybuilder.py b/aiida/orm/implementation/querybuilder.py
@@ -41,7 +41,7 @@ class EntityTypes(Enum):
 
 
 EntityRelationships: Dict[str, Set[str]] = {
-    'authinfo': set(),
+    'authinfo': {'with_computer', 'with_user'},
     'comment': {'with_node', 'with_user'},
     'computer': {'with_node'},
     'group': {'with_node', 'with_user'},