From fd29d6a53caeedec432318cdf12bf882ae0f069b Mon Sep 17 00:00:00 2001
From: Angelina Tingaeva <angelinatingaeva@yandex.ru>
Date: Wed, 22 Nov 2023 11:51:22 +0500
Subject: [PATCH] refactor(mapping): mapping optimization solving

---
 README.md      |  2 +-
 asana_users.py |  3 +-
 mapping.py     | 93 ++++++++++++++++++++++++++++++++++++++++++--------
 3 files changed, 80 insertions(+), 18 deletions(-)

diff --git a/README.md b/README.md
index 75f591b..f460168 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@
 | "created_at" | "createdAt" | "Дата создания задачи" |
 | "due_on" | "deadline" | "Срок завершения задачи- дэдлайн" |
 | "assignee" | "assignee" | "Логин или идентификатор исполнителя" |
-| "section" | "section" | "Категория" |
+| "section" | "milestone" | "Категория" |
 | "memberships" | "memberships" | "Участники задачи" |
 | " - " | "parent" | "Связка с родительской задачей" |
 | "completed" | "status" | "Степень завершения" |
diff --git a/asana_users.py b/asana_users.py
index fdbcd3c..c68d784 100644
--- a/asana_users.py
+++ b/asana_users.py
@@ -1,6 +1,5 @@
-import json
 import csv
-
+import json
 
 asana_data = './data/data_asana.json'
 
diff --git a/mapping.py b/mapping.py
index ac28230..5337489 100644
--- a/mapping.py
+++ b/mapping.py
@@ -4,28 +4,91 @@ import requests
 from yandex_tracker_client import TrackerClient
 
 client = TrackerClient(
-    token=<'TOKEN'>,
-    org_id=<'ORG_ID'>
+    token='YANDEX_TRACKER_TOKEN',
+    org_id='ORG_ID'
 )
 
 
-TOKEN = 'y0_AgAEA7qkfTFqAArTqQAAAADyduv806rRKh8MS6GOg01ZdwPE-F4ZHuQ'
+YANDEX_TRACKER_TOKEN = 'y0_AgAEA7qkfTFqAArTqQAAAADyduv806rRKh8MS6GOg01ZdwPE-F4ZHuQ'
 ORG_ID = '35d164dcf81e48f2b6a532f4badddd33'
 
 
-# Чтение данных JSON из Asana
-asana_data = json.loads(requests.get('https://api.asana.com/tasks').text)
+asana_data_json = './data/data_asana.json'
 
-# Маппинг данных и создание или обновление задач в Яндекс Трекер
-for task in asana_data['tasks']:
-    yandex_data = {
-        'Название': task['name'],
-        'Описание': task['description'],
-        # Другие поля
+
+# ниже функция для случаев, когда не будет готового файла для импорта
+# 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 transform_data(data):
+    """ Преобразование данных из Asana в Яндекс Трекер в совместимом формате """
+
+    transformed_data = []
+    for task in data:
+        transformed_task = {
+            'summary': task['name'],
+            'description': task['notes'],
+            'createdAt': task['created_at'],
+            'deadline': task['due_on'],
+            'assignee': task['assignee'],
+            'milestone': task['section'],
+            # 'memberships': task['memberships'],
+            # 'parent': task[''],
+            'status': task['completed'],
+        }
+        transformed_data.append(transformed_task)
+    return transformed_data
+
+
+def create_tasks_in_tracker(data):
+    """ Создание задач в Яндекс Трекер """
+
+    base_url = 'https://api.tracker.yandex.net/v2/issues'
+    headers = {
+        'Authorization': 'YANDEX_TRACKER_TOKEN',
+        'Content-Type': 'asana_data_json',
     }
+    for task in data:
+        payload = {
+            'summary': task['summary'],
+            'description': task['description'],
+            'createdAt': task['createdAt'],
+            'deadline': task['deadline'],
+            'assignee': task['assignee'],
+            'milestone': task['milestone'],
+            # 'memberships': task['memberships'],
+            # 'parent': task['parent'],
+            'status': task['status'],
+        }
+        response = requests.post(
+            base_url,
+            headers=headers,
+            data=json.dumps(payload),
+        )
+        if response.status_code == 201:
+            print('Задача успешно создана в Яндекс.Трекер')
+        else:
+            print(
+                'Ошибка при создании задачи в Яндекс.Трекер:',
+                response.content,
+            )
 
-    # Преобразование данных, если необходимо
-    # yandex_data['Дата'] = task['due_date'].split('T')[0]  # Преобразование даты
 
-    # Создание или обновление задачи в Яндекс Трекер
-    requests.post('https://api.yandextracker.com/v2/issues', data=yandex_data)
\ No newline at end of file
+def main():
+    """ Основная функция для переноса данных """
+    # asana_data = get_data_from_asana()
+    asana_data = asana_data_json
+    yandex_tracker_data = transform_data(asana_data)
+    create_tasks_in_tracker(yandex_tracker_data)
+
+
+# Запуск основной функции
+main()