pc

Author: hectorzin

.github/workflows/pre-commit.yml

@@ -19,6 +19,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          python -m pip install types-PyYAML types-requests
           python -m pip install --upgrade pip
           python -m pip install pre-commit
           pre-commit install-hooks

config.py

@@ -3,6 +3,7 @@
 import requests
 
 from datetime import datetime, timedelta
+from typing import Dict, Any
 
 # Rutas a los archivos de configuración
 CONFIG_PATH = "data/config.yaml"
@@ -15,7 +16,7 @@
 domain_percentage_table = {}
 all_users_configurations: dict[str, dict] = {}
 
-config_data = {
+config_data: Dict[str, Any] = {
     # Telegram
     "BOT_TOKEN": "",
     "DELETE_MESSAGES": "",

handlers/patterns.py

@@ -1,7 +1,14 @@
 from handlers.base_handler import PATTERN_URL_QUERY
-from typing import Dict, Optional
+from typing import Dict
 
-PATTERNS: Dict[str, Dict[str, Optional[str]]] = {
+from typing import TypedDict
+
+class PatternConfig(TypedDict):
+    pattern: str
+    format_template: str
+    affiliate_tag: str
+
+PATTERNS: Dict[str, PatternConfig] = {
     "amazon": {
         "pattern": (
             r"(https?://(?:www\.)?(?:amazon\.[a-z]{2,3}(?:\.[a-z]{2})?|amzn\.to|amzn\.eu)"