127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
import requests
|
|
import json
|
|
import datetime
|
|
from dotenv import load_dotenv
|
|
from os import getenv
|
|
|
|
load_dotenv()
|
|
ACCESS_TOKEN = getenv("TOKEN")
|
|
USERNAMES = ["kalinom6"] # Modify
|
|
URL = "https://globaljira.roche.com/rest/api/2"
|
|
CERT = getenv("CERT_LOCATION")
|
|
WEEKS_BACK = 1
|
|
VERBOSE = True
|
|
|
|
headers = {
|
|
"Authorization": f"Bearer {ACCESS_TOKEN}",
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
def send_request(url, method, payload=None):
|
|
if method == "GET":
|
|
response = requests.get(url, headers=headers, data=json.dumps(payload), verify=CERT)
|
|
if method == "POST":
|
|
response = requests.post(url, headers=headers, data=json.dumps(payload), verify=CERT)
|
|
if response.status_code == 200:
|
|
return response.json()
|
|
print(f"Error: {response.status_code}, {response.text}")
|
|
|
|
def get_issues(username):
|
|
jql = f"assignee WAS '{username}'"
|
|
payload = {
|
|
"jql": jql,
|
|
"startAt": 0,
|
|
"maxResults": 50,
|
|
"fields": [
|
|
"key", "summary"
|
|
]
|
|
}
|
|
data = send_request(f'{URL}/search', "POST", payload)
|
|
if data:
|
|
tasks = []
|
|
for issue in data.get("issues", []):
|
|
task = {
|
|
"task_id": issue["key"],
|
|
"summary": issue["fields"]["summary"],
|
|
"task_link": f"https://globaljira.roche.com/browse/{issue['key']}"
|
|
}
|
|
tasks.append(task)
|
|
return tasks
|
|
|
|
def get_worklogs(issues):
|
|
worklogs = []
|
|
for issue in issues:
|
|
data = send_request(f'{URL}/issue/{issue["task_id"]}/worklog', "GET")
|
|
if data:
|
|
for log in data.get("worklogs", []):
|
|
worklog = {
|
|
"ticket_id": issue["task_id"],
|
|
"jira_link": issue["task_link"],
|
|
"time_spent": log["timeSpent"],
|
|
"date": log["started"],
|
|
"author": log["author"]["name"]
|
|
}
|
|
if worklog["author"] in USERNAMES:
|
|
worklogs.append(worklog)
|
|
return worklogs
|
|
|
|
def get_days(day, weeks_back=WEEKS_BACK):
|
|
if WEEKS_BACK:
|
|
day = day - datetime.timedelta(weeks=weeks_back)
|
|
year, weeknum, day_of_week = day.isocalendar()
|
|
# start at monday
|
|
while day_of_week != 1:
|
|
day = day - datetime.timedelta(days=1)
|
|
day_of_week -= 1
|
|
days = []
|
|
# end at sunday
|
|
while day_of_week <= 7:
|
|
days.append(str(day))
|
|
day = day + datetime.timedelta(days=1)
|
|
day_of_week += 1
|
|
return days
|
|
|
|
def get_week_tickets(worklogs, days):
|
|
this_week_tickets = []
|
|
for worklog in worklogs:
|
|
worklog_date = datetime.datetime.strptime(worklog["date"].split('T')[0], '%Y-%m-%d').date()
|
|
current_month = get_max_month(days)
|
|
worklog_month = str(worklog_date).split('-')[1]
|
|
if str(worklog_date) in days and int(worklog_month) == current_month:
|
|
this_week_tickets.append(worklog)
|
|
return this_week_tickets
|
|
|
|
def get_max_month(days):
|
|
max = 0
|
|
for day in days:
|
|
month = int(day.split('-')[1])
|
|
if month > max:
|
|
max = month
|
|
return max
|
|
|
|
def main():
|
|
print('The script starts its work...')
|
|
|
|
for username in USERNAMES:
|
|
issues = get_issues(username)
|
|
worklogs = get_worklogs(issues)
|
|
today = datetime.date.today()
|
|
days = get_days(today)
|
|
worklogs = get_week_tickets(worklogs, days)
|
|
omnimat_string = ""
|
|
|
|
for worklog in worklogs:
|
|
if worklog["ticket_id"] not in omnimat_string:
|
|
omnimat_string += worklog["ticket_id"] + '\n'
|
|
|
|
if VERBOSE:
|
|
print('Issues of the user:')
|
|
print(json.dumps(issues, indent=4))
|
|
print(f"Worklogs for the current week:")
|
|
print(json.dumps(worklogs, indent=4))
|
|
print("Days:")
|
|
print(days)
|
|
print(f"Omnimat string:")
|
|
print(omnimat_string)
|
|
if __name__ == '__main__':
|
|
main() |