adss 1.0__py3-none-any.whl → 1.2__py3-none-any.whl

adss/models/user.py

@@ -0,0 +1,123 @@
+"""
+User-related data models for the Astronomy TAP Client.
+"""
+from dataclasses import dataclass, field
+from typing import List, Dict, Optional, Any
+from datetime import datetime
+
+from adss.utils import parse_datetime
+
+
+@dataclass
+class SchemaPermission:
+    """Schema-level permission."""
+    schema_name: str
+    permission: str  # 'read', 'write', or 'all'
+
+
+@dataclass
+class TablePermission:
+    """Table-level permission."""
+    schema_name: str
+    table_name: str
+    permission: str  # 'read', 'write', or 'all'
+
+
+@dataclass
+class RolePermissions:
+    """Permissions associated with a role."""
+    schema_permissions: List[SchemaPermission] = field(default_factory=list)
+    table_permissions: List[TablePermission] = field(default_factory=list)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'RolePermissions':
+        """Create a RolePermissions object from a dictionary."""
+        schema_perms = [
+            SchemaPermission(**p) 
+            for p in data.get('schema_permissions', [])
+        ]
+        
+        table_perms = [
+            TablePermission(**p) 
+            for p in data.get('table_permissions', [])
+        ]
+        
+        return cls(
+            schema_permissions=schema_perms,
+            table_permissions=table_perms
+        )
+
+
+@dataclass
+class Role:
+    """User role with associated permissions."""
+    id: int
+    name: str
+    description: Optional[str] = None
+    permissions: Optional[RolePermissions] = None
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'Role':
+        """Create a Role object from a dictionary."""
+        role_id = data.get('id')
+        name = data.get('name')
+        description = data.get('description')
+        
+        permissions_data = data.get('permissions')
+        permissions = None
+        if permissions_data:
+            permissions = RolePermissions.from_dict(permissions_data)
+        
+        return cls(
+            id=role_id,
+            name=name,
+            description=description,
+            permissions=permissions
+        )
+
+
+@dataclass
+class User:
+    """User model with authentication and role information."""
+    id: str
+    username: str
+    email: str
+    full_name: Optional[str] = None
+    is_active: bool = True
+    is_staff: bool = False
+    is_superuser: bool = False
+    created_at: Optional[datetime] = None
+    last_login: Optional[datetime] = None
+    roles: List[Role] = field(default_factory=list)
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'User':
+        """Create a User object from a dictionary."""
+        user_id = data.get('id')
+        username = data.get('username')
+        email = data.get('email')
+        full_name = data.get('full_name')
+        is_active = data.get('is_active', True)
+        is_staff = data.get('is_staff', False)
+        is_superuser = data.get('is_superuser', False)
+        
+        created_at = parse_datetime(data.get('created_at'))
+        last_login = parse_datetime(data.get('last_login'))
+        
+        roles = [
+            Role.from_dict(role_data) 
+            for role_data in data.get('roles', [])
+        ]
+        
+        return cls(
+            id=user_id,
+            username=username,
+            email=email,
+            full_name=full_name,
+            is_active=is_active,
+            is_staff=is_staff,
+            is_superuser=is_superuser,
+            created_at=created_at,
+            last_login=last_login,
+            roles=roles
+        )

adss/utils.py

@@ -0,0 +1,107 @@
+"""
+Utility functions for the Astronomy TAP Client.
+"""
+import json
+import pandas as pd
+import pyarrow.parquet as pq
+import io
+from datetime import datetime
+from typing import Dict, Any, Optional, Union, List
+
+from adss.exceptions import (
+    ADSSClientError, AuthenticationError, PermissionDeniedError,
+    ResourceNotFoundError, QueryExecutionError, ServerError
+)
+
+
+def handle_response_errors(response):
+    """Handles HTTP response errors and raises appropriate exceptions."""
+    if 200 <= response.status_code < 300:
+        return response
+    
+    try:
+        error_data = response.json()
+        error_message = error_data.get('detail', str(error_data))
+    except Exception:
+        error_message = response.text or f"HTTP Error {response.status_code}"
+    
+    # For 401 errors, check if it's an anonymous query attempt
+    if response.status_code == 401:
+        # If the error mentions "authentication required for protected schemas"
+        # or similar, it's likely an anonymous query trying to access restricted data
+        if "protected" in error_message.lower() or "requires authentication" in error_message.lower():
+            raise PermissionDeniedError(f"This query requires authentication: {error_message}", response)
+        else:
+            raise AuthenticationError(error_message, response)
+    elif response.status_code == 403:
+        raise PermissionDeniedError(error_message, response)
+    elif response.status_code == 404:
+        raise ResourceNotFoundError(error_message, response)
+    elif response.status_code >= 500:
+        raise ServerError(f"Server error: {error_message}", response)
+    else:
+        raise ADSSClientError(error_message, response)
+
+def parse_datetime(dt_str: Optional[str]) -> Optional[datetime]:
+    """
+    Parses a datetime string into a datetime object.
+    """
+    if not dt_str:
+        return None
+    
+    try:
+        return datetime.fromisoformat(dt_str.replace('Z', '+00:00'))
+    except (ValueError, TypeError):
+        return None
+
+
+def parquet_to_dataframe(parquet_data: bytes) -> pd.DataFrame:
+    """
+    Converts Parquet bytes to a pandas DataFrame.
+    """
+    try:
+        buffer = io.BytesIO(parquet_data)
+        table = pq.read_table(buffer)
+        return table.to_pandas()
+    except Exception as e:
+        raise ADSSClientError(f"Failed to convert Parquet data to DataFrame: {str(e)}")
+
+
+def format_table_name(schema: str, table: str) -> str:
+    """
+    Formats a schema and table name into a fully qualified table name.
+    """
+    return f"{schema}.{table}"
+
+
+def prepare_query_params(params: Dict[str, Any]) -> Dict[str, str]:
+    """
+    Prepares query parameters for API requests, handling different types.
+    """
+    processed_params = {}
+    
+    for key, value in params.items():
+        if value is None:
+            continue
+        elif isinstance(value, bool):
+            processed_params[key] = str(value).lower()
+        elif isinstance(value, (list, dict)):
+            processed_params[key] = json.dumps(value)
+        else:
+            processed_params[key] = str(value)
+    
+    return processed_params
+
+
+def format_permission(permission_type: str) -> str:
+    """
+    Validates and formats a permission type (read, write, all).
+    """
+    valid_permissions = {'read', 'write', 'all'}
+    permission = permission_type.lower()
+    
+    if permission not in valid_permissions:
+        raise ValueError(f"Invalid permission type: {permission_type}. "
+                         f"Must be one of: {', '.join(valid_permissions)}")
+    
+    return permission

{adss-1.0.dist-info → adss-1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: adss
-Version: 1.0
+Version: 1.2
 Summary: Astronomical Data Smart System
 Home-page: https://github.com/schwarzam/adss
 Author: Gustavo Schwarz

adss-1.2.dist-info/RECORD

@@ -0,0 +1,30 @@
+adss/__init__.py,sha256=TqNjMZ7PuNXc-FnOlVKSNHSAp9ZqcLMdHDuRYt47_Y8,834
+adss/adss_manager.py,sha256=vaS6y1IycVW8AjpAQeG58VwjCDLyMBUylJzYSeg6D9o,2068
+adss/auth.py,sha256=M8GAN8pdfmJfsEOqv_EW2ru0uP0H1JtK7IfBDrIZMyg,4025
+adss/client.py,sha256=Ojc6jqZtCBMwRoTGuBiF0SEDkdWs_-xbGwyk03nHkr8,29613
+adss/exceptions.py,sha256=YeN-xRHvlSmwyS8ni2jOEhhgZK9J1jsG11pOedy3Gfg,1482
+adss/table.py,sha256=Ua663njPk2sg8BtQPo1wZ-V09YvnjrEyIb_SmBhdOYY,13383
+adss/utils.py,sha256=0RISndgXnwVy8cLMFa4Mm7CfGqwGdX-X-HZ0NmPDVD0,3558
+adss/variables.py,sha256=kmbwxJBDC97yKakrnBvONRh1FVvSXU4YKqnjExAU2ZA,51
+adss/endpoints/__init__.py,sha256=Pr29901fT8ClCS2GasTjTiBNyn7DfVfxILpYDFsMvPA,488
+adss/endpoints/admin.py,sha256=S6ZrkeA_Lh_LCpF1NHyfMKqjbIiylYXUSV65H_WKg1U,16391
+adss/endpoints/images.py,sha256=ItAiBss_jQvWQWRUvy0c9Cjn1r9lDR8eOPauqOcPcZ8,35777
+adss/endpoints/metadata.py,sha256=RPrRP6Uz6-uPMIcntMgfss9vAd5iN7JXjZbF8SW0EYg,8238
+adss/endpoints/queries.py,sha256=5BONw_IcGORMPNe-5J6BpoFY6z7lKcktEVhqZ9j17_8,17286
+adss/endpoints/users.py,sha256=6Abkl3c3_YKdMYR_JWI-uL9HTHxcjlIOnE29GyN5_QE,10811
+adss/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+adss/executors/async_query.py,sha256=qmSm-Ex8ZJSjn8lmm_F5BufVpDV-w6vOqXie7krCp7k,3977
+adss/executors/sync_query.py,sha256=3e-ALG3GnA906o_oefci5XHNcdnoPWuc67ml-YATMKE,1243
+adss/models/__init__.py,sha256=ADWVaGy4dkpEMH3iS_6EnRSBlEgoM5Vy9zORQr-UG6w,404
+adss/models/metadata.py,sha256=6fdH_0BenVRmeXkkKbsG2B68O-N2FXTTRgxsEhAHRoU,4058
+adss/models/query.py,sha256=Af-iojZb-nO6qj-yMT_PlNM7Hip6EwBfNeaQPMJPNM0,4293
+adss/models/user.py,sha256=5qVT5qOktokmVLkGszPGCTZWv0wC-7aBMvJ8EeBOqdw,3493
+adss/operations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+adss/operations/cone_search.py,sha256=qfdFA2TGqnzuggz4nep21_y4LgmHP4ZMpVupxn87dB0,706
+adss/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+adss/utils/format_table.py,sha256=UYCQ6Xum3dPHrh0cAh_KCj6vHShAvdHlV0rtIv7J09Q,3695
+adss-1.2.dist-info/LICENSE,sha256=1aYqcyqjrdNXY9hqgZkCWprcoA112oKvdrfPyvMYPTc,1468
+adss-1.2.dist-info/METADATA,sha256=l8AVLDMarCFGUu3oHE3IWCLCBn88wFPxVioKYiIiIoQ,378
+adss-1.2.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+adss-1.2.dist-info/top_level.txt,sha256=ebD44L3R0PEvEFoRCJ-RjTIsQ9Yjpo2aAYC1BMtueLg,5
+adss-1.2.dist-info/RECORD,,

{adss-1.0.dist-info → adss-1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (71.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

adss-1.0.dist-info/RECORD

@@ -1,16 +0,0 @@
-adss/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adss/adss_manager.py,sha256=vaS6y1IycVW8AjpAQeG58VwjCDLyMBUylJzYSeg6D9o,2068
-adss/table.py,sha256=Ua663njPk2sg8BtQPo1wZ-V09YvnjrEyIb_SmBhdOYY,13383
-adss/variables.py,sha256=kmbwxJBDC97yKakrnBvONRh1FVvSXU4YKqnjExAU2ZA,51
-adss/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adss/executors/async_query.py,sha256=qmSm-Ex8ZJSjn8lmm_F5BufVpDV-w6vOqXie7krCp7k,3977
-adss/executors/sync_query.py,sha256=3e-ALG3GnA906o_oefci5XHNcdnoPWuc67ml-YATMKE,1243
-adss/operations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adss/operations/cone_search.py,sha256=qfdFA2TGqnzuggz4nep21_y4LgmHP4ZMpVupxn87dB0,706
-adss/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adss/utils/format_table.py,sha256=UYCQ6Xum3dPHrh0cAh_KCj6vHShAvdHlV0rtIv7J09Q,3695
-adss-1.0.dist-info/LICENSE,sha256=1aYqcyqjrdNXY9hqgZkCWprcoA112oKvdrfPyvMYPTc,1468
-adss-1.0.dist-info/METADATA,sha256=mXmdv0NUDgCapF58GJOkJj5_bCU5rjHgCjYI1cCDY7Q,378
-adss-1.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-adss-1.0.dist-info/top_level.txt,sha256=ebD44L3R0PEvEFoRCJ-RjTIsQ9Yjpo2aAYC1BMtueLg,5
-adss-1.0.dist-info/RECORD,,