Refactor auth submodule

pyown/auth/__init__.py

@@ -1 +1,3 @@
+from .enum import *
+from .hmac import *
 from .open import *

pyown/auth/enum.py

@@ -0,0 +1,15 @@
+from enum import IntEnum
+
+from ..messages import GenericMessage
+
+__all__ = [
+    "AuthAlgorithm",
+]
+
+
+class AuthAlgorithm(IntEnum):
+    SHA1 = 1
+    SHA256 = 2
+
+    def to_message(self) -> GenericMessage:
+        return GenericMessage(["98", str(self.value)])

pyown/auth/hmac.py

@@ -1,12 +1,10 @@
-from enum import IntEnum
 from hashlib import sha1, sha256
 from hmac import HMAC, compare_digest
 from secrets import token_bytes
 
-from ..messages.base import GenericMessage
+from .enum import AuthAlgorithm
 
 __all__ = [
-    "AuthAlgorithm",
     "client_hmac",
     "server_hmac",
     "compare_hmac",
@@ -15,14 +13,6 @@
 ]
 
 
-class AuthAlgorithm(IntEnum):
-    SHA1 = 1
-    SHA256 = 2
-
-    def to_message(self) -> GenericMessage:
-        return GenericMessage(["98", str(self.value)])
-
-
 def client_hmac(
         server_key: str,
         client_key: str,