diff --git a/trustgraph-base/trustgraph/api/library.py b/trustgraph-base/trustgraph/api/library.py index b421daf5..eb68fe40 100644 --- a/trustgraph-base/trustgraph/api/library.py +++ b/trustgraph-base/trustgraph/api/library.py @@ -196,14 +196,14 @@ class Library: return {} def start_processing( - self, document_id, processing_id, flow="0000", + self, id, document_id, flow="0000", user="trustgraph", collection="default", tags=[], ): input = { "operation": "add-processing", "processing-metadata": { - "id": processing_id, + "id": id, "document-id": document_id, "time": int(time.time()), "flow": flow, @@ -218,12 +218,12 @@ class Library: return {} def stop_processing( - self, processing_id, user="trustgraph" + self, id, user="trustgraph", ): input = { "operation": "remove-processing", - "processing-id": processing_id, + "processing-id": id, "user": user, } diff --git a/trustgraph-cli/scripts/tg-start-library-processing b/trustgraph-cli/scripts/tg-start-library-processing index cd49eee3..78749771 100644 --- a/trustgraph-cli/scripts/tg-start-library-processing +++ b/trustgraph-cli/scripts/tg-start-library-processing @@ -13,15 +13,24 @@ default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/') default_user = "trustgraph" def start_processing( - url, user, document_id, processing_id, flow, collection, tags + url, user, document_id, id, flow, collection, tags ): api = Api(url).library() - tags = tags.split(",") + if tags: + tags = tags.split(",") + else: + tags = [] - api.start_processing(document_id, processing_id, flow, user, - collection, tags) + api.start_processing( + id = id, + document_id = document_id, + flow = flow, + user = user, + collection = collection, + tags = tags + ) def main(): @@ -49,7 +58,7 @@ def main(): ) parser.add_argument( - '-d', '--did', '--document-id', + '-d', '--document-id', required=True, help=f'Document ID', ) @@ -79,7 +88,7 @@ def main(): url = args.api_url, user = args.user, document_id = args.document_id, - processing_id = args.id, + id = args.id, flow = args.flow_id, collection = args.collection, tags = args.tags diff --git a/trustgraph-cli/scripts/tg-stop-library-processing b/trustgraph-cli/scripts/tg-stop-library-processing index 732bd2b0..6602bd44 100644 --- a/trustgraph-cli/scripts/tg-stop-library-processing +++ b/trustgraph-cli/scripts/tg-stop-library-processing @@ -18,7 +18,7 @@ def stop_processing( api = Api(url).library() - api.stop_processing(user, id) + api.stop_processing(user = user, id = id) def main():