refactor(mapping): add task_parent not last version

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())
 

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,
+#         )