import os

current_file_path = __file__
print(f"current_file_path: {current_file_path}")

import inspect

current_file_name = inspect.getfile(inspect.currentframe())
print(f"current_file_name: {current_file_name}")

import inspect

def be_call_fun():
  # stack()返回的是調用棧列表。
  frame_stack = inspect.stack()
  # 0是標識當前函數(shù)的棧，1是標識上一層函數(shù)的棧，依此類推。
  # 也就是這個數(shù)值不一定是1，要看你要獲取其文件路徑的函數(shù)在第幾層
  caller_frame = frame_stack[1]
  caller_file_path = caller_frame.filename
  # 由于當前調用函數(shù)和被調用函數(shù)放在同一個文件，所以文件名還是當前文件名
  # 可將調用函數(shù)和被調用函數(shù)放到不同文件進行觀察
  print(f"caller_file_path: {caller_file_path}")

def caller_fun():
  be_call_fun()

if __name__ == "__main__":
  caller_fun()

import os

current_working_dir = os.getcwd()
print(f"current_working_dir: {current_working_dir}")

import os

# 文件絕對路徑
current_file_path = __file__
# 借助dirname()從絕對路徑中提取目錄
current_file_dir = os.path.dirname(current_file_path)
print(f"current_file_dir: {current_file_dir}")

# 類似地可以借助basename()從絕對路徑中提取文件名
# current_filename = os.path.basename(current_file_path)

# 另外建議使用os.path.join()來實現(xiàn)路徑拼接，這樣不用自己再關注路徑分隔符的問題
# 再有是os.path.join()參數(shù)并不一定要就是目錄+文件名形式，不管多少個參數(shù)都可以拼接
# current_file_path = os.path.join(current_file_dir, current_filename)

import os


file_name = "test.txt"
new_file_name = "new_test.txt"

# 創(chuàng)建空白文件
open(file_name, 'w').close()

# 刪除文件
os.remove(file_name)

# 重命名文件
os.rename(file_name, new_file_name)

# 讀文件
with open(file_name, 'r') as fd:
  # 讀取所有內容
  fd.read()
  # 讀取一行
  fd.readline()
  # 讀取所有行，返回為各行組成的形表
  fd.readlines()

# 遍歷文件各行簡潔寫法
for line in open(file_name, 'r', encoding='utf-8'):
  print(line)

# 寫文件
with open(file_name, 'w') as fd:
  # 向文件寫入內容，需要自行添加\n
  fd.write("test_str")
  # 向文件寫入內容，需要自行添加\n
  # 和write的區(qū)別是，其參數(shù)除了可以是字符串，還可以是字符串列表
  fd.writelines(["test_str", "test_str"])

import os


dir_name = "test_dir/dir_name"
new_dir_name = "test_dir/new_dir_name"

# 創(chuàng)建目錄。此種形式是的于shell的mkdir，在父目錄不存在時會創(chuàng)建失敗
os.mkdir(dir_name)
# 創(chuàng)建目錄。此種形式是的于shell的mkdir -p，在父目錄不存在時會自動創(chuàng)建父文件夾
os.makedirs(dir_name, exist_ok=True)

# 刪除目錄。此種形式相當于shell的rmdir，在目錄非空時刪除失敗
os.rmdir(dir_name)
# 刪除目錄。此種形式相當于shell的rm -rf
import shutil
shutil.rmtree(dir_name)

# 重命名目錄。
os.rename(dir_name, new_dir_name)

# 遍歷目錄
# dir_path是當前遍歷到的目錄。dir_names是dir_path下的文件夾列表。file_names是是dir_path下的文件列表
# 如果想實現(xiàn)目錄白名單，將白名單目錄從dir_names中去除即可
for (dir_path, dir_names, file_names) in os.walk(dir_name):
  for file_name in file_names:
    print(os.path.join(dir_path, file_name))