Jedd Morgan
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
from typing import List, Tuple
|
|
|
|
from specklepy.core.api.client import SpeckleClient
|
|
from specklepy.core.api.models.current import Version
|
|
|
|
from .account_manager import _client_cache
|
|
from .misc import format_relative_time
|
|
|
|
|
|
def get_versions_for_model(
|
|
account_id: str, project_id: str, model_id: str
|
|
) -> List[Tuple[str, str, str]]:
|
|
"""
|
|
fetches versions for a given model from the Speckle server
|
|
"""
|
|
try:
|
|
# Validate inputs
|
|
if not account_id or not project_id or not model_id:
|
|
print(
|
|
f"Error: Invalid inputs - account_id: {account_id}, project_id: {project_id}, model_id: {model_id}"
|
|
)
|
|
return []
|
|
|
|
client: SpeckleClient = _client_cache.get_client(account_id)
|
|
|
|
# Get versions
|
|
versions = client.version.get_versions(
|
|
project_id=project_id, model_id=model_id, limit=10
|
|
)
|
|
versions_list: List[Tuple[str, str, str]] = []
|
|
for version in versions.items:
|
|
if version.referenced_object != "":
|
|
versions_list.append(
|
|
(
|
|
version.id,
|
|
version.message
|
|
if version.message is not None
|
|
else "No message",
|
|
format_relative_time(version.created_at),
|
|
)
|
|
)
|
|
return versions_list
|
|
|
|
except Exception as e:
|
|
print(f"Error fetching versions: {str(e)}")
|
|
# Clear cache on error to prevent stale clients
|
|
_client_cache.clear()
|
|
return []
|
|
|
|
|
|
def get_latest_version(
|
|
account_id: str, project_id: str, model_id: str
|
|
) -> Tuple[str, str, str]:
|
|
try:
|
|
# Validate inputs
|
|
if not account_id or not project_id or not model_id:
|
|
print(
|
|
f"Error: Invalid inputs - account_id: {account_id}, project_id: {project_id}, model_id: {model_id}"
|
|
)
|
|
return ("", "", "")
|
|
|
|
# Get cached client
|
|
client: SpeckleClient = _client_cache.get_client(account_id)
|
|
|
|
# Get versions (limit to 1 since we only need the latest)
|
|
versions: List[Version] = client.version.get_versions(
|
|
project_id=project_id, model_id=model_id, limit=1
|
|
).items
|
|
|
|
if not versions:
|
|
print(f"Error: No versions found for model_id: {model_id}")
|
|
return ("", "", "")
|
|
|
|
latest = versions[0]
|
|
return (
|
|
latest.id,
|
|
latest.message if latest.message is not None else "No message",
|
|
format_relative_time(latest.created_at),
|
|
)
|
|
|
|
except Exception as e:
|
|
print(f"Error fetching latest version: {str(e)}")
|
|
# Clear cache on error to prevent stale clients
|
|
_client_cache.clear()
|
|
return ("", "", "")
|