109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
#! /usr/bin/env python
|
|
import cherrypy, tempfile, shutil, os
|
|
|
|
import crans.cp
|
|
from ClassesIntranet.ModuleBase import ModuleBase
|
|
import crans.utils.quota as quota
|
|
|
|
class main(ModuleBase):
|
|
def category(self):
|
|
return "Personnel"
|
|
def title(self):
|
|
return "Quotas"
|
|
def icon(self):
|
|
quotas = self._get_quota()
|
|
for a_quota in quotas:
|
|
if a_quota['quota'] < a_quota['usage']:
|
|
return "icon_alert.png"
|
|
return "icon.png"
|
|
|
|
def _get_quota(self):
|
|
if (cherrypy.config.configMap["global"]["server.environment"] == "development"):
|
|
return quota.fake_getUserQuota(cherrypy.session['uid'])
|
|
else:
|
|
return quota.getUserQuota(cherrypy.session['uid'])
|
|
|
|
|
|
|
|
##########################
|
|
# affichage
|
|
##########################
|
|
#
|
|
# methode qui affiche la template
|
|
#
|
|
def index(self ):
|
|
values = {}
|
|
try:
|
|
quotas = self._get_quota()
|
|
returned_quotas = []
|
|
for a_quota in quotas:
|
|
# calculate text equivalent
|
|
quota = a_quota['quota']
|
|
usage = a_quota['usage']
|
|
limite = a_quota['limite']
|
|
text_equiv = "["
|
|
text_equiv+= "#" * min( 10, int( 10 * usage / quota ) )
|
|
text_equiv+= "." * max(0, int( 10 * ( quota - usage ) / quota ) )
|
|
if limite > quota:
|
|
text_equiv+= "|"
|
|
limite_diff = limite - quota
|
|
diff_proportion = 10 * limite_diff / quota
|
|
depassement = max(0, usage - quota)
|
|
text_equiv+= "#" * min(0, int(diff_proportion* ( depassement / limite_diff ) ) )
|
|
text_equiv+= "." * max(0, int(diff_proportion*( limite_diff - depassement ) / limite_diff ) )
|
|
text_equiv+= "]"
|
|
a_returned_quota = {
|
|
"label":a_quota['label'],
|
|
"usage":a_quota['usage'],
|
|
"quota":a_quota['quota'],
|
|
"limite":a_quota['limite'],
|
|
"percents":a_quota['%'],
|
|
"%":a_quota['%'],
|
|
"text_equiv":text_equiv,
|
|
"svg_url":"barreSVG?filesystem=%s" % a_quota['filesystem'],
|
|
}
|
|
returned_quotas.append(a_returned_quota)
|
|
values = {'quotas': returned_quotas}
|
|
except Exception, e:
|
|
crans.cp.log('error getting quota for user %s : %s' % (cherrypy.session['uid'], str(e)), 'QUOTA', 1)
|
|
values = {'erreur':str(e)}
|
|
return {'template':'quota',
|
|
'values': values,
|
|
'stylesheets':['quota.css'],
|
|
'scripts':['quota.js', 'popup.js'],
|
|
}
|
|
index.exposed = True
|
|
|
|
def index_html(self ):
|
|
result = self.index()
|
|
result['template'] = 'quota_html'
|
|
return result
|
|
index_html.exposed = True
|
|
|
|
|
|
##########################
|
|
# SVG
|
|
##########################
|
|
#
|
|
# methode qui renvoie une barre en svg
|
|
#
|
|
def barreSVG(self, filesystem = ""):
|
|
try:
|
|
values = {'erreur':"Not found"}
|
|
quotas = self._get_quota()
|
|
for a_quota in quotas:
|
|
if a_quota['filesystem'] == filesystem:
|
|
values = {
|
|
"usage":a_quota['usage'],
|
|
"quota":a_quota['quota'],
|
|
"limite":a_quota['limite'],
|
|
"percents":a_quota['%'],
|
|
}
|
|
except Exception, e:
|
|
values = {'erreur':str(e) }
|
|
cherrypy.response.headers['Content-Type']="image/svg+xml"
|
|
return {'template':'barre.svg',
|
|
'values': values,
|
|
'standalone':True,
|
|
}
|
|
barreSVG.exposed= True
|