diff options
Diffstat (limited to 'gn3/db')
| -rw-r--r-- | gn3/db/sample_data.py | 27 |
1 files changed, 4 insertions, 23 deletions
diff --git a/gn3/db/sample_data.py b/gn3/db/sample_data.py index 3f7e2da..f79ce97 100644 --- a/gn3/db/sample_data.py +++ b/gn3/db/sample_data.py @@ -1,5 +1,5 @@ """Module containing functions that work with sample data""" -from typing import Any, Tuple, Dict, Callable +from typing import Any, Tuple, Dict, Callable, Optional import re import collections @@ -425,27 +425,8 @@ def insert_sample_data( raise MySQLdb.Error(_e) from _e -def get_case_attributes(conn) -> dict: - """Get all the case attributes as a dictionary from the database. Should there - exist more than one case attribute with the same name, put the id in - brackets.""" - results = {} +def get_case_attributes(conn) -> Optional[Tuple]: + """Get all the case attributes from the database.""" with conn.cursor() as cursor: cursor.execute("SELECT Id, Name, Description FROM CaseAttribute") - _r = cursor.fetchall() - _dups = [ - item - for item, count in collections.Counter( - [name for _, name, _ in _r] - ).items() - if count > 1 - ] - for _id, _name, _desc in _r: - _name = _name.strip() - _desc = _desc if _desc else "" - if _name in _dups: - results[f"{_name} ({_id})"] = _desc - else: - results[f"{_name}"] = _desc - - return results + return cursor.fetchall() |