fix: correcting bug and isort

Author: Safenein

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "moodle-painkillers"
-version = "1.0.6"
+version = "1.0.7"
 description = "Signs you up on Moodle so you get paid."
 readme = "README.md"
 requires-python = ">=3.12"

src/moodle_painkillers/__init__.py

@@ -1,14 +1,15 @@
 import argparse
 import logging
 import os
+from ast import ParamSpec, TypeVar
 from dataclasses import dataclass
-from typing import Any, Callable
+from typing import Callable
 
 import bs4
 import requests as rq
 
-from .notifications import send_notification
 from .moodle_authenticate import MoodleAuthenticatedSession
+from .notifications import send_notification
 
 log = logging.getLogger(__name__)
 
@@ -153,7 +154,7 @@ def parse_args():
     )
 
 
-def notify_on_fail(func: Callable[[Any], Any]):
+def notify_on_fail[R](func: Callable[..., R]) -> Callable[..., R]:
     """
     Decorator to send a notification in case of failure.
 
@@ -164,7 +165,7 @@ def notify_on_fail(func: Callable[[Any], Any]):
         callable: The decorated function.
     """
 
-    def wrapper(*args, **kwargs):
+    def wrapper(*args: object, **kwargs: object):
         try:
             return func(*args, **kwargs)
         except Exception as e:
@@ -212,11 +213,13 @@ def main() -> None:
 
     log.info("Presence registration process completed successfully")
     _ = send_notification(
-        "Sent presence status!", discord_webhook=args.discord_webhook
+        "Sent presence status!",
+        title=NOTIFICATION_TITLE,
+        discord_webhook=args.discord_webhook,
     )
 
     return
 
 
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     main()

src/moodle_painkillers/notifications/linux.py

@@ -1,5 +1,5 @@
-import subprocess
 import shutil
+import subprocess
 from logging import getLogger
 
 log = getLogger(__name__)

tests/test___init__.py

@@ -5,12 +5,8 @@
 import pytest
 import requests as rq
 
-from moodle_painkillers import (
-    main,
-    notify_on_fail,
-    parse_args,
-    register_presence_status,
-)
+from moodle_painkillers import (main, notify_on_fail, parse_args,
+                                register_presence_status)
 
 
 class TestRegisterPresenceStatus:
@@ -244,7 +240,9 @@ def failing_func():
         assert excinfo.value == test_exception
 
         # Verify send_notification was called with the exception message
-        mock_send_notification.assert_called_once_with('test error', title='Moodle Presence Registration')
+        mock_send_notification.assert_called_once_with(
+            "test error", title="Moodle Presence Registration"
+        )
 
     @patch("moodle_painkillers.send_notification")
     def test_with_arguments(self, mock_send_notification):
@@ -294,7 +292,9 @@ def test_main_successful_workflow(
         mock_session_class.assert_called_once_with("test_user", "test_pass")
         mock_register_presence.assert_called_once_with(mock_session)
         mock_send_notification.assert_called_once_with(
-            "Sent presence status!", discord_webhook="test_webhook"
+            "Sent presence status!",
+            title="Moodle Presence Registration",
+            discord_webhook="test_webhook",
         )
 
     @patch("moodle_painkillers.send_notification")
@@ -324,7 +324,9 @@ def test_main_without_webhook(
 
         # Verify notification called with None webhook
         mock_send_notification.assert_called_once_with(
-            "Sent presence status!", discord_webhook=None
+            "Sent presence status!",
+            title="Moodle Presence Registration",
+            discord_webhook=None,
         )
 
     @patch("moodle_painkillers.register_presence_status")

tests/test_moodle_authenticate.py

@@ -2,10 +2,8 @@
 import pytest
 import requests
 
-from moodle_painkillers.moodle_authenticate import (
-    MoodleAuthenticatedSession,
-    get_hidden_input_value,
-)
+from moodle_painkillers.moodle_authenticate import (MoodleAuthenticatedSession,
+                                                    get_hidden_input_value)
 
 
 class TestGetHiddenInputValue: