Update is_writeable() for 2 methods (#4744)
* Writeable test * Fix * Cleanup
This commit is contained in:
1 parent
ff35292520
commit
22ee6fb7c1
1 file changed
+13
-11
+13
-11
@@ -112,17 +112,19 @@ def user_config_dir(dir='Ultralytics'):
|
|||||||
return path
|
return path
|
||||||
|
|
||||||
|
|
||||||
def is_writeable(dir):
|
def is_writeable(dir, test=False):
|
||||||
# Return True if directory has write permissions
|
# Return True if directory has write permissions, test opening a file with write permissions if test=True
|
||||||
# return os.access(path, os.R_OK) # known issue on Windows
|
if test: # method 1
|
||||||
file = Path(dir) / 'tmp.txt'
|
file = Path(dir) / 'tmp.txt'
|
||||||
try:
|
try:
|
||||||
with open(file, 'w'):
|
with open(file, 'w'): # open file with write permissions
|
||||||
pass
|
pass
|
||||||
file.unlink() # remove file
|
file.unlink() # remove file
|
||||||
return True
|
return True
|
||||||
except IOError:
|
except IOError:
|
||||||
return False
|
return False
|
||||||
|
else: # method 2
|
||||||
|
return os.access(dir, os.R_OK) # possible issues on Windows
|
||||||
|
|
||||||
|
|
||||||
def is_docker():
|
def is_docker():
|
||||||
|
|||||||
Reference in new issue
Block a user