Get bitlocker status of drives and send mail (outlook and gmail)

import ctypes

import re

import socket

import subprocess

from datetime import datetime

from email.mime.multipart import MIMEMultipart

from email.mime.text import MIMEText

import smtplib

​

Receiver = itsm.getParameter('EmailTo') #Provide a Toemail address where the mail need to be sent. the datatype should be a string.

Sender = itsm.getParameter('EmailFrom') #Provide the From Email address from which the mail to be send. the datatype should be a string.

Password = itsm.getParameter('Password') #Provide app password for from email. the datatype should be a string.

MailFlag = itsm.getParameter('MailFlag')  #Provide mail flag 1 or 0 (1 - outlook, 0 - gmail). the datatype should be a int.

​

​

def gmail(sender_email, password, receiver, text):

    try:

        msg = MIMEMultipart()

        msg["From"] = sender_email

        msg["To"] = receiver

        msg["Subject"] = "Bitlocker Status"

        msg.attach(MIMEText(text, 'plain'))

        if MailFlag:

            server = smtplib.SMTP("smtp.office365.com", 587)

        else:

            server = smtplib.SMTP("smtp.gmail.com", 587)

        server.starttls()

        server.login(sender_email, password)

        server.sendmail(sender_email, receiver, msg.as_string())

        server.quit()

        return "Bitlocker Status has been mailed successfully"

    except Exception as e:

        return e

​

​

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)

​

​

Cmd = 'C:\\Windows\\System32\\manage-bde.exe -status'

Drvlist = 'fsutil fsinfo drives'

​

​

def executeCmd(ExcCMD):

    with disable_file_system_redirection():

        process = subprocess.Popen(ExcCMD, shell=True, stdout=subprocess.PIPE)

    result = process.communicate()[0]

    return result

​

​

excResult = ''

getDisk = []

driverList = executeCmd(Drvlist).split(" ")[1:-1]

for dsk in driverList:

    DiskStat = executeCmd(Cmd + " " + dsk.replace('\\', ''))

    if 'Protection Off' in DiskStat and 'Unlocked' in DiskStat:

        getDisk.append(dsk.replace('\\', ''))

        if excResult == '':

            excResult = excResult + DiskStat

        else:

            excResult = excResult + re.sub('BitLocker .+reserved.\r\n\r\n', '', DiskStat, flags=re.DOTALL)

if excResult == '':

    excResult = executeCmd(Cmd)

​

​

def getBitStatus(getlist):

    BitlockerStat = ""

    if len(getlist) != 1:

        for i in range(len(getlist)):

            if i == 0:

                BitlockerStat = BitlockerStat + 'Bitlocker for ' + getlist[i]

            elif i != len(getlist) - 1:

                BitlockerStat = BitlockerStat + ", " + getlist[i]

            elif i == len(getlist) - 1:

                BitlockerStat = BitlockerStat + " and " + getlist[i] + " drive is deactivated\n\n"

    elif len(getlist) == 1:

        BitlockerStat = BitlockerStat + "Bitlocker for " + getlist[0] + " drive is deactivated\n\n"

    elif len(getlist) == 0:

        BitlockerStat = BitlockerStat + 'Bitlocker is active\n\n'

    return BitlockerStat

​

​

content = socket.gethostname() + "\t\t" + datetime.now().strftime("LAST STATUS UPDATE: %m/%d/%y %H:%M:%S %p") \

          + "\n\n" + getBitStatus(getDisk) + excResult

​

if len(getDisk) != 0:

    print str(gmail(Sender, Password, Receiver, content)) + '\n'

print content

​