U hF @sLddlmZddlmZddlmZGdddejZGdddejZdS) )models)timezone) ContentTypecs eZdZdZfddZZS)ActivityCategoryManagerzNCustom manager for ActivityCategory model that returns only active categories.cstjddS)z*Override to return only active categories.T) is_active)super get_querysetfilterself __class__3/var/www/html/Focus/src/apps/activities/managers.pyr sz$ActivityCategoryManager.get_queryset)__name__ __module__ __qualname____doc__r __classcell__rrr rrsrc@sZeZdZdZddZddZddZdd Zdd d Zd dZ ddZ ddZ ddZ dS)ActivityManagerz Custom manager for Activity model with additional query methods. This manager provides convenient methods for querying activities with common filters and aggregations. cCs |j|dS)z Get activities for a specific user. Args: user: User instance to filter activities for Returns: QuerySet: Activities performed by the specified user )userr r rrrrfor_users zActivityManager.for_usercCs ||S)zAlias for for_user method.)rrrrrby_user"szActivityManager.by_usercCstj|}|j||jdS)z Get activities for a specific object. Args: obj: Model instance to get activities for Returns: QuerySet: Activities related to the specified object ) content_type object_id)robjects get_for_modelr pk)r objrrrr for_object&s zActivityManager.for_objectcCs |j|dS)z Get activities by category code. Args: category_code (str): Category code to filter by Returns: QuerySet: Activities in the specified category )Zcategory__coder)r Z category_coderrr by_category3s zActivityManager.by_categorycCs ttj|d}|j|dS)z Get recent activities within the specified number of days. Args: days (int): Number of days to look back (default: 7) Returns: QuerySet: Activities from the last N days )days)created_at__gte)rnow timedeltar )r r$ cutoff_daterrrrecent?s zActivityManager.recentcCs |j|dS)z Get activities by action type. Args: action (str): Action type to filter by Returns: QuerySet: Activities with the specified action )actionr)r r*rrr by_actionLs zActivityManager.by_actioncCs |jddS)z Get only successful activities. Returns: QuerySet: Activities that completed successfully TZ is_successfulrr rrr successfulXszActivityManager.successfulcCs |jddS)zt Get only failed activities. Returns: QuerySet: Activities that failed Fr,rr rrrfailedaszActivityManager.failedcCs|j||dS)a Get activities within a specific date range. Args: start_date: Start date for the range end_date: End date for the range Returns: QuerySet: Activities within the specified date range )r%Zcreated_at__lter)r start_dateend_daterrr in_date_rangejs zActivityManager.in_date_rangeN)r#) rrrrrrr!r"r)r+r-r.r1rrrrrs      rN) django.dbr django.utilsr"django.contrib.contenttypes.modelsrManagerrrrrrrs