litequant 3.0.1__tar.gz → 3.0.2__tar.gz

{litequant-3.0.1 → litequant-3.0.2}/LiteQuantClient.py

@@ -29,6 +29,7 @@ import io
 import json
 import logging
 import os
+import fnmatch
 import time
 import threading
 from typing import Any, Dict, List, Optional, Literal
@@ -75,6 +76,7 @@ class DataTicket:
     expires_at: datetime
     datasets: List[str]
     key_patterns: List[str]
+    category_patterns: Optional[List[str]] = None
     redis_username: Optional[str] = None
     redis_ssl: bool = False
     
@@ -312,7 +314,7 @@ class LiteQuantClient:
         try:
             data = self._api_post_json(
                 "/api/v1/data/ticket/",
-                {"client_version": "3.0.1"},
+                {"client_version": "3.0.2"},
                 timeout=30,
             )
             
@@ -330,6 +332,7 @@ class LiteQuantClient:
                 expires_at=datetime.now() + timedelta(seconds=data["expires_in"]),
                 datasets=scope.get("datasets", []),
                 key_patterns=scope.get("key_patterns", []),
+                category_patterns=scope.get("category_patterns", []),
                 redis_username=redis_username,
                 redis_ssl=redis_ssl,
             )
@@ -650,13 +653,20 @@ class LiteQuantClient:
         all_categories = [str(x) for x in categories]
         
         # 过滤：只保留授权数据集的类别
-        if self._ticket and self._ticket.datasets:
+        if self._ticket and self._ticket.category_patterns:
+            authorized_patterns = [str(x) for x in self._ticket.category_patterns]
+            filtered_categories = [
+                cat for cat in all_categories
+                if any(fnmatch.fnmatchcase(cat, pattern) for pattern in authorized_patterns)
+            ]
+            self.logger.debug("??????: %s -> %s", len(all_categories), len(filtered_categories))
+        elif self._ticket and self._ticket.datasets:
             authorized_datasets = set(self._ticket.datasets)
             filtered_categories = [
                 cat for cat in all_categories
                 if any(cat.startswith(f"{ds}#") for ds in authorized_datasets)
             ]
-            self.logger.debug("过滤类别: %s -> %s", len(all_categories), len(filtered_categories))
+            self.logger.debug("??????: %s -> %s", len(all_categories), len(filtered_categories))
         else:
             filtered_categories = all_categories
         

{litequant-3.0.1 → litequant-3.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: litequant
-Version: 3.0.1
+Version: 3.0.2
 Summary: LiteQuant Python client SDK
 Author: LiteQuant Team
 License: MIT

{litequant-3.0.1 → litequant-3.0.2}/__init__.py

@@ -26,7 +26,7 @@ from .exceptions import (
     APIError,
 )
 
-__version__ = "3.0.1"
+__version__ = "3.0.2"
 __all__ = [
     # Client
     "LiteQuantClient",

{litequant-3.0.1 → litequant-3.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "litequant"
-version = "3.0.1"
+version = "3.0.2"
 description = "LiteQuant Python client SDK"
 readme = "README.md"
 license = {text = "MIT"}