日: 2021年11月5日

[Python] 313 自作モジュールに実行スクリプト名を渡す

自作モジュール内で実行スクリプト名により条件分岐するif文を作ってみました。

os.path.basename(__file__)を使っています。

from my_library import moduleA
from os

# 実行スクリプト名の取得
source_file = str(os.path.basename(__file__))
print(f"source_file: {source_file}")

# moduleAに実行スクリプト名を渡す
moduleA.main(source_file)
--------------------------------------------------

出力
--------------------------------------------------
source_file: test.py
# 自作モジュールの一例
def main(file):
    if "test" in file:
        profit = ws.cell(column=5,row=maxrow -1).value
    else:
        profit = ws.cell(column=5,row=maxrow).value