[work] python 判断文件夹不存在时创建文件夹

迷南。 2022-04-02 03:44 401阅读 0赞

1.  `if not os.path.exists(directory):`
2.  `os.makedirs(directory)`

今天敲代码的时候，发现一个问题，找不到文件夹，我的代码的文件夹是写死的，先放一段代码

1.  `file_path = os.path.dirname(os.getcwd()) + '/TestPython/Screenshots/'`
2.  `rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))`
3.  `screen_name = file_path + rq + '.png'`
4.  `try:`
5.  `self.driver.get_screenshot_as_file(screen_name)`
6.  `logger.info("开始截图并保存")`
7.  
8.  `except Exception as e:`
9.  `logger.error("出现异常", format(e))`

这样很不方便，需要自己手动在根目录下创建文件夹，如果忘了就会报错，这时候就需要通过代码自动创建，我们需要给一个判断，有文件夹时候就执行下面的代码，没有的就自行创建。

1.  `file_path = os.path.dirname(os.getcwd()) + '/TestPython/Screenshots/'`
2.  `if not os.path.exists(file_path):`
3.  `os.mkdir(file_path)`
4.  `else:`
5.  `rq = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))`
6.  `screen_name = file_path + rq + '.png'`
7.  `try:`
8.  `self.driver.get_screenshot_as_file(screen_name)`
9.  `logger.info("开始截图并保存")`
10. 
11. `except Exception as e:`
12. `logger.error("出现异常", format(e))`

这是修改以后的，可以明显看出来。

这里会有一个问题，如果一个文件在调用`os.path.exists`和`os.makedirs`之间被创建了,将会出现一个`OSError`.遗憾的是捕获`OSError`异常继续进行并不是万无一失的,它将会忽略像磁盘空间不足,没有足够权限等一些其他造成文件创建失败的因素.

一个做法是捕获`OSError`异常并检查返回的错误代码(前提是知道错误代码对应的是什么).然而,还有另一种可能,第二次的`os.path.exists`.假如恰好在第一次检查的时候创建了文件夹,然后在第二次检查的时候删掉~