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 | class JiraReader(BaseReader):
"""Jira reader. Reads data from Jira issues from passed query.
Args:
Optional basic_auth:{
"email": "email",
"api_token": "token",
"server_url": "server_url"
}
Optional oauth:{
"cloud_id": "cloud_id",
"api_token": "token"
}
"""
def __init__(
self,
email: Optional[str] = None,
api_token: Optional[str] = None,
server_url: Optional[str] = None,
BasicAuth: Optional[BasicAuth] = None,
Oauth2: Optional[Oauth2] = None,
) -> None:
from jira import JIRA
if email and api_token and server_url:
if BasicAuth is None:
BasicAuth = {}
BasicAuth["email"] = email
BasicAuth["api_token"] = api_token
BasicAuth["server_url"] = server_url
if Oauth2:
options = {
"server": f"https://api.atlassian.com/ex/jira/{Oauth2['cloud_id']}",
"headers": {"Authorization": f"Bearer {Oauth2['api_token']}"},
}
self.jira = JIRA(options=options)
else:
self.jira = JIRA(
basic_auth=(BasicAuth["email"], BasicAuth["api_token"]),
server=f"https://{BasicAuth['server_url']}",
)
def load_data(self, query: str) -> List[Document]:
relevant_issues = self.jira.search_issues(query)
issues = []
assignee = ""
reporter = ""
epic_key = ""
epic_summary = ""
epic_descripton = ""
for issue in relevant_issues:
# Iterates through only issues and not epics
if "parent" in (issue.raw["fields"]):
if issue.fields.assignee:
assignee = issue.fields.assignee.displayName
if issue.fields.reporter:
reporter = issue.fields.reporter.displayName
if issue.raw["fields"]["parent"]["key"]:
epic_key = issue.raw["fields"]["parent"]["key"]
if issue.raw["fields"]["parent"]["fields"]["summary"]:
epic_summary = issue.raw["fields"]["parent"]["fields"]["summary"]
if issue.raw["fields"]["parent"]["fields"]["status"]["description"]:
epic_descripton = issue.raw["fields"]["parent"]["fields"]["status"][
"description"
]
issues.append(
Document(
text=f"{issue.fields.summary} \n {issue.fields.description}",
extra_info={
"id": issue.id,
"title": issue.fields.summary,
"url": issue.permalink(),
"created_at": issue.fields.created,
"updated_at": issue.fields.updated,
"labels": issue.fields.labels,
"status": issue.fields.status.name,
"assignee": assignee,
"reporter": reporter,
"project": issue.fields.project.name,
"issue_type": issue.fields.issuetype.name,
"priority": issue.fields.priority.name,
"epic_key": epic_key,
"epic_summary": epic_summary,
"epic_description": epic_descripton,
},
)
)
return issues
|