refactor(resolved_tasks):add secrets to the .env

mapping.py

@@ -30,7 +30,6 @@ headers = {
 }
 
 
-
 def get_assignee_data():
     """ Получение данных для поля assignee """
 
@@ -82,7 +81,7 @@ def transform_data(data):
     for task in data:
         status, completed_at = get_task_status(task)
         transformed_task = {
-            'gid':task['gid'],
+            'gid': task['gid'],
             'summary': task['name'],
             'description': task['notes'],
             'createdAt': task['created_at'],
@@ -91,12 +90,12 @@ def transform_data(data):
             'status': status,
             'completedAt': completed_at,
             'parent': task['parent'],
-            'subtasks' : transform_data(task['subtasks'])
+            'subtasks': transform_data(task['subtasks'])
-            # 'parent': task['parent']  # это я добавила, чтоб parent добавить
         }
         transformed_data.append(transformed_task)
     return transformed_data
 
+
 def create_task(task):
     base_url = 'https://api.tracker.yandex.net/v2/issues/_import'
     print(task)
@@ -104,7 +103,7 @@ def create_task(task):
         ya_assignee = assignee.get(task['assignee']['gid'], 'dr.cyrill')
     else:
         ya_assignee = 'dr.cyrill'
-    
+
     payload = {
         'queue': QUEUE_NAME,
         'summary': task['summary'],
@@ -115,28 +114,28 @@ def create_task(task):
         'assignee': ya_assignee,
         'status': task['status'],
     }
-    
+
     logger.debug('Request: %s', json.dumps(payload))
-    
+
     response = requests.post(
         base_url,
         headers=headers,
         data=json.dumps(payload),
         params=urlencode({'parent': task['parent']})  # добавила параметр для parent в url-запрос 
     )
-    
+
     if response.status_code == 201:
         print('Задача успешно создана в ЯндексТрекер')
-    
+
         # здесь добавлено про родительскую задачу
         return response.json()['id']
-    
+
     else:
         print(
             'Ошибка при создании задачи в ЯндексТрекер:',
             response.content,
         )
-    
+
     logger.debug('Response: %s', response.content)
 
 
@@ -158,16 +157,9 @@ def create_tasks_in_tracker(data, limit=10):
             logger.info('Found subtask gid = %s for id = %s',subtask['gid'], task_id)
             subtask_id = create_task(subtask)
             assign_parent_task(subtask_id, task_id)
-            
-     
-    # вот здесь тоже про родительскую задачу добавлено
-    #parent_task_ids = get_parent_task_ids(task, ya_imported_task_response)
-
-    #for gid, parent_id in parent_task_ids.items():
-     #   assign_parent_task(ya_imported_task_response[gid], parent_id)
 
 
-#и эта функция для родительской задачи написана
+#  и эта функция для родительской задачи написана
 def assign_parent_task(task_id, parent_id):
     """ Назначение родительской задачи """
 
@@ -203,8 +195,51 @@ def assign_parent_task(task_id, parent_id):
 # здесь я закончила добавлять про родительскую задачу с большой надеждой
 
 
+#   пробую закрыть задачи уже в Трекере
+def close_tasks_for_completedAt(queue_name):
+    base_url = 'https://api.tracker.yandex.net/v2/issues/_search'
+    payload = {
+        "filter": {
+            "queue": queue_name,
+            "status": "resolved"  # Фильтр для задач cо статусом closed
+        }
+    }
+
+    response = requests.post(
+        base_url,
+        headers=headers,
+        data=json.dumps(payload)
+    )
+
+    if response.status_code == 200:
+        logger.debug('Response: %s', response.content)
+
+        tasks = response.json()
+
+        for task in tasks:
+            task_id = task['id']
+            close_payload = {
+                'resolution': 'fixed'
+            }
+            close_url = f'https://api.tracker.yandex.net/v2/issues/{task_id}/transitions/close/_execute'
+            print(close_url)
+            print(headers)
+            close_response = requests.post(
+                close_url,
+                headers=headers,
+                data=json.dumps(close_payload)
+            )
+            if close_response.status_code == 200:
+                print(f'Задача {task_id} закрыта успешно')
+            else:
+                print(f'Ошибка при закрытии задачи {task_id}: {close_response.content}')
+    else:
+        print(f'Ошибка при получении задач для очереди {queue_name}: {response.content}')
+
+
 file = open(asana_data_json, "r", encoding="utf8")
 json_data = json.loads(file.read())
 
 yandex_tracker_data = transform_data(json_data['data'])
 create_tasks_in_tracker(yandex_tracker_data, limit=3)
+close_tasks_for_completedAt('TESTIMPORT')

requirements.txt

@@ -1,6 +1,7 @@
 certifi==2023.11.17
 charset-normalizer==3.3.2
 idna==3.4
+dotenv==1.0.1
 requests==2.31.0
 setuptools==69.0.0
 six==1.16.0

resolved_task.py

@@ -1,14 +1,14 @@
-import csv
 import json
 import os
 import sys
-import time
 
 import requests
 
 
+from dotenv import load_dotenv
 from logging_config import setup_logging
-from urllib.parse import urlencode
+
+load_dotenv()
 
 logger = setup_logging(__name__)
 
@@ -21,20 +21,14 @@ sys.stdout.reconfigure(encoding='utf-8')
 
 
 QUEUE_NAME = 'TESTIMPORT'
-
+TOKEN = os.getenv('TOKEN')
-# headers = {
-#     'Host': 'api.tracker.yandex.net',
-#     'Authorization': 'OAuth y0_AgAEA7qkB3KjAArkXwAAAADzMlP9oR1lwMzBS2e94jHzdnII8Laxi7k',
-#     'X-Org-ID': '7095769',
-#     'Content-Type': 'appication/json',
-# }
-
 
 def close_tasks_in_queue(queue_name):
     base_url = 'https://api.tracker.yandex.net/v2/issues/_search'
+    # base_url = 'https://api.tracker.yandex.net/v2/issues/_search?expand=transitions'
     headers = {
         'Host': 'api.tracker.yandex.net',
-        'Authorization': 'OAuth y0_AgAEA7qkB3KjAArkXwAAAADzMlP9oR1lwMzBS2e94jHzdnII8Laxi7k',
+        'Authorization': TOKEN,
         'X-Org-ID': '7095769',
         'Content-Type': 'application/json',
     }
@@ -46,31 +40,39 @@ def close_tasks_in_queue(queue_name):
         }
     }
 
-    response = requests.post(
+    while True:
-        base_url,
+
-        headers=headers,
+        response = requests.post(
-        data=json.dumps(payload)
+            base_url,
-    )
+            headers=headers,
+            data=json.dumps(payload)
+        )
+
+        if response.status_code == 200:
+            logger.debug('Response: %s', response.content)
+
+            tasks = response.json()
+
+            for task in tasks:
+                task_id = task['id']
+                close_payload = {
+                    'resolution': 'fixed'
+                }
+                close_url = f'https://api.tracker.yandex.net/v2/issues/{task_id}/transitions/close/_execute'
+                print(close_url)
+                print(headers)
+                close_response = requests.post(
+                    close_url,
+                    headers=headers,
+                    data=json.dumps(close_payload)
+                )
+                if close_response.status_code == 200:
+                    print(f'Задача {task_id} закрыта успешно')
+                else:
+                    print(f'Ошибка при закрытии задачи {task_id}: {close_response.content}')
+        else:
+            print(f'Ошибка при получении задач для очереди {queue_name}: {response.content}')
 
-    if response.status_code == 200:
-        tasks = response.json().get('issues', [])
-        for task in tasks:
-            task_id = task['id']
-            close_payload = {
-                'status': 'resolved'  # Установка статуса "Закрыт"
-            }
-            close_url = f'https://api.tracker.yandex.net/v2/issues/{task_id}'
-            close_response = requests.patch(
-                close_url,
-                headers=headers,
-                data=json.dumps(close_payload)
-            )
-            if close_response.status_code == 200:
-                print(f'Задача {task_id} закрыта успешно')
-            else:
-                print(f'Ошибка при закрытии задачи {task_id}: {close_response.content}')
-    else:
-        print(f'Ошибка при получении задач для очереди {queue_name}: {response.content}')
 
 # Вызов функции для закрытия задач в очереди 'TESTIMPORT'
 close_tasks_in_queue('TESTIMPORT')