wuam.py

#! python3

# ************************************************************
# Imports
# ************************************************************
import sys

import csv

import sqlite3

# Verify Python version
if sys.version_info[0] <= 3 and sys.version_info[1] < 7:
    sys.exit("This script requires at least Python version 3.7.")

from importlib import util

pyodbc_spec = util.find_spec("pyodbc")
if pyodbc_spec is None:
    sys.exit("Requires Python SQL Driver (pyodbc) - https://docs.microsoft.com/en-us/sql/connect/python/pyodbc/python-sql-driver-pyodbc?view=sql-server-2017")

import pyodbc

import globs

import argparse

import sqlite3

import wuapis

import BamLogger

import logging

import multiprocessing as mp

from support.utils import exitfunction, util_logconfig

try:
    globs.DBWSUSCONN = pyodbc.connect(globs.connstr)
except pyodbc.OperationalError as error:
    print(error)
    sys.exit("Must be able to connect to WSUS")

def displayhelp(parserh):
    '''
    displays help prompt
    '''
    parserh.print_help()

def parsecommandline(parser):
    '''
    parses arguments given to commandline
    '''
    parser.add_argument(
        "-wu", "--winupdateinfo", 
        help="Optional (Requires pyodbc)."
        "should be added to database.",
        action='store_true')
    if len(sys.argv) == 1:
        displayhelp(parser)
        exitfunction()

    return parser.parse_args()

if __name__ == "__main__":    

    PARSER = argparse.ArgumentParser()
    ARGS = parsecommandline(PARSER)

    print("Connected to: " + globs.connstr)

    globqueue = mp.Manager().Queue(-1)
    mainlogger = logging.getLogger("BAM.wuam")
    qh = logging.handlers.QueueHandler(globqueue)
    mainlogger.addHandler(qh)
    mainlogger.setLevel(logging.DEBUG)

    loggerProcess = mp.Process(target=BamLogger.log_listener, args=(globqueue, BamLogger.log_config))
    loggerProcess.start()

    wuapis.db_logconfig(globqueue)

    if ARGS.winupdateinfo:
        try:
            '''        
                Update the BAM DB with superseding and superseded information
            '''

            wuapis.updatewuentrysecedenceinfo()

        except sqlite3.Error as error:
            print("Error caught: ", error.args[0])
        except Exception as e:
            print("Error caught", str(e))
    
    globs.DBCONN.close()
    globs.DBWSUSCONN.close()
    globqueue.put_nowait(None)
    loggerProcess.join()