1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
'''
Created on Jan 18, 2012
@author: leifj
'''
from celery.task import periodic_task,task
from celery.schedules import crontab
from meetingtools.apps.cluster.models import ACCluster
from meetingtools.ac import ac_api_client, ac_api_client_nocache,\
ac_api_client_direct
from meetingtools.apps.room.models import Room
import iso8601
from django.contrib.auth.models import User
from django.core.cache import cache
from django.core.exceptions import ObjectDoesNotExist
import logging
from datetime import datetime,timedelta
from lxml import etree
from django.db.models import Q
import math
def _owner_username(api,sco):
logging.debug(sco)
key = '_sco_owner_%s' % sco.get('sco-id')
logging.debug(key)
try:
if cache.get(key) is None:
fid = sco.get('folder-id')
if not fid:
logging.debug("No folder-id")
return None
folder_id = int(fid)
r = api.request('sco-info',{'sco-id':folder_id},False)
if r.status_code() == 'no-data':
return None
parent = r.et.xpath("//sco")[0]
logging.debug("p %s",repr(parent))
logging.debug("r %s",etree.tostring(parent))
name = None
if parent:
logging.debug("parent: %s" % parent)
if parent.findtext('name') == 'User Meetings':
name = sco.findtext('name')
else:
name = _owner_username(api,parent)
cache.set(key,name)
return cache.get(key)
except Exception,e:
logging.debug(e)
return None
def _extended_info(api,sco_id):
r = api.request('sco-info',{'sco-id':sco_id},False)
if r.status_code == 'no-data':
return None
return (r.et,_owner_username(api,r.et.xpath('//sco')[0]))
def _import_one_room(acc,api,row):
sco_id = int(row.get('sco-id'))
last = iso8601.parse_date(row.findtext("date-modified[0]"))
room = None
try:
room = Room.objects.get(acc=acc,deleted_sco_id=sco_id)
if room != None:
return # We hit a room in the process of being cleaned - let it simmer until next pass
except ObjectDoesNotExist:
pass
except Exception,e:
logging.debug(e)
return
try:
logging.debug("finding acc=%s,sco_id=%d in our DB" % (acc,sco_id))
room = Room.objects.get(acc=acc,sco_id=sco_id)
if room.deleted_sco_id != None:
return # We hit a room in the process of being cleaned - let it simmer until next pass
room.trylock()
except ObjectDoesNotExist:
pass
last = last.replace(tzinfo=None)
lastupdated = None
if room:
lastupdated = room.lastupdated.replace(tzinfo=None) # make the compare work - very ugly
logging.debug("last %s" % last)
logging.debug("lastupdated %s" % lastupdated)
if not room or lastupdated < last:
(r,username) = _extended_info(api, sco_id)
logging.debug("owner: %s" % username)
if username is None:
return
urlpath = row.findtext("url[0]").strip("/")
name = row.findtext('name[0]')
description = row.findtext('description[0]')
folder_sco_id = int(r.xpath('//sco[0]/@folder-id') or 0) or None
source_sco_id = int(r.xpath('//sco[0]/@source-sco-id') or 0) or None
if room == None:
user,created = User.objects.get_or_create(username=username)
if created:
user.set_unusable_password()
room = Room.objects.create(acc=acc,sco_id=sco_id,creator=user,name=name,description=description,folder_sco_id=folder_sco_id,source_sco_id=source_sco_id,urlpath=urlpath)
room.trylock()
else:
room.folder_sco_id = folder_sco_id
room.source_sco_id = source_sco_id
room.description = description
room.urlpath = urlpath
room.save()
room.unlock()
else:
room.unlock()
def _import_acc(acc):
api = ac_api_client_direct(acc)
backthen = datetime.now()-timedelta(seconds=6000000)
backthen = backthen.replace(microsecond=0)
logging.debug(backthen.isoformat())
r = api.request('report-bulk-objects',{'filter-type': 'meeting','filter-gt-date-modified': backthen.isoformat()})
for row in r.et.xpath("//row"):
_import_one_room(acc,api,row)
@periodic_task(run_every=crontab(hour="*", minute="*/1", day_of_week="*"))
def import_all_rooms():
for acc in ACCluster.objects.all():
_import_acc(acc)
def start_user_counts_poll(room,niter):
poll_user_counts.apply_async(args=[room],kwargs={'niter': niter})
@task(name='meetingtools.apps.room.tasks.poll_user_counts',rate_limit="10/s")
def poll_user_counts(room,niter=0):
logging.debug("rechecking user_counts for room %s" % room.name)
api = ac_api_client_direct(room.acc)
(nusers,nhosts) = api.poll_user_counts(room)
if nusers > 0:
logging.debug("room occupied by %d users and %d hosts, checking again in 20 ..." % (nusers,nhosts))
poll_user_counts.apply_async(args=[room],kwargs={'niter': 0},countdown=20)
elif niter > 0:
logging.debug("room empty, will check again in 5 for %d more times ..." % (niter -1))
poll_user_counts.apply_async(args=[room],kwargs={'niter': niter-1},countdown=5)
return (nusers,nhosts)
# belts and suspenders - we setup polling for active rooms aswell...
@periodic_task(run_every=crontab(hour="*", minute="*/5", day_of_week="*"))
def import_recent_user_counts():
for acc in ACCluster.objects.all():
api = ac_api_client_direct(acc)
then = datetime.now()-timedelta(seconds=600)
for room in Room.objects.filter((Q(lastupdated__gt=then) | Q(lastvisited__gt=then)) & Q(acc=acc)):
api.poll_user_counts(room)
|
