aboutsummaryrefslogtreecommitdiff
path: root/gn3/db/case_attributes.py
blob: bb1524884345c0b372fc0feb31c1bbc83d01ec41 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
"""Module that contains functions for editing case-attribute data"""
from typing import Any, Optional, Tuple

import json
import MySQLdb


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")
        return cursor.fetchall()


def get_unreviewed_diffs(conn: Any) -> Optional[tuple]:
    """Fetch all case attributes in GN"""
    with conn.cursor() as cursor:
        cursor.execute(
            "SELECT id, editor, json_diff_data FROM "
            "caseattributes_audit WHERE status = 'review'"
        )
        return cursor.fetchall()


def insert_case_attribute_audit(
    conn: Any, status: str, author: str, data: str
) -> int:
    """Update the case_attribute_audit table"""
    rowcount = 0
    try:
        with conn.cursor() as cursor:
            cursor.execute(
                "INSERT INTO caseattributes_audit "
                "(status, editor, json_diff_data) "
                "VALUES (%s, %s, %s)",
                (status, author, data,),
            )
            rowcount = cursor.rowcount
    except Exception as _e:
        raise MySQLdb.Error(_e) from _e
    return rowcount


def reject_case_attribute(conn: Any, case_attr_audit_id: int) -> int:
    """Given the id of the json_diff in the case_attribute_audit table, reject
    it"""
    rowcount = 0
    try:
        with conn.cursor() as cursor:
            cursor.execute(
                "UPDATE caseattributes_audit SET "
                "status = 'rejected' WHERE id = %s",
                (case_attr_audit_id,),
            )
            rowcount = cursor.rowcount
    except Exception as _e:
        raise MySQLdb.Error(_e) from _e
    return rowcount


def approve_case_attribute(conn: Any, case_attr_audit_id: int) -> int:
    """Given the id of the json_diff in the case_attribute_audit table,
    approve it

    """
    rowcount = 0
    try:
        with conn.cursor() as cursor:
            cursor.execute(
                "SELECT json_diff_data FROM caseattributes_audit "
                "WHERE id = %s",
                (case_attr_audit_id,),
            )
            diff_data = cursor.fetchone()
            if diff_data:
                diff_data = json.loads(diff_data[0])
                # Insert (Most Important)
                if diff_data.get("Insert"):
                    data = diff_data.get("Insert")
                    cursor.execute(
                        "INSERT INTO CaseAttribute "
                        "(Name, Description) VALUES "
                        "(%s, %s)",
                        (
                            data.get("name").strip(),
                            data.get("description").strip(),
                        ),
                    )
                # Delete
                elif diff_data.get("Deletion"):
                    data = diff_data.get("Deletion")
                    cursor.execute(
                        "DELETE FROM CaseAttribute WHERE Id = %s",
                        (data.get("id"),),
                    )
                # Modification
                elif diff_data.get("Modification"):
                    data = diff_data.get("Modification")
                    if desc_ := data.get("description"):
                        cursor.execute(
                            "UPDATE CaseAttribute SET "
                            "Description = %s WHERE Id = %s",
                            (
                                desc_.get("Current"),
                                diff_data.get("id"),
                            ),
                        )
                    if name_ := data.get("name"):
                        cursor.execute(
                            "UPDATE CaseAttribute SET "
                            "Name = %s WHERE Id = %s",
                            (
                                name_.get("Current"),
                                diff_data.get("id"),
                            ),
                        )
                if cursor.rowcount:
                    cursor.execute(
                        "UPDATE caseattributes_audit SET "
                        "status = 'approved' WHERE id = %s",
                        (case_attr_audit_id,),
                    )
            rowcount = cursor.rowcount
    except Exception as _e:
        raise MySQLdb.Error(_e) from _e
    return rowcount