From 2c9cdc0b8e7de592214f32a51ec0aafa8c8e0daa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20tom=20W=C3=B6rden?= <henrik@trineo.org>
Date: Thu, 15 Aug 2019 08:11:43 +0200
Subject: [PATCH] ENH: allow to set datatype when using assure_has_property
---
src/caosadvancedtools/cfood.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/caosadvancedtools/cfood.py b/src/caosadvancedtools/cfood.py
index 3058e58f..bd5cb510 100644
--- a/src/caosadvancedtools/cfood.py
+++ b/src/caosadvancedtools/cfood.py
@@ -299,7 +299,7 @@ def assure_has_parent(entity, parent, to_be_updated=None, verbosity=INFO):
def assure_has_property(entity, name, value, to_be_updated=None,
- verbosity=INFO):
+ verbosity=INFO, datatype=None):
"""
Checks whether `entity` has a property `name` with the value `value`.
@@ -332,7 +332,11 @@ def assure_has_property(entity, name, value, to_be_updated=None,
name,
value, entity.id))
print("/"*60)
- entity.add_property(name=name, value=value)
+
+ if datatype is None:
+ entity.add_property(name=name, value=value)
+ else:
+ entity.add_property(name=name, value=value, datatype=datatype)
if to_be_updated is None:
get_ids_for_entities_with_names([entity])
--
GitLab