pythonでimportすると「ImportError: cannot import name 'webdriver'」になる。

Question

実行するとエラーになります。

「ImportError: cannot import name 'webdriver'」
どうしたらいいでしょうか？

script

test_ut.py (unittest実行用)

import unittest
import selenium.webtest1

def suite():
  suite = unittest.TestSuite()
  suite.addTest(selenium.webtest1('TestCase1'))
  return suite


if __name__ == '__main__':
  runner = unittest.TextTestRunner()
  test_suite = suite()
  runner.run(test_suite)

selenium/webtest1.py　(selenium実行ファイル)

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re, datetime, os

class TestCase1(unittest.TestCase):
        def setUp(self):
                self.driver = webdriver.Firefox()
                self.driver.implicitly_wait(30)
                self.base_url = "https://yahoo.co.jp/"
                self.verificationErrors = []
                self.accept_next_alert = True

        def test_untitled_test_case(self):
                driver = self.driver
                driver.get("https://yahoo.co.jp")
                print(driver.title)

        def tearDown(self):
                self.driver.quit()
                self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
        unittest.main()

単体での実行結果

# python36 selenium/webtest1.py
Yahoo! JAPAN
.
----------------------------------------------------------------------
Ran 1 test in 2.544s

OK

それをimportしたunittest側の実行結果

# python36 test_ut.py
Traceback (most recent call last):
  File "test_ut.py", line 2, in <module>
    import selenium.webtest1
  File "/ababa/selenium/webtest1.py", line 2, in <module>
    from selenium import webdriver
ImportError: cannot import name 'webdriver'

moduleとして実行しても変わらず

# python36 -m test_ut
Traceback (most recent call last):
  File "/usr/lib64/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib64/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/ababa/test_ut.py", line 2, in <module>
    import selenium.webtest1
  File "/ababa/selenium/webtest1.py", line 2, in <module>
    from selenium import webdriver
ImportError: cannot import name 'webdriver'

環境

python 3.6
centos7

Answer 1

ImportError: cannot import name 'webdriver'

グーグル翻訳
ImportError：名前 'webdriver'をインポートできません

File "/XXXX/selenium/webtest1.py", line 2, in <module>

ディレクトリ名がseleniumモジュールとかぶっています。別の名前に変更してくださいな。

---

◇補足説明

File "/XXXX/test_ut.py", line 2, in <module>
import selenium.webtest1
File "/XXXX/selenium/webtest1.py", line 2, in <module>
from selenium import webdriver

例外スタックトレースによるとプログラム階層が以下になります。

XXXX
 +- test_ut.py
 +- selenium
    +- webtest1.py
    +- __init__.py # ←質問文に記載がないのですが、このファイルがあるのでは？。

python36 selenium/webtest1.py
python36 test_ut.py

この2つのコマンドの動作の違いは、6.1.2. モジュール検索パスの以下の部分です。

入力されたスクリプトのあるディレクトリ (あるいはファイルが指定されなかったときはカレントディレクトリ)。
～中略
スクリプトファイルを含むディレクトリが検索パスの先頭、標準ライブラリパスよりも前に追加されます。なので、ライブラリのディレクトリにあるファイルよりも、そのディレクトリにある同じ名前のスクリプトが優先してインポートされます。

seleniumフォルダにある__init__.pyファイルによりseleniumフォルダが6.4. パッケージ化されて
モジュールの解決に影響を与えました。

IDE(Pycharm community edition)などを使用すると、こういうトラブルは防げるかと。

Comments

[ustst]

ありがとうございます！
元々は別のエラーが出ていて検証していたんですが、
ここでの質問用に整理して専用ディレクトリとスクリプトを作ってたら
こんな感じになっちゃいました。
ひとまず今のエラーについては解決できそうで何よりです。
ディレクトリ名についてなんて考えたことなかった。

ちょっと急な仕事でまだ検証できていないのですが、取り急ぎレスさせて頂きます。
ありがとうございました。