terrakio-core 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

terrakio_core/user_management.py

@@ -1,227 +0,0 @@
-import requests
-from typing import Dict, Any, List, Optional
-from .exceptions import APIError, ConfigurationError
-
-class UserManagement:
-    def __init__(self, api_url: str, api_key: str, verify: bool = True, timeout: int = 60):
-        """
-        Initialize the User Management client.
-        
-        Args:
-            api_url: API base URL
-            api_key: API key for authentication
-            verify: Verify SSL certificates
-            timeout: Request timeout in seconds
-        """
-        self.api_url = api_url.rstrip('/')
-        self.api_key = api_key
-        self.verify = verify
-        self.timeout = timeout
-        self.session = requests.Session()
-        self.session.headers.update({
-            'x-api-key': self.api_key,
-            'Content-Type': 'application/json'
-        })
-
-    def get_user_by_id(self, user_id: str) -> Dict[str, Any]:
-        """
-        Retrieve user info by user ID.
-        
-        Args:
-            user_id: User ID to retrieve
-        
-        Returns:
-            User information as a dictionary
-            
-        Raises:
-            APIError: If the API request fails
-        """
-        endpoint = f"{self.api_url}/admin/users/{user_id}"
-        try:
-            response = self.session.get(
-                endpoint, 
-                timeout=self.timeout, 
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")
-
-    def get_user_by_email(self, email: str) -> Dict[str, Any]:
-        """
-        Retrieve user info by email.
-        
-        Args:
-            email: User email to retrieve
-            
-        Returns:
-            User information as a dictionary
-            
-        Raises:
-            APIError: If the API request fails
-        """
-
-        endpoint = f"{self.api_url}/admin/users/email/{email}"
-        try:
-            response = self.session.get(
-                endpoint, 
-                timeout=self.timeout, 
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")
-
-    def edit_user(
-        self,
-        user_id: str,
-        uid: Optional[str] = None,
-        email: Optional[str] = None,
-        role: Optional[str] = None,
-        apiKey: Optional[str] = None,
-        groups: Optional[List[str]] = None,
-        quota: Optional[int] = None
-    ) -> Dict[str, Any]:
-        """
-        Edit user info. Only provided fields will be updated.
-        
-        Args:
-            user_id: User ID (required)
-            uid: New user ID (optional)
-            email: New user email (optional)
-            role: New user role (optional)
-            apiKey: New API key (optional)
-            groups: New list of groups (optional)
-            quota: New quota value (optional)
-            
-        Returns:
-            Updated user information
-            
-        Raises:
-            APIError: If the API request fails
-        """
-        endpoint = f"{self.api_url}/admin/users"
-        payload = {"uid": user_id}
-
-        if uid is not None:
-            payload["uid"] = uid
-        if email is not None:
-            payload["email"] = email
-        if role is not None:
-            payload["role"] = role
-        if apiKey is not None:
-            payload["apiKey"] = apiKey
-        if groups is not None:
-            payload["groups"] = groups
-        if quota is not None:
-            payload["quota"] = quota
-
-        try:
-            response = self.session.patch(
-                endpoint,
-                json=payload,
-                timeout=self.timeout,
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")
-
-    def list_users(self, substring: Optional[str] = None, uid: bool = False) -> List[Dict[str, Any]]:
-        """
-        List users, optionally filtering by a substring.
-        
-        Args:
-            substring: Optional substring to filter users
-            uid: If True, includes the user ID in the response (default: False)
-        
-        Returns:
-            List of users
-            
-        Raises:
-            APIError: If the API request fails
-        """
-        # Use the base API URL instead of hardcoding
-        endpoint = "https://terrakio-server-lark-573248941006.australia-southeast1.run.app/admin/users"
-
-        params = {}
-        if substring:
-            params["substring"] = substring
-        if uid:
-            params["uid"] = "true"
-
-        try:
-            response = self.session.get(
-                endpoint,
-                params=params,
-                timeout=self.timeout,
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")
-
-    def reset_quota(self, email: str, quota: Optional[int] = None) -> Dict[str, Any]:
-        """
-        Reset the quota for a user by email.
-        
-        Args:
-            email: The user's email (required)
-            quota: The new quota value (optional)
-            
-        Returns:
-            API response as a dictionary
-            
-        Raises:
-            APIError: If the API request fails
-        """
-        endpoint = f"{self.api_url}/admin/users/reset_quota/{email}"
-        payload = {"email": email}
-        if quota is not None:
-            payload["quota"] = quota
-        try:
-            response = self.session.patch(
-                endpoint,
-                json=payload,
-                timeout=self.timeout,
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")
-
-    def delete_user(self, uid: str) -> Dict[str, Any]:
-        """
-        Delete a user by UID.
-        
-        Args:
-            uid: The user's UID (required)
-            
-        Returns:
-            API response as a dictionary
-            
-        Raises:
-            APIError: If the API request fails
-        """
-        endpoint = f"{self.api_url}/admin/users/{uid}"
-        try:
-            response = self.session.delete(
-                endpoint,
-                timeout=self.timeout,
-                verify=self.verify
-            )
-            if not response.ok:
-                raise APIError(f"API request failed: {response.status_code} {response.reason}")
-            return response.json()
-        except requests.RequestException as e:
-            raise APIError(f"Request failed: {str(e)}")

terrakio_core-0.2.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-terrakio_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-terrakio_core/auth.py,sha256=Y3X5CcRUO7rAsrv995cSedZFKJAsW6ObDinYCbcQMpc,7605
-terrakio_core/client.py,sha256=A8W-Tun8ZeMnQhr7d6i2ekyl4sylWTVq5amVLUhV6kE,36446
-terrakio_core/config.py,sha256=AwJ1VgR5K7N32XCU5k7_Dp1nIv_FYt8MBonq9yKlGzA,2658
-terrakio_core/dataset_management.py,sha256=hhO35fwStS6HYFQdKP9wkr3DxHgjvpctmIU8UWH6w6U,8742
-terrakio_core/exceptions.py,sha256=9S-I20-QiDRj1qgjFyYUwYM7BLic_bxurcDOIm2Fu_0,410
-terrakio_core/mass_stats.py,sha256=AqYJsd6nqo2BDh4vEPUDgsv4T0UR1_TPDoXa3WO3gTU,9284
-terrakio_core/user_management.py,sha256=MMNWkz0V_9X7ZYjjteuRU4H4W3F16iuQw1dpA2wVTGg,7400
-terrakio_core-0.2.1.dist-info/METADATA,sha256=n1judXAN5fxpx5jDOM6fIVNJXBeGkaghGwwdOuZH-3U,1518
-terrakio_core-0.2.1.dist-info/WHEEL,sha256=QZxptf4Y1BKFRCEDxD4h2V0mBFQOVFLFEpvxHmIs52A,91
-terrakio_core-0.2.1.dist-info/top_level.txt,sha256=5cBj6O7rNWyn97ND4YuvvXm0Crv4RxttT4JZvNdOG6Q,14
-terrakio_core-0.2.1.dist-info/RECORD,,

terrakio_core-0.2.1.dist-info/top_level.txt

@@ -1 +0,0 @@
-terrakio_core