TPD-Keys/Helpers/wvd_check.py
2023-12-08 13:53:24 -05:00

22 lines
703 B
Python

# Import dependencies
import os
import glob
# 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