feat: Add more fields to types.VFolderMount (#122)

* These may be useful for searching/analyzing vfolder mounts
lablup · Mar 10, 2022 · 6c031bb · 6c031bb
1 parent 8d67988
commit 6c031bb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/changes/122.feature.md b/changes/122.feature.md
@@ -0,0 +1 @@
+Add more fields to `types.VFolderMount` as a follow-up of #121
diff --git a/src/ai/backend/common/types.py b/src/ai/backend/common/types.py
@@ -684,13 +684,17 @@ def as_trafaret(cls) -> t.Trafaret:
 @attr.define(slots=True)
 class VFolderMount(JSONSerializableMixin):
     name: str
+    vfid: uuid.UUID
+    vfsubpath: PurePosixPath
     host_path: PurePosixPath
     kernel_path: PurePosixPath
     mount_perm: MountPermission
 
     def to_json(self) -> dict[str, Any]:
         return {
             'name': self.name,
+            'vfid': str(self.vfid),
+            'vfsubpath': str(self.vfsubpath),
             'host_path': str(self.host_path),
             'kernel_path': str(self.kernel_path),
             'mount_perm': self.mount_perm.value,
@@ -705,6 +709,8 @@ def as_trafaret(cls) -> t.Trafaret:
         from . import validators as tx
         return t.Dict({
             t.Key('name'): t.String,
+            t.Key('vfid'): tx.UUID,
+            t.Key('vfsubpath', default="."): tx.PurePath,
             t.Key('host_path'): tx.PurePath,
             t.Key('kernel_path'): tx.PurePath,
             t.Key('mount_perm'): tx.Enum(MountPermission),