refactor: sorted imports

Author: trbtm

postgresql_watcher/watcher.py

@@ -1,10 +1,11 @@
-from typing import Optional, Callable
-from psycopg2 import connect, extensions
+from logging import Logger, getLogger
 from multiprocessing import Process, Pipe
 from multiprocessing.connection import Connection
-import time
-from select import select
-from logging import Logger, getLogger
+from time import sleep, time
+from typing import Optional, Callable
+
+from psycopg2 import connect, extensions
+
 from .casbin_channel_subscription import (
     casbin_channel_subscription,
     _ChannelSubscriptionMessage,
@@ -115,7 +116,7 @@ def start(self):
                     message = int(self.parent_conn.recv())
                     if message == _ChannelSubscriptionMessage.IS_READY:
                         break
-                time.sleep(1 / 1000)  # wait for 1 ms
+                sleep(1 / 1000)  # wait for 1 ms
 
     def _cleanup_connections_and_processes(self) -> None:
         # Clean up potentially existing Connections and Processes
@@ -155,9 +156,7 @@ def update(self) -> None:
         # Can only receive notifications when not in transaction, set this for easier usage
         conn.set_isolation_level(extensions.ISOLATION_LEVEL_AUTOCOMMIT)
         curs = conn.cursor()
-        curs.execute(
-            f"NOTIFY {self.channel_name},'casbin policy update at {time.time()}'"
-        )
+        curs.execute(f"NOTIFY {self.channel_name},'casbin policy update at {time()}'")
         conn.close()
 
     def should_reload(self) -> bool:

tests/test_postgresql_watcher.py

@@ -1,13 +1,13 @@
-import sys
-import unittest
-from unittest.mock import MagicMock
-from multiprocessing.connection import Pipe
+from logging import DEBUG, getLogger, StreamHandler
+from multiprocessing import connection, context
 from time import sleep
-import logging
+from unittest import TestCase, main
+from unittest.mock import MagicMock
+import sys
 
 from postgresql_watcher import PostgresqlWatcher
 from postgresql_watcher.casbin_channel_subscription import CASBIN_CHANNEL_SELECT_TIMEOUT
-from multiprocessing import connection, context
+
 
 # Warning!!! , Please setup yourself config
 HOST = "127.0.0.1"
@@ -16,9 +16,9 @@
 PASSWORD = "123456"
 DBNAME = "postgres"
 
-logger = logging.getLogger()
-logger.level = logging.DEBUG
-stream_handler = logging.StreamHandler(sys.stdout)
+logger = getLogger()
+logger.level = DEBUG
+stream_handler = StreamHandler(sys.stdout)
 logger.addHandler(stream_handler)
 
 
@@ -43,7 +43,7 @@ def get_watcher(channel_name):
     _winapi = None
 
 
-class TestConfig(unittest.TestCase):
+class TestConfig(TestCase):
     def test_pg_watcher_init(self):
         pg_watcher = get_watcher("test_pg_watcher_init")
         if _winapi:
@@ -104,4 +104,4 @@ def test_update_handler_not_called(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    main()