From 3abc8ca3d0a60c1fa0790b4381a1b9a56b723497 Mon Sep 17 00:00:00 2001
From: Maximilian Moser <maximilian.moser@tuwien.ac.at>
Date: Fri, 29 Jan 2021 09:23:40 +0100
Subject: [PATCH] drafts: remove direct import of DraftMetadata
* get the DraftMetadata class from the record service instead of
directly importing it from invenio-rdm-records
* prevents issues with renaming of the metadata class, etc.
---
invenio_utilities_tuw/cli/drafts.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/invenio_utilities_tuw/cli/drafts.py b/invenio_utilities_tuw/cli/drafts.py
index 738ea61..718c3fb 100644
--- a/invenio_utilities_tuw/cli/drafts.py
+++ b/invenio_utilities_tuw/cli/drafts.py
@@ -8,7 +8,6 @@ from os.path import basename, isdir, isfile, join
import click
from flask.cli import with_appcontext
from invenio_files_rest.models import ObjectVersion
-from invenio_rdm_records.records.models import DraftMetadata
from ..utils import get_draft_file_service, get_record_service
from .utils import (
@@ -60,7 +59,7 @@ def list_drafts(user):
service = get_record_service()
recids = [
dm.json["id"]
- for dm in DraftMetadata.query.all()
+ for dm in service.draft_cls.model_cls.query.all()
if dm is not None and dm.json is not None
]
--
GitLab