|
| 1 | +"""This module has functions to do with journal note related calls to the KMD Nova api.""" |
| 2 | + |
| 3 | +import base64 |
| 4 | +import uuid |
| 5 | +import urllib.parse |
| 6 | +from datetime import datetime |
| 7 | + |
| 8 | +import requests |
| 9 | + |
| 10 | +from itk_dev_shared_components.kmd_nova.authentication import NovaAccess |
| 11 | +from itk_dev_shared_components.kmd_nova.nova_objects import JournalNote |
| 12 | + |
| 13 | + |
| 14 | +def add_text_note(case_uuid: str, note_title: str, note_text: str, approved: bool, nova_access: NovaAccess) -> str: |
| 15 | + """Add a text based journal note to a Nova case. |
| 16 | +
|
| 17 | + Args: |
| 18 | + case_uuid: The uuid of the case to add the journal note to. |
| 19 | + note_title: The title of the note. |
| 20 | + note_text: The text content of the note. |
| 21 | + approved: Whether the journal note should be marked as approved in Nova. |
| 22 | + nova_access: The NovaAccess object used to authenticate. |
| 23 | +
|
| 24 | + Returns: |
| 25 | + The uuid of the created journal note. |
| 26 | + """ |
| 27 | + note_uuid = str(uuid.uuid4()) |
| 28 | + |
| 29 | + url = urllib.parse.urljoin(nova_access.domain, "api/Case/Update") |
| 30 | + params = {"api-version": "1.0-Case"} |
| 31 | + |
| 32 | + payload = { |
| 33 | + "common": { |
| 34 | + "transactionId": str(uuid.uuid4()), |
| 35 | + "uuid": case_uuid |
| 36 | + }, |
| 37 | + "journalNotes": [ |
| 38 | + { |
| 39 | + "uuid": note_uuid, |
| 40 | + "approved": approved, |
| 41 | + "journalNoteAttributes": { |
| 42 | + "journalNoteDate": datetime.today().isoformat(), |
| 43 | + "title": note_title, |
| 44 | + "journalNoteType": "Bruger", |
| 45 | + "format": "Text", |
| 46 | + "note": _encode_text(note_text) |
| 47 | + } |
| 48 | + } |
| 49 | + ] |
| 50 | + } |
| 51 | + |
| 52 | + headers = {'Content-Type': 'application/json', 'Authorization': f"Bearer {nova_access.get_bearer_token()}"} |
| 53 | + |
| 54 | + response = requests.patch(url, params=params, headers=headers, json=payload, timeout=60) |
| 55 | + response.raise_for_status() |
| 56 | + |
| 57 | + return note_uuid |
| 58 | + |
| 59 | + |
| 60 | +def _encode_text(string: str) -> str: |
| 61 | + """Encode a string to a base64 string. |
| 62 | + Ensure the base64 string doesn't contain padding by inserting spaces at the end of the input string. |
| 63 | + There is a bug in the Nova api that corrupts the string if it contains padding. |
| 64 | + The extra spaces will not show up in the Nova user interface. |
| 65 | +
|
| 66 | + Args: |
| 67 | + string: The string to encode. |
| 68 | +
|
| 69 | + Returns: |
| 70 | + A base64 string containing no padding. |
| 71 | + """ |
| 72 | + def b64(s: str) -> str: |
| 73 | + """Helper function to convert a string to base64.""" |
| 74 | + return base64.b64encode(s.encode()).decode() |
| 75 | + |
| 76 | + while (s := b64(string)).endswith("="): |
| 77 | + string += ' ' |
| 78 | + |
| 79 | + return s |
| 80 | + |
| 81 | + |
| 82 | +def get_notes(case_uuid: str, nova_access: NovaAccess, offset: int = 0, limit: int = 100) -> tuple[JournalNote, ...]: |
| 83 | + """Get all journal notes from the given case. |
| 84 | +
|
| 85 | + Args: |
| 86 | + case_uuid: The uuid of the case to get notes from. |
| 87 | + nova_access: The NovaAccess object used to authenticate. |
| 88 | + offset: The number of journal notes to skip. |
| 89 | + limit: The maximum number of journal notes to get (1-500). |
| 90 | +
|
| 91 | + Returns: |
| 92 | + A tuple of JournalNote objects. |
| 93 | + """ |
| 94 | + url = urllib.parse.urljoin(nova_access.domain, "api/Case/GetList") |
| 95 | + params = {"api-version": "1.0-Case"} |
| 96 | + |
| 97 | + payload = { |
| 98 | + "common": { |
| 99 | + "transactionId": str(uuid.uuid4()), |
| 100 | + "uuid": case_uuid |
| 101 | + }, |
| 102 | + "paging": { |
| 103 | + "startRow": offset+1, |
| 104 | + "numberOfRows": limit |
| 105 | + }, |
| 106 | + "caseGetOutput": { |
| 107 | + "journalNotes": { |
| 108 | + "uuid": True, |
| 109 | + "approved": True, |
| 110 | + "journalNoteAttributes": { |
| 111 | + "title": True, |
| 112 | + "format": True, |
| 113 | + "note": True, |
| 114 | + "createdTime": True |
| 115 | + } |
| 116 | + } |
| 117 | + } |
| 118 | + } |
| 119 | + |
| 120 | + headers = {'Content-Type': 'application/json', 'Authorization': f"Bearer {nova_access.get_bearer_token()}"} |
| 121 | + |
| 122 | + response = requests.put(url, params=params, headers=headers, json=payload, timeout=60) |
| 123 | + response.raise_for_status() |
| 124 | + |
| 125 | + note_dicts = response.json()['cases'][0]['journalNotes']['journalNotes'] |
| 126 | + |
| 127 | + notes_list = [] |
| 128 | + |
| 129 | + for note_dict in note_dicts: |
| 130 | + notes_list.append( |
| 131 | + JournalNote( |
| 132 | + uuid=note_dict['uuid'], |
| 133 | + title=note_dict['journalNoteAttributes']['title'], |
| 134 | + approved=note_dict['approved'], |
| 135 | + journal_date=note_dict['journalNoteAttributes']['createdTime'], |
| 136 | + note=note_dict['journalNoteAttributes']['note'], |
| 137 | + note_format=note_dict['journalNoteAttributes']['format'] |
| 138 | + ) |
| 139 | + ) |
| 140 | + |
| 141 | + return tuple(notes_list) |
0 commit comments