add import statements for the types

Signed-off-by: Soumik Dutta <shalearkane@gmail.com>
rucio · Mar 11, 2023 · 13c5e47 · 13c5e47
1 parent 8994069
commit 13c5e47
Show file tree

Hide file tree

Showing 32 changed files with 137 additions and 5 deletions.
diff --git a/lib/rucio/daemons/abacus/account.py b/lib/rucio/daemons/abacus/account.py
@@ -20,6 +20,7 @@
 import logging
 import threading
 import time
+from typing import TYPE_CHECKING
 
 import rucio.db.sqla.util
 from rucio.common import exception
@@ -28,6 +29,10 @@
 from rucio.core.account_counter import get_updated_account_counters, update_account_counter, fill_account_counter_history_table
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 
 

diff --git a/lib/rucio/daemons/abacus/collection_replica.py b/lib/rucio/daemons/abacus/collection_replica.py
@@ -20,13 +20,18 @@
 import logging
 import threading
 import time
+from typing import TYPE_CHECKING
 
 import rucio.db.sqla.util
 from rucio.common import exception
 from rucio.common.logging import setup_logging
 from rucio.core.replica import get_cleaned_updated_collection_replicas, update_collection_replica
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 
 

diff --git a/lib/rucio/daemons/abacus/rse.py b/lib/rucio/daemons/abacus/rse.py
@@ -20,6 +20,7 @@
 import logging
 import threading
 import time
+from typing import TYPE_CHECKING
 
 import rucio.db.sqla.util
 from rucio.common import exception
@@ -28,6 +29,10 @@
 from rucio.core.rse_counter import get_updated_rse_counters, update_rse_counter, fill_rse_counter_history_table
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 
 

diff --git a/lib/rucio/daemons/atropos/atropos.py b/lib/rucio/daemons/atropos/atropos.py
@@ -37,6 +37,8 @@
 
 if TYPE_CHECKING:
     from rucio.daemons.common import HeartbeatHandler
+    from types import FrameType
+    from typing import Optional
 
 GRACEFUL_STOP = threading.Event()
 

diff --git a/lib/rucio/daemons/automatix/automatix.py b/lib/rucio/daemons/automatix/automatix.py
@@ -40,6 +40,9 @@
 
 
 if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
     from rucio.daemons.common import HeartbeatHandler
 
 METRICS = MetricManager(module=__name__)

diff --git a/lib/rucio/daemons/badreplicas/minos.py b/lib/rucio/daemons/badreplicas/minos.py
@@ -40,6 +40,9 @@
 
 
 if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
     from rucio.daemons.common import HeartbeatHandler
     from rucio.common.types import InternalAccount
 

diff --git a/lib/rucio/daemons/badreplicas/minos_temporary_expiration.py b/lib/rucio/daemons/badreplicas/minos_temporary_expiration.py
@@ -37,6 +37,9 @@
 
 
 if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
     from rucio.daemons.common import HeartbeatHandler
 
 graceful_stop = threading.Event()

diff --git a/lib/rucio/daemons/badreplicas/necromancer.py b/lib/rucio/daemons/badreplicas/necromancer.py
@@ -38,6 +38,9 @@
 from rucio.daemons.common import run_daemon
 
 if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
     from rucio.daemons.common import HeartbeatHandler
 
 graceful_stop = threading.Event()

diff --git a/lib/rucio/daemons/bb8/bb8.py b/lib/rucio/daemons/bb8/bb8.py
@@ -34,6 +34,9 @@
 
 
 if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
     from rucio.daemons.common import HeartbeatHandler
 
 graceful_stop = threading.Event()

diff --git a/lib/rucio/daemons/c3po/c3po.py b/lib/rucio/daemons/c3po/c3po.py
@@ -24,6 +24,7 @@
 from threading import Event, Thread
 from time import sleep
 from uuid import uuid4
+from typing import TYPE_CHECKING
 
 from requests import post
 from requests.auth import HTTPBasicAuth
@@ -41,6 +42,10 @@
 
 from queue import Queue
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 GRACEFUL_STOP = Event()
 
 

diff --git a/lib/rucio/daemons/conveyor/finisher.py b/lib/rucio/daemons/conveyor/finisher.py
@@ -23,6 +23,7 @@
 import os
 import re
 import threading
+from typing import TYPE_CHECKING
 
 from dogpile.cache.api import NoValue
 from sqlalchemy.exc import DatabaseError
@@ -45,6 +46,10 @@
 
 from urllib.parse import urlparse
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 
 region = make_region_memcached(expiration_time=900)

diff --git a/lib/rucio/daemons/conveyor/poller.py b/lib/rucio/daemons/conveyor/poller.py
@@ -26,6 +26,7 @@
 import threading
 import time
 from itertools import groupby
+from typing import TYPE_CHECKING
 
 from requests.exceptions import RequestException
 from configparser import NoOptionError
@@ -45,6 +46,10 @@
 from rucio.transfertool.fts3 import FTS3Transfertool
 from rucio.transfertool.globus import GlobusTransferTool
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 METRICS = MetricManager(module=__name__)
 

diff --git a/lib/rucio/daemons/conveyor/receiver.py b/lib/rucio/daemons/conveyor/receiver.py
@@ -24,6 +24,7 @@
 import threading
 import time
 import traceback
+from typing import TYPE_CHECKING
 
 import stomp
 
@@ -39,6 +40,10 @@
 from rucio.db.sqla.session import transactional_session
 from rucio.transfertool.fts3 import FTS3CompletionMessageTransferStatusReport
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 logging.getLogger("stomp").setLevel(logging.CRITICAL)
 
 METRICS = MetricManager(module=__name__)

diff --git a/lib/rucio/daemons/conveyor/stager.py b/lib/rucio/daemons/conveyor/stager.py
@@ -20,6 +20,7 @@
 import functools
 import logging
 import threading
+from typing import TYPE_CHECKING
 
 from configparser import NoOptionError
 
@@ -34,6 +35,10 @@
 from rucio.db.sqla.constants import RequestType
 from rucio.transfertool.fts3 import FTS3Transfertool
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/conveyor/submitter.py b/lib/rucio/daemons/conveyor/submitter.py
@@ -20,6 +20,7 @@
 import functools
 import logging
 import threading
+from typing import TYPE_CHECKING
 
 from configparser import NoOptionError
 
@@ -37,6 +38,10 @@
 from rucio.transfertool.fts3 import FTS3Transfertool
 from rucio.transfertool.globus import GlobusTransferTool
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/conveyor/throttler.py b/lib/rucio/daemons/conveyor/throttler.py
@@ -21,6 +21,7 @@
 import threading
 import traceback
 from collections import defaultdict
+from typing import TYPE_CHECKING
 
 from sqlalchemy import null
 
@@ -35,6 +36,10 @@
 from rucio.daemons.common import run_daemon
 from rucio.db.sqla.constants import RequestState, TransferLimitDirection
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 METRICS = MetricManager(module=__name__)
 

diff --git a/lib/rucio/daemons/follower/follower.py b/lib/rucio/daemons/follower/follower.py
@@ -18,6 +18,7 @@
 import socket
 import threading
 import time
+from typing import TYPE_CHECKING
 
 import rucio.db.sqla.util
 from rucio.common import exception
@@ -26,6 +27,10 @@
 from rucio.core.did import create_reports
 from rucio.core.heartbeat import live, die, sanity_check
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 graceful_stop = threading.Event()
 
 

diff --git a/lib/rucio/daemons/hermes/hermes.py b/lib/rucio/daemons/hermes/hermes.py
@@ -25,6 +25,7 @@
 import socket
 import threading
 import traceback
+from typing import TYPE_CHECKING
 
 from email.mime.text import MIMEText
 from typing import List, Dict, Any
@@ -40,6 +41,10 @@
 from rucio.core.monitor import MetricManager
 from rucio.daemons.common import HeartbeatHandler, run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 logging.getLogger('requests').setLevel(logging.CRITICAL)
 logging.getLogger('stomp').setLevel(logging.CRITICAL)
 

diff --git a/lib/rucio/daemons/hermes/hermes2.py b/lib/rucio/daemons/hermes/hermes2.py
@@ -50,7 +50,9 @@
 from rucio.daemons.common import run_daemon
 
 if TYPE_CHECKING:
-    from typing import Callable, List, Dict
+    from types import FrameType
+    from typing import Callable, Optional, List, Dict
+
     from rucio.daemons.common import HeartbeatHandler
 
 logging.getLogger("requests").setLevel(logging.CRITICAL)

diff --git a/lib/rucio/daemons/judge/cleaner.py b/lib/rucio/daemons/judge/cleaner.py
@@ -24,6 +24,7 @@
 from datetime import datetime, timedelta
 from random import randint
 from re import match
+from typing import TYPE_CHECKING
 
 from sqlalchemy.exc import DatabaseError
 
@@ -36,6 +37,10 @@
 from rucio.daemons.common import run_daemon
 from rucio.db.sqla.util import get_db_time
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/judge/evaluator.py b/lib/rucio/daemons/judge/evaluator.py
@@ -25,6 +25,7 @@
 from datetime import datetime, timedelta
 from random import randint
 from re import match
+from typing import TYPE_CHECKING
 
 from sqlalchemy.exc import DatabaseError
 from sqlalchemy.orm.exc import FlushError
@@ -37,6 +38,10 @@
 from rucio.core.rule import re_evaluate_did, get_updated_dids, delete_updated_did
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/judge/injector.py b/lib/rucio/daemons/judge/injector.py
@@ -24,6 +24,7 @@
 from datetime import datetime, timedelta
 from random import randint
 from re import match
+from typing import TYPE_CHECKING
 
 from sqlalchemy.exc import DatabaseError
 
@@ -35,6 +36,10 @@
 from rucio.core.rule import inject_rule, get_injected_rules, update_rule
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/judge/repairer.py b/lib/rucio/daemons/judge/repairer.py
@@ -25,6 +25,7 @@
 from datetime import datetime, timedelta
 from random import randint
 from re import match
+from typing import TYPE_CHECKING
 
 from sqlalchemy.exc import DatabaseError
 
@@ -36,6 +37,10 @@
 from rucio.core.rule import repair_rule, get_stuck_rules
 from rucio.daemons.common import run_daemon
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()
 

diff --git a/lib/rucio/daemons/oauthmanager/oauthmanager.py b/lib/rucio/daemons/oauthmanager/oauthmanager.py
@@ -31,6 +31,7 @@
 import threading
 import traceback
 from re import match
+from typing import TYPE_CHECKING
 
 from sqlalchemy.exc import DatabaseError
 
@@ -44,6 +45,10 @@
 from rucio.daemons.common import run_daemon
 from rucio.daemons.common import HeartbeatHandler
 
+if TYPE_CHECKING:
+    from types import FrameType
+    from typing import Optional
+
 METRICS = MetricManager(module=__name__)
 graceful_stop = threading.Event()