fix lint issues

Author: Nidhi091999

pro_tes/config_models.py

@@ -1,14 +1,15 @@
 """Custom app config models."""
 
 from typing import Optional
-from pathlib import Path
-
 from pydantic import BaseModel  # pylint: disable=no-name-in-module
 from typing import List
 from pro_wes.ga4gh.wes.models import ServiceInfoBase as ServiceInfo
+import string
+
 
 # pragma pylint: disable=too-few-public-methods
 
+
 class DB(BaseModel):
     """DB config for post_task.
 
@@ -18,6 +19,7 @@ class DB(BaseModel):
     Attributes:
         insert_attempts: Number of attempts to insert a new task in DB.
     """
+
     insert_attempts: int = 10
 
 
@@ -32,6 +34,7 @@ class TaskID(BaseModel):
         charset: Characters to use when generating task IDs.
         length: Length of the generated task ID.
     """
+
     charset: str = string.ascii_uppercase + string.digits
     length: int = 6
 
@@ -49,6 +52,7 @@ class Timeout(BaseModel):
         poll: Timeout for polling.
         job: Timeout for job execution (None disables timeout).
     """
+
     post: Optional[int] = None
     poll: int = 2
     job: Optional[int] = None
@@ -65,6 +69,7 @@ class Polling(BaseModel):
         wait: Wait time between polling attempts.
         attempts: Max polling attempts before failure.
     """
+
     wait: int = 3
     attempts: int = 100
 
@@ -84,6 +89,7 @@ class PostTask(BaseModel):
         timeout: Timeout settings.
         polling: Polling behavior.
     """
+
     db: DB = DB()
     task_id: TaskID = TaskID()
     timeout: Timeout = Timeout()
@@ -99,6 +105,7 @@ class ListTasks(BaseModel):
     Attributes:
         default_page_size: Default pagination size.
     """
+
     default_page_size: int = 5
 
 
@@ -111,6 +118,7 @@ class Monitor(BaseModel):
     Attributes:
         timeout: Timeout to wait for Celery monitoring.
     """
+
     timeout: float = 0.1
 
 
@@ -125,6 +133,7 @@ class Celery(BaseModel):
         monitor: Monitor settings.
         message_maxsize: Maximum allowed message size.
     """
+
     monitor: Monitor = Monitor()
     message_maxsize: int = 16777216
 
@@ -142,12 +151,14 @@ class Controllers(BaseModel):
         list_tasks: Settings for GET /tasks.
         celery: Celery background task settings.
     """
+
     post_task: PostTask = PostTask()
     list_tasks: ListTasks = ListTasks()
     celery: Celery = Celery()
 
+
 class Tes(BaseModel):
-  """TES backend configuration.
+    """TES backend configuration.
 
     Args:
         service_list: List of available TES services.
@@ -156,35 +167,45 @@ class Tes(BaseModel):
         service_list: List of available TES services.
     """
 
-   service_list: List[str] = [
-    "https://csc-tesk-noauth.rahtiapp.fi",
-    "https://funnel.cloud.e-infra.cz/",
-    "https://tesk-eu.hypatia-comp.athenarc.gr",
-    "https://tesk-na.cloud.e-infra.cz",
-    "https://vm4816.kaj.pouta.csc.fi/",
-]
+    service_list: List[str] = [
+        "https://csc-tesk-noauth.rahtiapp.fi",
+        "https://funnel.cloud.e-infra.cz/",
+        "https://tesk-eu.hypatia-comp.athenarc.gr",
+        "https://tesk-na.cloud.e-infra.cz",
+        "https://vm4816.kaj.pouta.csc.fi/",
+    ]
+
 
 class StoreLogs(BaseModel):
-   """Logging configuration.
+    """Logging configuration.
 
     Args:
         execution_trace: Whether to store execution trace logs.
 
     Attributes:
         execution_trace: Whether to store execution trace logs.
     """
-      execution_trace: True
+
+    execution_trace: bool = True
+
 
 class Middlewares(BaseModel):
     middlewares: List[List[str]] = [
         [
-            "pro_tes.plugins.middlewares.task_distribution.distance.TaskDistributionDistance",
-            "pro_tes.plugins.middlewares.task_distribution.random.TaskDistributionRandom",
+            (
+                "pro_tes.plugins.middlewares.task_distribution.distance."
+                "TaskDistributionDistance"
+            ),
+            (
+                "pro_tes.plugins.middlewares.task_distribution.random."
+                "TaskDistributionRandom"
+            ),
         ]
     ]
 
+
 class CustomConfig(BaseModel):
-  """Custom app configuration.
+    """Custom app configuration.
 
     Args:
         controllers: All controller-related config.
@@ -200,9 +221,9 @@ class CustomConfig(BaseModel):
         middlewares: Middleware class paths.
         service_info: Metadata about the service.
     """
+
     controllers: Controllers = Controllers()
     tes: Tes = Tes()
     storeLogs: StoreLogs = StoreLogs()
     middlewares: Middlewares = Middlewares()
     service_info: ServiceInfo
-