django-cfg 1.4.20__py3-none-any.whl → 1.4.23__py3-none-any.whl

django_cfg/apps/agents/managers/registry.py

@@ -2,22 +2,23 @@
 Custom managers for agent registry models.
 """
 
+from typing import Any, Dict, List, Optional
+
 from django.db import models
 from django.utils import timezone
-from typing import Optional, Dict, Any, List
 
 
 class AgentDefinitionQuerySet(models.QuerySet):
     """Custom queryset for AgentDefinition."""
-    
+
     def active(self):
         """Get active agent definitions."""
         return self.filter(is_active=True)
-    
+
     def public(self):
         """Get public agent definitions."""
         return self.filter(is_public=True)
-    
+
     def for_user(self, user):
         """Get agent definitions available for specific user."""
         return self.filter(
@@ -26,29 +27,29 @@ class AgentDefinitionQuerySet(models.QuerySet):
             models.Q(allowed_users=user) |
             models.Q(allowed_groups__in=user.groups.all())
         ).distinct()
-    
+
     def by_category(self, category: str):
         """Filter by category."""
         return self.filter(category=category)
-    
+
     def by_creator(self, user):
         """Get agents created by specific user."""
         return self.filter(created_by=user)
-    
+
     def popular(self, min_usage: int = 10):
         """Get popular agents."""
         return self.filter(usage_count__gte=min_usage)
-    
+
     def recent(self, days: int = 7):
         """Get recently created agents."""
         from datetime import timedelta
         cutoff = timezone.now() - timedelta(days=days)
         return self.filter(created_at__gte=cutoff)
-    
+
     def with_executions(self):
         """Include execution statistics."""
         return self.prefetch_related('executions')
-    
+
     def search(self, query: str):
         """Search agents by name, description, or instructions."""
         return self.filter(
@@ -61,46 +62,46 @@ class AgentDefinitionQuerySet(models.QuerySet):
 
 class AgentDefinitionManager(models.Manager):
     """Custom manager for AgentDefinition."""
-    
+
     def get_queryset(self):
         return AgentDefinitionQuerySet(self.model, using=self._db)
-    
+
     def active(self):
         """Get active agents."""
         return self.get_queryset().active()
-    
+
     def public(self):
         """Get public agents."""
         return self.get_queryset().public()
-    
+
     def for_user(self, user):
         """Get agents available for user."""
         return self.get_queryset().for_user(user)
-    
+
     def by_category(self, category: str):
         """Get agents by category."""
         return self.get_queryset().by_category(category)
-    
+
     def by_creator(self, user):
         """Get agents by creator."""
         return self.get_queryset().by_creator(user)
-    
+
     def popular(self, min_usage: int = 10):
         """Get popular agents."""
         return self.get_queryset().popular(min_usage)
-    
+
     def recent(self, days: int = 7):
         """Get recent agents."""
         return self.get_queryset().recent(days)
-    
+
     def with_executions(self):
         """Get agents with executions."""
         return self.get_queryset().with_executions()
-    
+
     def search(self, query: str):
         """Search agents."""
         return self.get_queryset().search(query)
-    
+
     def create_agent(
         self,
         name: str,
@@ -123,7 +124,7 @@ class AgentDefinitionManager(models.Manager):
             created_by=created_by,
             **kwargs
         )
-    
+
     def get_categories(self) -> List[str]:
         """Get all unique categories."""
         return list(
@@ -133,11 +134,11 @@ class AgentDefinitionManager(models.Manager):
             .exclude(category='')
             .order_by('category')
         )
-    
+
     def get_stats(self) -> Dict[str, Any]:
         """Get overall agent statistics."""
         queryset = self.get_queryset()
-        
+
         return {
             'total': queryset.count(),
             'active': queryset.active().count(),
@@ -154,25 +155,25 @@ class AgentDefinitionManager(models.Manager):
 
 class AgentTemplateQuerySet(models.QuerySet):
     """Custom queryset for AgentTemplate."""
-    
+
     def active(self):
         """Get active templates."""
         return self.filter(is_active=True)
-    
+
     def by_category(self, category: str):
         """Filter by category."""
         return self.filter(category=category)
-    
+
     def by_creator(self, user):
         """Get templates created by specific user."""
         return self.filter(created_by=user)
-    
+
     def recent(self, days: int = 7):
         """Get recently created templates."""
         from datetime import timedelta
         cutoff = timezone.now() - timedelta(days=days)
         return self.filter(created_at__gte=cutoff)
-    
+
     def search(self, query: str):
         """Search templates by name or description."""
         return self.filter(
@@ -183,30 +184,30 @@ class AgentTemplateQuerySet(models.QuerySet):
 
 class AgentTemplateManager(models.Manager):
     """Custom manager for AgentTemplate."""
-    
+
     def get_queryset(self):
         return AgentTemplateQuerySet(self.model, using=self._db)
-    
+
     def active(self):
         """Get active templates."""
         return self.get_queryset().active()
-    
+
     def by_category(self, category: str):
         """Get templates by category."""
         return self.get_queryset().by_category(category)
-    
+
     def by_creator(self, user):
         """Get templates by creator."""
         return self.get_queryset().by_creator(user)
-    
+
     def recent(self, days: int = 7):
         """Get recent templates."""
         return self.get_queryset().recent(days)
-    
+
     def search(self, query: str):
         """Search templates."""
         return self.get_queryset().search(query)
-    
+
     def create_template(
         self,
         name: str,
@@ -231,7 +232,7 @@ class AgentTemplateManager(models.Manager):
             created_by=created_by,
             **kwargs
         )
-    
+
     def get_categories(self) -> List[str]:
         """Get all unique categories."""
         return list(
@@ -241,11 +242,11 @@ class AgentTemplateManager(models.Manager):
             .exclude(category='')
             .order_by('category')
         )
-    
+
     def get_stats(self) -> Dict[str, Any]:
         """Get template statistics."""
         queryset = self.get_queryset()
-        
+
         return {
             'total': queryset.count(),
             'active': queryset.active().count(),