diskquota.py

from util       import run_cmd, capture,syshost
from datetime import *
import re, sys

def diskquota():

  # primitive argument input for userid - no error checking
  if len(sys.argv)==2:
    userid=sys.argv[1]
  else:
    userid=capture("whoami").rstrip()
  #userid="u6006455"
  host=syshost()
  grepcmd1="curl --data 'unid={0}' https://www.chpc.utah.edu/apps/systems/curl_post/CHPCUserQuota.php -k -s | tail -n +2".format(userid)      
  myfs=capture(grepcmd1).splitlines()
  
  goodcolor = "32m"
  badcolor = "31m"
  for space in myfs:
      #print(space)
   # if ( (host=="ls4" and space=="/home1") or (host=="maverick" and space=="/home") or (host=="ls5" and space=="/home1")):
      rawinfo=space.split("\t")
      #print(rawinfo)
      if (rawinfo[0]==userid) or (userid in rawinfo[0]):
        usage = rawinfo[1][:-4]
        files = rawinfo[3]
        quota = rawinfo[2][:-1]
        if rawinfo[0]==userid:
          fsuse = rawinfo[5][:-4]
          fsmax = rawinfo[6][:-1]
          #if ((fsmax != 0) and (float(fsuse)/float(fsmax) > 0.9)):
          if (isinstance(fsuse, int) and isinstance(fsmax, int)):
            if (float(fsuse)/float(fsmax) > 0.9):
              fscolor=badcolor
          else:
            fscolor=goodcolor;
        if quota=="0":
          badusage=100
        else:
          badusage=int(quota)
        if float(usage) > badusage:
          ucolor=badcolor
          umsg = "\033[1;31mare over the quota limit\033[0m"
        else:
          ucolor=goodcolor
          umsg = "have quota limit"
        if int(files) > 100000:
          fcolor=badcolor
        else:
          fcolor=goodcolor
      
        if rawinfo[0]==userid:
          if (rawinfo[2]=="0G"):
            if (files==0):
              print("\tIn \033[1;36m{0}\033[0m you have no quota limit, use \033[1;{2}{1} GB\033[0m".format("General Home Directory",usage,ucolor))
            else:
              print("\tIn \033[1;36m{0}\033[0m you have no quota limit, use \033[1;{3}{1} GB\033[0m and have \033[1;{4}{2}\033[0m files".format("General Home Directory",usage,files,ucolor,fcolor))
            if (fsuse!=0):
              print("\t   (total file system usage is \033[1;{2}{0} GB\033[0m out of \033[1;{2}{1} GB\033[0m)".format(fsuse,fsmax,fscolor))
            #print("\tOn \033[1;36m{0}\033[0m you have no quota limit, use \033[1;{3}{1} GB\033[0m (total file system usage is {5} GB out of {6} GB) and have \033[1;{4}{2}\033[0m files".format(rawinfo[0],usage,files,ucolor,fcolor,fsuse,fsmax))
          else:
            if (files=='0'):
              print("\tIn \033[1;36m{0}\033[0m you {4} \033[1;{3}{2} GB\033[0m and use \033[1;{3}{1} GB\033[0m".format("General home directory",usage,quota,ucolor,umsg))
            else:
              print("\tIn \033[1;36m{0}\033[0m you {6} \033[1;{4}{3} GB\033[0m, use \033[1;{4}{1} GB\033[0m and have \033[1;{5}{2}\033[0m files".format("General home directory",usage,files,quota,ucolor,fcolor,umsg))
            if (fsuse!='0'):
              print("\t   (total file system usage is \033[1;{2}{0} GB\033[0m out of \033[1;{2}{1} GB\033[0m)".format(fsuse,fsmax,fscolor))
        else:
          print("\tIn \033[1;36m{0}\033[0m you {6} \033[1;{7}{3} GB\033[0m, use \033[1;{4}{1} GB\033[0m and have \033[1;{5}{2}\033[0m files".format("PE Home Directory",usage,files,quota,ucolor,fcolor,umsg,ucolor))