script.py

import os
os.system('python -m pip install pywinusb pandas tk pillow pywin32 psutil requests')
from time import sleep
from msvcrt import kbhit
from configfile import *
import pywinusb.hid as hid
import threading
from tkinter import *
from tkinter import ttk
import tkinter
from PIL import Image, ImageTk
import pandas as pd
import time
import sys
import time
import win32evtlog
import win32evtlogutil
import requests
import datetime
import subprocess
import logging

logger=logging.getLogger("MTR_log")
logger.setLevel(logging.DEBUG)
fh=logging.FileHandler(log_file_path)
fh.setLevel(logging.DEBUG)
logger.addHandler(fh)
device_id=""

try:
	slno=subprocess.run("wmic bios get serialnumber",stdout=subprocess.PIPE)
	slnob= slno.stdout.decode()
	fnl= slnob.split()
	dev_slno= fnl[1]
	url='https://login.microsoftonline.com/'+tenant_id+'/oauth2/v2.0/token'
	header ={}
	header['Content-Type']= 'application/x-www-form-urlencoded'
	data={}
	data['client_id']=client_id
	data['scope']='https://graph.microsoft.com/.default'
	data['client_secret']=client_secret
	data['grant_type']='client_credentials'
	x=requests.post(url,headers=header,data=data)
	y=x.json()
	tok=(y["access_token"])
	url='https://graph.microsoft.com/beta/teamwork/devices/'
	header={}
	
	data={}
	header['Authorization']='Bearer'+' '+tok
	x=requests.get(url,headers=header,data=data)
	y=x.json()
	for sl in y["value"] :
		if sl["hardwareDetail"]["serialNumber"]== dev_slno :
			device_id=sl["id"]
			break
	print(device_id)
except:
	pass

def is_idle():
	if device_id!="":
		return is_idle2()
	else:
		return True

def is_idle2():
	global dispname
	global devname
	sleep(1)
	url='https://login.microsoftonline.com/'+tenant_id+'/oauth2/v2.0/token'
	header ={}
	header['Content-Type']= 'application/x-www-form-urlencoded'
	data={}
	data['client_id']=client_id
	data['scope']='https://graph.microsoft.com/.default'
	data['client_secret']=client_secret
	data['grant_type']='client_credentials'
	x=requests.post(url,headers=header,data=data)
	y=x.json()
	tok=(y["access_token"])
	url='https://graph.microsoft.com/beta/teamwork/devices/'+device_id
	header={}
	data={}
	header['Authorization']='Bearer'+' '+tok
	z=requests.get(url,headers=header,data=data)
	a=z.json()
	state=a['activityState']
	dispname=a['hardwareDetail']['serialNumber']
	devname=a['currentUser']['displayName']
	return state=='idle'

def events(message):
	DUMMY_EVT_APP_NAME = "Authentication request MTR"
	DUMMY_EVT_ID = 7040  
	DUMMY_EVT_CATEG = 9876
	DUMMY_EVT_STRS = [message]
	DUMMY_EVT_DATA = message.encode()
	win32evtlogutil.ReportEvent(DUMMY_EVT_APP_NAME, DUMMY_EVT_ID, eventCategory=DUMMY_EVT_CATEG, eventType=win32evtlog.EVENTLOG_INFORMATION_TYPE, strings=DUMMY_EVT_STRS,data=DUMMY_EVT_DATA)

def validcheck(val):
	global dispname
	global devname
	try:
		val=str(val)
		df=pd.read_csv(data_path)
		my_df=df.query("serial=="+val)
		x=my_df['user id'].values[0]
		#t1=threading.Thread(target=pop,name='t1',args=(str(x),val))
		#t1.start()
		destroy_window()
		log=((str(datetime.datetime.now()))[:19] +" SUCCESS on Device Sl No "+ dispname+" on Device Name "+devname+" with Security Key "+val+" by UserID "+str(x))
		write_log(log)
	except Exception as e:
		x="No User Id found"
		log=((str(datetime.datetime.now()))[:19] +" FAILED on Device Sl No "+ dispname+" on Device Name "+devname+" with Security Key "+val)
		write_log(log)
		write_log(str(e))

def write_log(stmt):
	logger.info(stmt)
	events(stmt)
      

def setup():
	global flag
	global desttime
	desttime=0
	flag=False
	window()

	
def destroy_window():
	global root1
	global flag
	global desttime
	desttime=time.time()
	flag=False
	#t8=threading.Thread(target=team_start,name='t8')
	#t8.start()
	root1.withdraw()


def exit_window(e):
        global dispname
        global devname
        write_log((str(datetime.datetime.now()))[:19] +" DEBUG on Sl No "+ dispname+" on Device Name "+devname+" by pressing ESC on Keyboard")
        destroy_window()
		
def lockup():
	global flag
	global desttime
	while use_service:
		if not flag:
			currtime=time.time()
			if currtime-desttime>delay_time and is_idle():
				show_window()
		
def show_window():
	global root1
	global flag
	if is_idle():
		flag=True
		root1.deiconify()

def window():
	global flag
	global root1
	flag=True
	root1 = Tk()
	root1.attributes('-fullscreen',True)
	root1.attributes('-topmost',True)
	root1.attributes('-alpha',0.4)
	image1=Image.open(image_file_path)
	image1=image1.resize((root1.winfo_screenwidth(),root1.winfo_screenheight()))
	test=	ImageTk.PhotoImage(image1)
	label1=tkinter.Label(image=test)
	label1.image=test
	label1.place(x=-2,y=-2)
	root1.bind('<Escape>',exit_window)
	root1.mainloop()

def sample_handler(data):
    global flag
    data=data[10:18]
    slno=int(bytes(data).decode())
    print("handler before if", slno, " ", flag)
    if flag:
        print("handler check ",slno)
        validcheck(slno)
    
def raw_test2():
	while True:
		raw_test()

def raw_test():	
    all_hids = hid.find_all_hid_devices()
    if all_hids:
        while True:
            index_option="0"
            for index, device in enumerate(all_hids):
                device_name = str("{0.vendor_name} {0.product_name}" \
                        "(vID=0x{1:04x}, pID=0x{2:04x})"\
                        "".format(device, device.vendor_id, device.product_id))
                device_name=device_name.lower()              
                if 'baltech' in device_name.lower() or 'kofax' in device_name.lower():
                    index_option=str(index+1)
                    break
            if index_option.isdigit() and int(index_option) <= len(all_hids):
                break;
        int_option = int(index_option)
        if int_option:
            device = all_hids[int_option-1]
            try:
                device.open()
                device.set_raw_data_handler(sample_handler)
                
                print("Tap Yubikey on reader")
                while device.is_plugged():
                    sleep(0.5)
                return
            finally:
                device.close()
    else:
        print("Device unavailable")

is_idle()        

if __name__ == '__main__' and use_service:
	sleep(1)
	t2=threading.Thread(target=raw_test2,name='t2')
	t2.start()
	sleep(1)
	t6=threading.Thread(target=setup,name='t6')
	t6.start()
	sleep(5)
	lockup()