From 915be612a30abd656ba25c6eb088408a27308da6 Mon Sep 17 00:00:00 2001 From: Angelina Tingaeva Date: Tue, 5 Dec 2023 17:04:11 +0500 Subject: [PATCH] refactor(mapping): add task_parent not last version --- mapping.py | 93 +++++++++++++++++++++++++-------------------- mapping_logging.bak | 37 ++++++++++++++++++ 2 files changed, 89 insertions(+), 41 deletions(-) diff --git a/mapping.py b/mapping.py index ec73632..7fae977 100644 --- a/mapping.py +++ b/mapping.py @@ -19,16 +19,6 @@ sys.stdout.reconfigure(encoding='utf-8') QUEUE_NAME = 'TESTIMPORT' -# ниже функция для случаев, когда не будет готового файла для импорта -# def get_data_from_asana(): -# """ Получение данных из Asana """ - -# headers = { -# 'Authorization': 'ASANA_TOKEN', -# } -# response = requests.get('https://api.asana.com/projects/PROJECT_ID/tasks', headers=headers) -# data = response.json() -# return data def get_assignee_data(): @@ -59,6 +49,21 @@ def get_task_status(task): completed_at = '' return status, completed_at +# здесь функция для дальнейшей работы с parent +def get_parent_task_ids(data, ya_imported_task_response): + """ Получение соответствий между идентификаторами подзадач в Асане и Яндекс Трекере """ + + parent_task_ids = {} + + for task in data["data"]: + if task.get('parent') and 'gid' in task['parent']: + parent_gid = task['parent']['gid'] + if parent_gid in ya_imported_task_response: + parent_id = ya_imported_task_response[parent_gid] + parent_task_ids[task['gid']] = parent_id + + return parent_task_ids + def transform_data(data): """ Преобразование данных из Asana в ЯндексТрекер в совместимом формате """ @@ -91,6 +96,10 @@ def create_tasks_in_tracker(data): } datalen = len(data) logger.info('Import started. Task count: %d', datalen) + + # этот датасет тоже для parent добавлен + ya_imported_task_response = {} + for task in data: if task.get('assignee') and 'gid' in task['assignee']: ya_assignee = assignee.get(task['assignee']['gid'], 'dr.cyrill') @@ -98,7 +107,7 @@ def create_tasks_in_tracker(data): ya_assignee = 'dr.cyrill' payload = { - 'queue': QUEUE_NAME, # Вынести в переменную + 'queue': QUEUE_NAME, 'summary': task['summary'], 'description': task['description'], 'createdAt': task['createdAt'], @@ -118,6 +127,9 @@ def create_tasks_in_tracker(data): if response.status_code == 201: print('Задача успешно создана в ЯндексТрекер') + + # здесь добавлено про родительскую задачу + ya_imported_task_response[task['gid']] = response.json()['id'] else: print( 'Ошибка при создании задачи в ЯндексТрекер:', @@ -126,45 +138,44 @@ def create_tasks_in_tracker(data): logger.debug('Response: %s', response.content) - ya_imported_task_response = response.json() - if task.get('parent'): - parent_task_gid = task['parent']['gid'] - if parent_task_gid in ya_imported_task_response: - parent_task_id = ya_imported_task_response[parent_task_gid]['id'] - logger.debag( - f'В задаче {task['summary']} есть связка "parent"', - ensure_ascii=False - ) - update_parent_task( - base_url, - headers, - ya_imported_task_response['id'], - parent_task_id, - ) + # вот здесь тоже про родительскую задачу добавлено + parent_task_ids = get_parent_task_ids(data, ya_imported_task_response) + for gid, parent_id in parent_task_ids.items(): + assign_parent_task(gid, parent_id) -def update_parent_task(base_url, headers, task_id, parent_task_id): - """ Обновление данных родительской задачи в ЯндексТрекере""" +#и эта функция для родительской задачи написана +def assign_parent_task(child_gid, parent_id): + """ Назначение родительской задачи """ - data = { - 'parent': parent_task_id + base_url = f'https://api.tracker.yandex.net/v2/issues/{child_gid}' + headers = { + 'Host': 'api.tracker.yandex.net', + 'Authorization': 'OAuth y0_AgAEA7qkB3KjAArkXwAAAADzMlP9oR1lwMzBS2e94jHzdnII8Laxi7k', + 'X-Org-ID': '7095769', + 'Content-Type': 'application/json', } - payload = json.dumps(data) - update_url = f"{base_url}/{task_id}" - response = requests.post(update_url, headers=headers, data=payload) - result = json.loads(response.text) - logger.warning(json.dumps(result, ensure_ascii=False)) + + payload = { + 'parent': parent_id, + } + + logger.debug('Request: %s', json.dumps(payload)) + + response = requests.patch( + base_url, + headers=headers, + data=json.dumps(payload), + ) if response.status_code == 200: - logger.info('Данные о связи с задачей "parent" успешно обновлены') + print('Родительская задача успешно назначена') else: - logger.error( - 'Ошибка при обновлении данных о связи с задачей "parent": %s', - response.content, - ) + print('Ошибка при назначении родительской задачи:', response.content) + logger.debug('Response: %s', response.content) +# здесь я закончила добавлять про родительскую задачу с большой надеждой -# asana_data = get_data_from_asana() file = open(asana_data_json, "r", encoding="utf8") json_data = json.loads(file.read()) diff --git a/mapping_logging.bak b/mapping_logging.bak index 6b0499a..5f5dbbb 100644 --- a/mapping_logging.bak +++ b/mapping_logging.bak @@ -43,3 +43,40 @@ def logRoundtrip(response, *args, **kwargs): session = requests.Session() session.hooks['response'].append(logRoundtrip) session.get('http://secariolabs.com') + + + + +# ya_imported_task_response = response.json() +# if task.get('parent'): +# parent_task_gid = task['parent']['gid'] +# if parent_task_gid in ya_imported_task_response: +# parent_task_id = ya_imported_task_response[parent_task_gid]['id'] +# logger.debug(f'В задаче {task["summary"]} есть связка "parent"', ensure_ascii=False) +# update_parent_task( +# base_url, +# headers, +# ya_imported_task_response['id'], +# parent_task_id, +# ) + + +# def update_parent_task(base_url, headers, task_id, parent_task_id): +# """ Обновление данных родительской задачи в ЯндексТрекере""" + +# data = { +# 'parent': parent_task_id +# } +# payload = json.dumps(data) +# update_url = f"{base_url}/{task_id}" +# response = requests.post(update_url, headers=headers, data=payload) +# result = json.loads(response.text) +# logger.warning(json.dumps(result, ensure_ascii=False)) + +# if response.status_code == 200: +# logger.info('Данные о связи с задачей "parent" успешно обновлены') +# else: +# logger.error( +# 'Ошибка при обновлении данных о связи с задачей "parent": %s', +# response.content, +# )