implement addSession (#159)

Author: Affie

src/navability/entities/session.py

@@ -43,7 +43,7 @@ class SessionSchema(Schema):
     id = fields.UUID(allow_none=True)
     label = fields.String(required=True)
 
-    metadata = fields.Method("get_metadata", "set_metadata")
+    metadata = fields.Method("get_metadata", "set_metadata", allow_none=True)
     originLatitude = fields.String(allow_none=True)
     originLongitude = fields.String(allow_none=True)
     createdTimestamp = fields.DateTime(required=True)
@@ -63,7 +63,7 @@ class SessionSchema(Schema):
     @post_load
     def make_session(self, data, **kwargs):
         return Session(**data)
-    
+
     def get_metadata(self, obj):
         return base64.b64encode(json.dumps(obj).encode())
 

src/navability/services/__init__.py

@@ -10,6 +10,7 @@
 from .blobentry import *
 from .loader import *
 from .graph import *
+from .session import *
 
 # Maps
 from .map import *

src/navability/services/blobentry.py

@@ -119,6 +119,7 @@ async def listBlobEntriesAsync(
         entry['label'] for entry in entries
     ]
 
+
 def listBlobEntries(
     fgclient: DFGClient,
     variableLabel: str,

src/navability/services/loader.py

@@ -7,6 +7,7 @@
 from gql import gql  # used to parse GraphQL queries
 from graphql import GraphQLSyntaxError  # used to handle GraphQL errors
 
+
 # Define a class to represent a GraphQL fragment with a name and data
 class Fragment:
     def __init__(self, name: str, data: str):

src/navability/services/session.py

@@ -0,0 +1,52 @@
+import logging
+import asyncio
+from navability.entities.navabilityclient import (
+    NavAbilityClient,
+    # QueryOptions,
+    MutationOptions,
+)
+from navability.entities.session import SessionSchema
+from navability.services.loader import GQL_OPERATIONS
+from navability.common.versions import payload_version
+
+import nest_asyncio
+nest_asyncio.apply()
+
+logger = logging.getLogger(__name__)
+
+
+async def addSessionAsync(
+    navAbilityClient: NavAbilityClient,
+    userLabel: str,
+    robotLabel: str,
+    sessionLabel: str,
+):
+    """Add a new session to the given user and robot.
+
+    Args:
+        navAbilityClient (NavAbilityClient): The NavAbility client.
+    """
+    params = {
+        "userLabel": userLabel,
+        "robotLabel": robotLabel,
+        "sessionLabel": sessionLabel,
+        "version": payload_version,
+    }
+    result = await navAbilityClient.mutate(
+        MutationOptions(
+            GQL_OPERATIONS["MUTATION_ADD_SESSION"].data,
+            params,
+        )
+    )
+
+    return SessionSchema().load(result["addSessions"]['sessions'][0])
+
+
+def addSession(
+    navAbilityClient: NavAbilityClient,
+    userLabel: str,
+    robotLabel: str,
+    sessionLabel: str,
+):
+    tsk = addSessionAsync(navAbilityClient, userLabel, robotLabel, sessionLabel)
+    return asyncio.run(tsk)

src/sdkCommonGQL

@@ -0,0 +1,15 @@
+import random
+import string
+from navability.entities.navabilityclient import NavAbilityHttpsClient
+from navability.services import addSession
+
+
+def test_addSession():
+    client = NavAbilityHttpsClient()
+    userLabel = "[email protected]"
+    robotLabel = "TestRobot"
+    sessionLabel = "Test_addSession_" + "".join(
+        random.choices(string.ascii_letters + string.digits, k=4)
+    )
+    session = addSession(client, userLabel, robotLabel, sessionLabel)
+    assert session.label == sessionLabel