Merge pull request #176 from sebastienthebaud/master

update GmailConnector create_label : do not raise error if user try t…

melusine/_config.py

@@ -1,6 +1,7 @@
 """
 Module which handles the package configuration.
 """
+
 import copy
 import logging
 import os

melusine/backend/base_backend.py

@@ -7,6 +7,7 @@
     BaseTransformerBackend,
 ]
 """
+
 from abc import ABC, abstractmethod
 from typing import Any, Callable, List, Optional
 

melusine/backend/pandas_backend.py

@@ -5,6 +5,7 @@
     PandasBackend,
 ]
 """
+
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 import numpy as np

melusine/base.py

@@ -11,6 +11,7 @@
     MelusineFeatureEncoder
 ]
 """
+
 from __future__ import annotations
 
 import copy

melusine/connectors/gmail.py

@@ -11,6 +11,7 @@
 from google.oauth2.credentials import Credentials
 from google_auth_oauthlib.flow import InstalledAppFlow
 from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
 from tqdm import tqdm
 
 logger = logging.getLogger(__name__)
@@ -158,8 +159,15 @@ def create_label(self, label_name: str) -> Dict[str, str]:
         Returns:
             Dict[str, str]: return from the api with label and its informations
         """
-        label = self.service.users().labels().create(userId="me", body=dict(name=label_name)).execute()
-        logger.info(f"Label {label_name} has been created.")
+        try:
+            label = self.service.users().labels().create(userId="me", body=dict(name=label_name)).execute()
+            logger.info(f"Label {label_name} has been created.")
+        except HttpError as error:
+            if error.resp.status == 409:  # Conflict error if label already exists
+                logger.error(f"Label '{label_name}' already exists.")
+                return {}
+            else:
+                raise
         return label
 
     @staticmethod

melusine/detectors.py

@@ -5,6 +5,7 @@
 ReplyDetector, TransferDetector, RecipientsDetector]
 
 """
+
 from typing import Any, Dict, List, Tuple
 
 from melusine.base import MelusineDetector, MelusineItem, MelusineRegex

melusine/io/_classes.py

@@ -3,6 +3,7 @@
 
 Contained classes: [IoMixin]
 """
+
 from __future__ import annotations
 
 import logging

melusine/message.py

@@ -5,6 +5,7 @@
 
 Implemented classes: [Message]
 """
+
 import re
 from datetime import datetime
 from typing import Iterable, List, Optional, Tuple

melusine/pipeline.py

@@ -3,6 +3,7 @@
 
 Implemented classes: [PipelineConfigurationError, MelusinePipeline]
 """
+
 from __future__ import annotations
 
 import copy

melusine/processors.py

@@ -18,6 +18,7 @@
     Cleaner,
 ]
 """
+
 from __future__ import annotations
 
 import logging
@@ -1901,7 +1902,6 @@ def clean(self, text: str) -> str:
 
 
 class DateProcessor(MelusineTransformer):
-
     """
     Parse string date to iso format string date
     """

melusine/regex/__init__.py

@@ -1,6 +1,7 @@
 """
 The melusine.regex module includes tools for handling regexes.
 """
+
 from melusine.regex.emergency_regex import EmergencyRegex
 from melusine.regex.reply_regex import ReplyRegex
 from melusine.regex.thanks_regex import ThanksRegex

melusine/testing/pipeline_testing.py

@@ -1,6 +1,7 @@
 """
 Module that contains utility functions for tests (in /tests).
 """
+
 from typing import Any, Dict
 
 from melusine.base import MelusineTransformer

melusine/utils/__init__.py

@@ -1,6 +1,7 @@
 """
 The melusine.utils module includes utils functionalitites.
 """
+
 from melusine.utils.show_versions import show_versions
 
 __all__ = ["show_versions"]

tests/detectors/test_reply_detector.py

@@ -1,6 +1,7 @@
 """
 Unit tests of the ReplyDetector.
 """
+
 import pandas as pd
 import pytest
 from pandas import DataFrame

tests/detectors/test_thanks_detector.py

@@ -2,6 +2,7 @@
 Unit test of the ThanksDetector.
 
 """
+
 from tempfile import TemporaryDirectory
 
 import pandas as pd

tests/detectors/test_transfer_detector.py

@@ -2,7 +2,6 @@
 Unit tests of the TransferDetector.
 """
 
-
 import pandas as pd
 import pytest
 from pandas import DataFrame

tests/detectors/test_vacation_reply_detector.py

@@ -1,6 +1,7 @@
 """
 Unit tests of the VacationReplyDetector
 """
+
 import pandas as pd
 import pytest
 from pandas import DataFrame

tests/functional/test_emails_fixtures.py

@@ -41,6 +41,7 @@
 }
 ========================================================================================
 """
+
 import pytest
 
 testcase_initial_cleaning_1 = dict(

tests/pipeline/test_pipeline.py

@@ -1,6 +1,7 @@
 """
 Unit test for pipeline.py
 """
+
 import pandas as pd
 import pytest
 

tests/pipeline/test_pipeline_basic.py

@@ -1,6 +1,7 @@
 """
 Example script to fit a minimal preprocessing pipeline
 """
+
 import pandas as pd
 import pytest
 

tests/processors/test_processors.py

@@ -1,6 +1,7 @@
 """
 Unit test for processors.py
 """
+
 import pytest
 
 from melusine.processors import (