From 32213cdb49a08bd0f2012baaefb031d0726dc4bf Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Fri, 18 Feb 2022 16:15:00 +0100
Subject: move dataref export to cli.py

---
 lib/cli.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'lib')

diff --git a/lib/cli.py b/lib/cli.py
index 1ee1f33..3e7865b 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -143,6 +143,18 @@ def model_quality_table(header, result_lists, info_list):
             print(buf)
 
 
+def export_dataref(dref_file, dref):
+    with open(dref_file, "w") as f:
+        for k, v in sorted(dref.items()):
+            if type(v) is not tuple:
+                v = (v, None)
+            if v[1] is None:
+                prefix = r"\drefset{"
+            else:
+                prefix = r"\drefset" + f"[unit={v[1]}]" + "{"
+            print(f"{prefix}/{k}" + "}{" + str(v[0]) + "}", file=f)
+
+
 def add_standard_arguments(parser):
     parser.add_argument(
         "--export-dref",
-- 
cgit v1.2.3