U hf6@s(dZddlZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZdd lmZmZmZmZmZmZeeZ Gd d d ej!Z"Gdddej#Z$Gdddej!Z%Gddde$Z&Gddde%Z'Gddde%Z(Gddde%Z)dS)aV Enterprise-Ready Model Managers Module This module provides custom model managers that extend Django's default manager functionality with enterprise-level features including caching, filtering, searching, and performance optimizations. Features: - Base managers with common query patterns - Caching managers for performance optimization - Search and filtering capabilities - Soft delete management - Audit trail management - Performance monitoring and optimization - Bulk operations with validation - Advanced querying methods Author: Focus Development Team Version: 2.0.0 License: Proprietary N)datetime timedelta)OptionalListDictAnyUnion)settings)timezone)cache)models transaction)get_user_model)ValidationError)QCountAvgSumMaxMincs8eZdZdZfddZfddZfddZZS) ActiveManagerz Custom manager that excludes soft deleted records by default. This manager automatically filters out records where is_deleted=True, providing a clean interface for working with active records only. cstjddS)z Return queryset excluding soft deleted records. Returns: QuerySet: All objects where is_deleted=False F is_deletedsuper get_querysetfilterself __class__//var/www/html/Focus/src/apps/common/managers.pyr0szActiveManager.get_querysetcs tS)z Return queryset including soft deleted records. Returns: QuerySet: All objects including soft deleted ones )rrrrr!r" with_deleted9szActiveManager.with_deletedcstjddS)z Return queryset with only soft deleted records. Returns: QuerySet: Only objects where is_deleted=True Trrrrr!r" deleted_onlyBszActiveManager.deleted_only)__name__ __module__ __qualname____doc__rr#r$ __classcell__r!r!rr"r(s rc@seZdZdZddZddZddZdd Zded d d Z ded ddZ ddZ e dddZ de ee dddZddZddZdS) BaseQuerySetz Base QuerySet with common filtering and optimization methods. This QuerySet provides standard filtering capabilities and performance optimizations that are commonly needed across different models in the application. cCs |jddS)Return only active objects.T is_activerrr!r!r"activeUszBaseQuerySet.activecCs |jddS)Return only inactive objects.Fr,r.rr!r!r"inactiveYszBaseQuerySet.inactivecCs |jddS)Return only published objects.T is_publishedr.rr!r!r" published]szBaseQuerySet.publishedcCs |jddS)z Return only unpublished objects.Fr3r.rr!r!r" unpublishedaszBaseQuerySet.unpublisheddayscCstt|d}|j|dS)*Return objects created in the last N days.r8)created_at__gter nowrrrr9 cutoff_dater!r!r"recenteszBaseQuerySet.recentcCstt|d}|j|dS)z!Return objects older than N days.r8)Zcreated_at__ltr<r>r!r!r" older_thanjszBaseQuerySet.older_thancCs |j|dS)*Return objects created by a specific user. created_byr.ruserr!r!r"by_useroszBaseQuerySet.by_userstatuscCs |j|dS)z&Return objects with a specific status.rHr.)rrIr!r!r" by_statussszBaseQuerySet.by_statusNqueryfieldscCs>|r|s |St}|D]}|tf|d|iO}q||S)z Perform text search across specified fields. Args: query: Search query string fields: List of fields to search in Returns: Filtered QuerySet Z __icontains)rr)rrLrMZ search_filterfieldr!r!r"searchws zBaseQuerySet.searchcCs|jtdtjddS)z'Add common annotations to the queryset.id created_at) total_countZ created_date)annotaterr functions TruncDaterr!r!r"with_annotationss zBaseQuerySet.with_annotationscCs |S)z+Apply common optimizations to the queryset.)select_relatedprefetch_relatedrr!r!r" optimizedszBaseQuerySet.optimized)r7)r7)N)r%r&r'r(r/r1r5r6intr@rArGstrrJrrOrVrYr!r!r!r"r*Lsr*c@seZdZdZddZddZddZdd Zded d d Z ddZ de e e dddZ de ejeedddZddZedddZdS) BaseManagerz Base manager class providing common functionality. This manager provides standard methods and optimizations that are commonly needed across different models. cCst|j|jdS)zReturn the base queryset.using)r*model_dbrr!r!r"rszBaseManager.get_querysetcCs |S)r+)rr/rr!r!r"r/szBaseManager.activecCs |S)r0)rr1rr!r!r"r1szBaseManager.inactivecCs |S)r2)rr5rr!r!r"r5szBaseManager.publishedr7r8cCs||S)r:)rr@)rr9r!r!r"r@szBaseManager.recentcCs||S)rB)rrGrEr!r!r"rGszBaseManager.by_userNrKcCs|||S)z,Perform text search across specified fields.)rrO)rrLrMr!r!r"rOszBaseManager.searchF)objs batch_sizeignore_conflictscCs"|D] }|q|j|||dS)a? Bulk create objects with validation. Args: objs: List of model instances to create batch_size: Number of objects to create in each batch ignore_conflicts: Whether to ignore conflicts Returns: List of created objects )rbrc) full_clean bulk_create)rrarbrcobjr!r!r"bulk_create_with_validations z'BaseManager.bulk_create_with_validationcKs.z|jf|WS|jjk r(YdSXdS)z Get an object or return None if not found. Args: **kwargs: Lookup parameters Returns: Model instance or None N)getr_ DoesNotExistrkwargsr!r!r" get_or_nones zBaseManager.get_or_none)returncKs|jf|S)z Check if an object exists with given parameters. Args: **kwargs: Lookup parameters Returns: Boolean indicating existence )rexistsrjr!r!r" exists_bys zBaseManager.exists_by)r7)N)NF)r%r&r'r(rr/r1r5rZr@rGr[rrOr Modelboolrgrlror!r!r!r"r\s" r\cs0eZdZdZddZfddZddZZS)SoftDeleteQuerySetz QuerySet for models with soft delete functionality. This QuerySet automatically excludes soft-deleted objects from standard queries while providing methods to access deleted objects when needed. cCs|jdtdS)z/Perform soft delete on all objects in queryset.T)r deleted_at)updater r=rr!r!r"deleteszSoftDeleteQuerySet.deletecs tS))Permanently delete objects from database.)rrurrr!r" hard_deleteszSoftDeleteQuerySet.hard_deletecCs|jddddS)zRestore soft-deleted objects.FN)rrs deleted_by)rtrr!r!r"restores zSoftDeleteQuerySet.restore)r%r&r'r(rurwryr)r!r!rr"rrs rrc@s0eZdZdZddZddZddZdd Zd S) SoftDeleteManagerz Manager for models with soft delete functionality. This manager automatically excludes soft-deleted objects from standard queries while providing methods to access deleted objects. cCst|j|jdjddS)z4Override to exclude soft-deleted objects by default.r]Frrrr_r`rrr!r!r"rszSoftDeleteManager.get_querysetcCst|j|jdjddS)z!Return only soft-deleted objects.r]Trr{rr!r!r"deletedszSoftDeleteManager.deletedcCst|j|jdS)z/Return all objects including soft-deleted ones.r])rrr_r`rr!r!r"r#szSoftDeleteManager.with_deletedcCst|j|jdS)rvr])rrr_r`rwrr!r!r"rw#szSoftDeleteManager.hard_deleteN)r%r&r'r(rr|r#rwr!r!r!r"rzs rzc@sVeZdZdZdZdZeedddZeddd Zedd d Z deed ddZ d S) CachedManagerz Manager with caching capabilities for improved performance. This manager provides automatic caching of frequently accessed objects and query results to improve application performance. iZcached_manager)keyrmcCs|jd|jjjd|S)z%Generate cache key for the given key.:) cache_prefixr__metalabel)rr~r!r!r"_get_cache_key3szCachedManager._get_cache_key) cache_keycKs^||}t|}|dkrZz |jf|}t|||jWn|jjk rXd}YnX|S)z Get an object from cache or database. Args: cache_key: Key to use for caching **kwargs: Lookup parameters Returns: Model instance or None N)rr rhset cache_timeoutr_ri)rrrkfull_cache_keyrfr!r!r" get_cached7s    zCachedManager.get_cachedcCs||}t|dS)zInvalidate cached object.N)rr ru)rrrr!r!r"invalidate_cacheNs zCachedManager.invalidate_cacheN)rqueryset_methodcKsT||}t|}|dkrP|jf|}|r8t||}t|}t|||j|S)a& Get a list of objects from cache or database. Args: cache_key: Key to use for caching queryset_method: Method to call on queryset **kwargs: Filter parameters Returns: List of model instances N)rr rhrgetattrlistrr)rrrrkrobjectsquerysetr!r!r"get_cached_listSs    zCachedManager.get_cached_list)N) r%r&r'r(rrr[rrrrr!r!r!r"r}(sr}c@sreZdZdZdeejeeeddddZeddd Z d d Z ejd d dZ e e dddZ dedddZdS) AuditManagerz Manager for audit log functionality. This manager provides methods for creating and querying audit log entries with proper filtering and optimization. NZAuditLog)actionrfchanges ip_address user_agentrmc CsTddlm}|||pi||d}|rH||j|t|jt|d|jf|S)a Create an audit log entry. Args: action: Action performed user: User who performed the action obj: Object affected by the action changes: Dictionary of changes made ip_address: IP address of the user user_agent: User agent string Returns: Created AuditLog instance r ContentType)rrDrrr) object_type object_id object_repr)"django.contrib.contenttypes.modelsrrtr get_for_modelr[pkcreate) rrrFrfrrrrZ audit_datar!r!r" log_actionts  zAuditManager.log_actionrcCs |j|dS)z(Return audit logs for a specific action.rr.)rrr!r!r" by_actionszAuditManager.by_actioncCs |j|dS)z&Return audit logs for a specific user.rCr.rEr!r!r"rGszAuditManager.by_user)rfcCs,ddlm}|j|}|j|t|jdS)z(Return audit logs for a specific object.rr)rr)rrrrrr[r)rrfr content_typer!r!r" by_objects  zAuditManager.by_object) start_dateend_datecCs|j||gdS)z&Return audit logs within a date range.)Zcreated_at__ranger.)rrrr!r!r" by_date_rangeszAuditManager.by_date_ranger7r8cCsVtt|d}|j||ddjtddd}tdd|Dt ||d S) z Get user activity summary for the last N days. Args: user: User to get activity for days: Number of days to look back Returns: Dictionary with activity statistics r8)rDr;rrP)countz-countcss|]}|dVqdS)rNr!).0itemr!r!r" sz5AuditManager.user_activity_summary..)Z total_actionsZactions_by_typeZ period_days) r r=rrvaluesrSrorder_bysumr)rrFr9r?Zactivityr!r!r"user_activity_summarys z"AuditManager.user_activity_summary)NNNN)r7)r%r&r'r(r[r rprrrrGrrrrZrr!r!r!r"rls  $r)*r(loggingrrtypingrrrrr django.confr django.utilsr django.core.cacher django.dbr r django.contrib.authrdjango.core.exceptionsrdjango.db.modelsrrrrrr getLoggerr%loggerManagerrQuerySetr*r\rrrzr}rr!r!r!r"s$       $K[D