TPD-Keys/Helpers/wvd_check.py
TPD94 577d36920a v1.21
Added code to allow exe build.

Minor GUI changes
2023-12-30 13:44:35 -05:00

23 lines
724 B
Python

# Import dependencies
import os
import glob
from sys import exit
# Define WVD device check
def wvd_check():
try:
# Check to see if the WVDs folder exist, if not create it
if 'WVDs' not in os.listdir(fr'{os.getcwd()}'):
os.makedirs(f'{os.getcwd()}/WVDs')
# Use glob to get the name of the .wvd
extracted_device = glob.glob(f'{os.getcwd()}/WVDs/*.wvd')[0]
# Return the device path
return extracted_device
except:
# Check to see if the WVDs folder exist, if not create it
if 'WVDs' not in os.listdir(fr'{os.getcwd()}'):
os.makedirs(f'{os.getcwd()}/WVDs')
# Stop the program and print out instructions
return None