From 99dbe2104624ccf52496d4f60485d8caf8c55ffb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alexa=20Valentov=C3=A1?= <git@imaniti.org>
Date: Thu, 28 Nov 2024 16:11:01 +0100
Subject: [PATCH] handle transport server errors
---
shared/people_import.py | 15 ++++++++++++---
1 file changed, 12 insertions(+), 3 deletions(-)
diff --git a/shared/people_import.py b/shared/people_import.py
index 9f4564b4..2d210e92 100644
--- a/shared/people_import.py
+++ b/shared/people_import.py
@@ -7,6 +7,7 @@ import requests
from django.conf import settings
from django.core.files.images import ImageFile
from gql import Client, gql
+from gql.transport.exceptions import TransportServerError
from gql.transport.aiohttp import AIOHTTPTransport
from wagtail.images.models import Image
from wagtail.models.media import Collection
@@ -447,7 +448,7 @@ class PeopleTeamImporter(ImporterMixin):
if os.path.exists(lock_file_name):
os.remove(lock_file_name)
- def get_people_ids_from_group(self):
+ def get_people_ids_from_team(self):
query = gql(
f"""
query {{
@@ -473,7 +474,15 @@ class PeopleTeamImporter(ImporterMixin):
"""
)
- result = self.client.execute(query)
+ try:
+ result = self.client.execute(query)
+ except TransportServerError as exception:
+ logger.warning(
+ f"Error getting data for group %s: %s",
+ self.team_shortcut,
+ str(exception)
+ )
+ return []
users = []
@@ -540,7 +549,7 @@ class PeopleTeamImporter(ImporterMixin):
)
try:
- people_ids = self.get_people_ids_from_group()
+ people_ids = self.get_people_ids_from_team()
people_profiles = self.get_processed_people_profiles(people_ids)
people_instances = self.create_and_update_people_models(people_profiles)
--
GitLab