pandazx's blog

Hadoop, データ分析など雑多な技術ブログ

python2で実行スクリプト名の判定

以下のようにしておけば、sample.py が他のスクリプトから呼ばれた時に main()を実行することを避けられる。

# sample.py
import sys

def calc(val):
    x = val + 1

def main():
    calc(1)

if sys.argsv[0] == __file__:
    main()

これにより、前に作ったスクリプトのメソッドを流用したい時に、簡単にimportできるようになる。