30 lines
784 B
Python
30 lines
784 B
Python
'''
|
|
Various helper functions and classes
|
|
'''
|
|
|
|
import time, os
|
|
|
|
def mkdirSafe(path, logger):
|
|
'''
|
|
Makes new dir if path does not exist, and aborts program if path exists and
|
|
path is a file not a dir. Else does nothing
|
|
'''
|
|
if not os.path.exists(path):
|
|
os.mkdir(path)
|
|
elif os.path.isfile(path):
|
|
logger.error('%s is present but is a file (vs a directory). ' \
|
|
'Please (re)move this file to prevent data loss', path)
|
|
raise SystemExit
|
|
|
|
def waitForPath(path, logger, timeout=30):
|
|
'''
|
|
Waits for a path to appear. Useful for procfs and sysfs where devices
|
|
regularly (dis)appear. Timeout given in seconds
|
|
'''
|
|
for i in range(0, timeout):
|
|
if os.path.exists(path):
|
|
return
|
|
time.sleep(1)
|
|
logger.error('Could not find %s after %s seconds', path, timeout)
|
|
raise SystemExit
|