Monitor bandwidth usage

Sizenumber=10

SizeDef="MB"   #KB or MB or GB      #SizeDef is case sensitive it should be declared in caps letters

import os

import ctypes

import re

import math

import sys

ps_content=r'''

Get-NetAdapterStatistics

'''

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)

def alert(arg):

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

def ecmd(command):

    import ctypes

    from subprocess import PIPE, Popen

​

    with disable_file_system_redirection():

        obj = Popen(command, shell = True, stdout = PIPE, stderr = PIPE)

    out, err = obj.communicate()

    ret=obj.returncode

    if ret==0:

        if out:

            return out.strip()

        else:

            return ret

    else:

        if err:

            return err.strip()

        else:

            return ret

def convert_size(B):

   'Return the given bytes as a human friendly KB, MB, GB, or TB string'

   B = float(B)

   KB = float(1024)

   MB = float(KB ** 2) # 1,048,576

   GB = float(KB ** 3) # 1,073,741,824

   TB = float(KB ** 4) # 1,099,511,627,776

​

   if B < KB:

      return '{0} {1}'.format(B,'Bytes' if 0 == B > 1 else 'Byte')

   elif KB <= B < MB:

      return '{0:.2f} KB'.format(B/KB)

   elif MB <= B < GB:

      return '{0:.2f} MB'.format(B/MB)

   elif GB <= B < TB:

      return '{0:.2f} GB'.format(B/GB)

   elif TB <= B:

      return '{0:.2f} TB'.format(B/TB)

def parseSize(size):

    units = {"Byte": 1, "KB": 10**3, "MB": 10**6, "GB": 10**9, "TB": 10**12}

    number, unit = [string.strip() for string in size.split()]

    return int(float(number)*units[unit])

​

def bandwidth():

    file_name='powershell_file.ps1'

    file_path=os.path.join(os.environ['TEMP'], file_name)

    with open(file_path, 'wb') as wr:

        wr.write(ps_content)

    ecmd('powershell "Set-ExecutionPolicy RemoteSigned"')

    ki=ecmd('powershell "%s"'%file_path)

    ki=str(ki)

    import re

    sam=re.findall('Ethernet(.*)',ki)

    os.remove(file_path)

    if sam>0:

        sam1=re.sub(' +',' ',sam[0])

        sam1=sam1.replace(" ",",")

        sam1=sam1.split(",")

        return sam1[1]

def comparision(Sizenumber,SizeDef,data):

    num=re.findall("Received Bytes: (.*)",data)[0]

    num=int(num)

    num=convert_size(num)

    num=parseSize(num)

    convert=(int(output))

    convert=convert_size(convert)

    convert=parseSize(convert)

    ki=parseSize(str(Sizenumber)+" "+SizeDef)

    Total=convert-num

    Total=abs(Total)

    if Total>ki:

        print "You accessed more data"

        print "You have used "+str(convert_size(Total))

        ale=1

        return ale

    else:

        print "You are using limited data"

        print "You have used "+str(convert_size(Total))

        ale=0

        return ale

path=os.path.join(os.environ['programdata'],"temp","Bandwidth.txt")

if os.path.exists(path):

    output=bandwidth()

    with open(path,"r")as f:

        data=f.read()

    ale=comparision(Sizenumber,SizeDef,data)

    with open(path,"w+")as f:

        f.write("Received Bytes: " + output)   

else:

    try:

        os.mkdir(os.path.join(os.environ['programdata'],"temp"))

    except:

        pass

    output=bandwidth()

    with open(path,"w")as f:

        f.write("Received Bytes: " + output)

    ale=0

    print "Script is running for first time"

​

if ale==1:

    alert(1)

else:

    alert(0)

​