def project_usage(request, project_id, machine_category_id=settings.DEFAULT_MC):
machine_category = get_object_or_404(MachineCategory, pk=machine_category_id)
project = get_object_or_404(Project, pk=project_id)
if not project.can_view(request.user) and not getattr(settings, 'USAGE_IS_PUBLIC', False):
return HttpResponseForbidden('<h1>Access Denied</h1>')
if project.machine_categories.count() == 1:
machine_category = project.machine_categories.all()[0]
usage_list = []
total, total_jobs = 0, 0
querystring = request.META.get('QUERY_STRING', '')
start, end = get_date_range(request)
start_str = start.strftime('%Y-%m-%d')
end_str = end.strftime('%Y-%m-%d')
mc_ids = tuple([(int(m.id)) for m in machine_category.machine_set.all()])
if len(mc_ids) == 1:
mc_ids = "(%i)" % mc_ids[0]
# Custom SQL as need to get users that were removed from project too
cursor = connection.cursor()
sql = "SELECT user_id from cpu_job where project_id = '%s' and `machine_id` IN %s AND `date` >= '%s' AND `date` <= '%s' GROUP BY user_id" % (str(project.pid), mc_ids, start_str, end_str)
cursor.execute(sql)
rows = list(cursor.fetchall())
cursor.close()
for uid in rows:
u = UserAccount.objects.get(id=uid[0]).user
time, jobs = u.get_usage(project, start, end)
if time:
total += time
total_jobs += jobs
if jobs > 0:
usage_list.append({'user': u, 'usage': time, 'jobs': jobs})
for u in usage_list:
if total == 0:
u['percent'] = 0
else:
u['percent'] = (u['usage'] / total) * 100
usage_list = dictsortreversed(usage_list, 'usage')
count = 0
for i in usage_list:
i['colour'] = get_colour(count)
count += 1
graph = get_project_trend_graph_url(project, start, end, machine_category)
return render_to_response('usage/project_usage.html', locals(), context_instance=RequestContext(request))
def gen_project_graph(self, project, start, end, machine_category):
"""Generates a bar graph for a project
Keyword arguments:
project -- Project
start -- start date
end -- end date
machine_category -- MachineCategory object
"""
today = datetime.date.today()
start_str = start.strftime('%Y-%m-%d')
end_str = end.strftime('%Y-%m-%d')
start_t = start.strftime('%d/%m/%y')
end_t = end.strftime('%d/%m/%y')
fig = Figure(figsize=(6,3))
ax = fig.add_axes([0.2, 0.2, 0.7, 0.7])
period = (end-start).days
ax.set_xlim(0,period+1)
if period < 10:
step = 1
elif period < 100:
step = 20
else:
step = 50
ax.set_xticks(arange(period+1, step=step))
#print drange(start, end, datetime.timedelta(days=2))
ax.set_title('%s %s - %s' % (project.pid, start_t, end_t))
ax.set_ylabel("CPU Time (hours)")
ax.set_xlabel("Days")
mc_ids = tuple([(int(m.id)) for m in machine_category.machine_set.all()])
if len(mc_ids) == 1:
mc_ids = "(%i)" % mc_ids[0]
t_start = start
t_end = end
b_total = 0
user_data = []
cursor = connection.cursor()
SQL = "SELECT user_id from cpu_job where project_id = '%s' and `machine_id` IN %s AND `date` >= '%s' AND `date` <= '%s' GROUP BY user_id" % (str(project.pid), mc_ids, start_str, end_str)
cursor.execute(SQL)
rows = list(cursor.fetchall())
cursor.close()
for uid in rows:
try:
ua = UserAccount.objects.get(id=uid[0])
except UserAccount.DoesNotExist:
# Usage is from an unknown user.
continue
u = ua.user
cursor = connection.cursor()
SQL = "SELECT date, SUM( cpu_usage ) FROM `cpu_job` WHERE `project_id` LIKE '%s' AND `user_id` = %s AND `machine_id` IN %s AND `date` >= '%s' AND `date` <= '%s' Group By date" % (str(project.pid), str(ua.id), mc_ids, start_str, end_str)
cursor.execute(SQL)
rows = dict(cursor.fetchall())
if rows:
data, dates_y, labels = [], [], []
start = t_start
end = t_end
while start <= end:
if start != today:
try:
total = float(rows[start])
except:
total = 0
data.append(total / 3600.00)
labels.append(str(start))
b_total += total
start = start + datetime.timedelta(days=1)
user_data.append({'user': u, 'data': data, 'total': sum(data)})
#print ua.user
#print data
#print '-----'
#print 'prev: %s ' % prev_data
#ax.bar(dates, data, color=colours[count], edgecolor=colours[count])
#prev_data = data
#print b_total/ 3600
count = 0
prev_data = None
# majloc = dates.AutoDateLocator()
# majfmt = dates.AutoDateFormatter(majloc)
# ax.xaxis.set_major_locator(majloc)
# ax.xaxis.set_major_formatter(majfmt)
user_data = dictsortreversed(user_data, 'total')
user_data = [d['data'] for d in user_data]
for data in user_data:
x_data = range(1, len(data) + 1)
if prev_data:
ax.bar(x_data, data, color=get_colour(count), edgecolor=get_colour(count), bottom=prev_data, align='center')
p = 0
while p < len(prev_data):
prev_data[p] += data[p]
p += 1
else:
ax.bar(x_data, data, color=get_colour(count), edgecolor=get_colour(count), align='center')
prev_data = data
count += 1
canvas = FigureCanvasAgg(fig)
canvas.print_figure("%s/projects/%s_%s-%s_%i.png" % (str(settings.GRAPH_ROOT), str(project.pid), str(start_str), str(end_str), machine_category.id))
