工作中经常碰到这种情况:脚本开发是在本地(Windows),然后等脚本写好测好后放到生产服务器(Linux)。这就涉及到脚本在两种操作系统上的迁移和文件目录使用问题。以下小脚本虽然简单但还是很实用的。
import os
import platform
# set path
def doc_path():
if platform.system() == "Linux":
if os.path.exists("/root/"):
pass
else:
os.makedirs("/root/")
return "/root/"
elif platform.system() == "Windows":
if os.path.exists("D:/script/"):
pass
else:
os.makedirs("D:/script/")
return "D:/script/"
else:
exit()