src/cm/activity.py
author raph
Mon, 23 Nov 2009 15:14:29 +0100
changeset 0 40c8f766c9b8
child 282 b5deb8e32219
permissions -rw-r--r--
import from internal svn r 4007
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     1
from cm.models import Activity, Text
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     2
from django.contrib.auth.models import User
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     3
from django.db.models import signals
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     4
from datetime import datetime, timedelta
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     5
from time import mktime
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     6
import django.dispatch
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     7
import logging
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     8
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
     9
def register_activity(request, type, text=None, comment=None, user=None, text_version=None):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    10
    signal_activity.send(sender=text, request=request, type=type, comment=comment, user=user, text_version=text_version)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    11
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    12
# activity signal
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    13
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    14
signal_activity = django.dispatch.Signal(providing_args=["request", "type", "comment"])
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    15
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    16
def _save_activity(sender, **kwargs):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    17
    request = kwargs['request']
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    18
    type = kwargs['type']
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    19
    comment = kwargs['comment']
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    20
    user = kwargs['user']
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    21
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    22
    text = sender
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    23
    text_version = kwargs.get('text_version', None)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    24
    if not text_version and text:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    25
    	text_version = text.last_text_version
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    26
        
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    27
    if request.user.is_anonymous():
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    28
        originator_user = None
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    29
    else:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    30
        originator_user = request.user
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    31
    ip = request.META['REMOTE_ADDR']
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    32
    Activity.objects.create(text=text, user=user, text_version=text_version, comment=comment, type=type, ip=ip, originator_user=originator_user)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    33
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    34
def connect_all():
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    35
    signal_activity.connect(_save_activity)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    36
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    37
connect_all()
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    38
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    39
# activity processing
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    40
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    41
def seconds(t_delta):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    42
    return abs(t_delta.seconds + t_delta.days * 86400)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    43
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    44
VISIT_DURATION = timedelta(seconds=30 * 60) # 30 minutes
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    45
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    46
def get_activity(text='all', user='all', reference_date=None, nb_slots=31, slot_timedelta=timedelta(days=1), action="all", kind=''):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    47
    """
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    48
    text : text: specific text
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    49
           'all': all texts
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    50
    user : user: specific user
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    51
           None: anonymous users
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    52
           'all': all users
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    53
    """
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    54
    # calc activities used    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    55
    if not reference_date:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    56
        reference_date = datetime.now()
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    57
    from_date = reference_date - slot_timedelta * nb_slots
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    58
    activities = Activity.objects.filter(created__gt=from_date)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    59
    if action != 'all':
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    60
        activities = activities.filter(type=action)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    61
    if text != 'all':
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    62
        activities = activities.filter(text=text)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    63
    if user != 'all':
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    64
        activities = activities.filter(originator_user=user)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    65
    activities = activities.order_by('created').only('created', 'originator_user', 'text')
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    66
    #print 'got %d activities' % len(activities), [a.created for a in activities]
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    67
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    68
    if kind == 'raw':
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    69
        visits = activities
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    70
    else:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    71
        # calc visits
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    72
        visits = []
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    73
        if activities:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    74
            for i in range(len(activities)):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    75
                activity = activities[i]
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    76
                found = False
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    77
                for j in range(i - 1, -1, -1):
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    78
                    prev_act = activities[j]                
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    79
                    if activity.created > prev_act.created + VISIT_DURATION: # out of session bounds: add act
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    80
                        visits.append(activity)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    81
                        found = True
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    82
                        break
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    83
                    else:
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    84
                        if not activity.is_same_user(prev_act): # another user: just ignore prev_act
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    85
                            continue
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    86
                        else: # in session: do not count act
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    87
                            found = True
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    88
                            break
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    89
                if not found:                    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    90
                    visits.append(activity)
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    91
        #print 'got %d visits' % len(visits), [(v.created, v.user) for v in visits]
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    92
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    93
    # hist by slot_timedelta
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    94
    slots = [seconds(from_date - v.created) // seconds(slot_timedelta) for v in visits]
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    95
    
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    96
    # TODO: could be more efficient...
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    97
    res = [slots.count(index) for index in range(nb_slots)]
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    98
    return res
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
    99
        
40c8f766c9b8 import from internal svn r 4007
raph
parents:
diff changeset
   100