From 875503b4a372529a7f58dfd77e6ae0c3aa63574d Mon Sep 17 00:00:00 2001
From: Jonathan Weth <git@jonathanweth.de>
Date: Tue, 7 Apr 2020 15:19:13 +0200
Subject: [PATCH] Add util function connect_untis_fields
---
aleksis/apps/untis/util/mysql/util.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/aleksis/apps/untis/util/mysql/util.py b/aleksis/apps/untis/util/mysql/util.py
index c7edfc4..fc8c024 100644
--- a/aleksis/apps/untis/util/mysql/util.py
+++ b/aleksis/apps/untis/util/mysql/util.py
@@ -132,3 +132,18 @@ def compare_m2m(
ids_a = sorted([i.id for i in a])
ids_b = sorted([i.id for i in b])
return ids_a == ids_b
+
+
+def connect_untis_fields(obj: Model, attr: str, limit: int) -> List[str]:
+ """ Connects data from multiple DB fields """
+
+ all_data = []
+
+ for i in range(1, limit + 1):
+ attr_name = "{}{}".format(attr, i)
+ raw_data = getattr(obj, attr_name, "")
+ if raw_data not in ("", None):
+ data = untis_split_first(raw_data)
+ all_data += data
+
+ return all_data
--
GitLab