Script to Monitor Squid Cache File Size

import os

import ctypes

import sys

​

def alert(arg):

    sys.stderr.write("%d%d%d" % (arg, arg, arg))

​

def getfoldersize(path):

    total_size = 0

    spath = path

    for path, dirs, files in os.walk(spath):

        for f in files:

            fp = os.path.join(path, f)

            total_size += os.path.getsize(fp)

    return str(total_size)

​

class disable_file_system_redirection:

    _disable = ctypes.windll.kernel32.Wow64DisableWow64FsRedirection

    _revert = ctypes.windll.kernel32.Wow64RevertWow64FsRedirection

    def __enter__(self):

        self.old_value = ctypes.c_long()

        self.success = self._disable(ctypes.byref(self.old_value))

    def __exit__(self, type, value, traceback):

        if self.success:

            self._revert(self.old_value)

​

Squid_Path = r"{0}\Squid_ITarian\var\cache".format(os.environ['systemdrive'])

if os.path.exists(Squid_Path):

    print("Squid cache folder is present on the system")

    filesize = getfoldersize(Squid_Path)

    if int(filesize) > 2684354560:

        print("The 2.5 GB quota for the Squid cache folder is exceeded")

        print("The cache folder will be deleted")

        alert(1)

    else:

        print("The 2.5 GB Squid quota for the cache folder is NOT exceeded")

        alert(0)

else:

    print("Squid cache folder is NOT present on the system !!!")

    print("This procedure will stop")

    alert(0)