summaryrefslogtreecommitdiff
path: root/api/v1/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'api/v1/util.py')
-rw-r--r--api/v1/util.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/api/v1/util.py b/api/v1/util.py
index f2dddad..3b79b82 100644
--- a/api/v1/util.py
+++ b/api/v1/util.py
@@ -3,6 +3,8 @@ import typing
from flask import jsonify, request
from peewee import Model, ModelSelect
+from api.v1.db import DataPack, Tag, TagRelation
+
_V = typing.TypeVar('_V', bound=Model)
@@ -59,3 +61,8 @@ def query_paginator(query: ModelSelect, mapping: typing.Callable[[_V], dict]):
'next': offset + size if more else None,
'results': res,
})
+
+
+def get_tags_for_datapack(datapack: DataPack):
+ return [tag.name for tag in
+ Tag.select().join(TagRelation).join(DataPack).where(TagRelation.pack.id == int(datapack.id))]