Автоматизировать студент тестов запустить

Question

У меня есть структура папок для некоторых задач, они так:

- student_id1/answers.py 
- student_id2/answers.py 
- student_id3/answers.py 
- student_id4/answers.py 
- ... 

У меня есть основной файл: run_tests.py:

from student_id1.answers import run_test as test1 
from student_id2.answers import run_test as test2 
... 


try: 
    test1() 
    print("student_id1: OK") 
except Exception as ee: 
    print("Error for student_id1") 

try: 
    test2() 
    print("student_id2: OK") 
except Exception as ee: 
    print("Error for student_id2") 
... 

Там может быть несколько папок, как они добавляются с каждым новым учеником. Я хотел бы назвать все тесты одной командой, но не хочу добавлять столько строк с каждым новым учеником.

Как я могу автоматизировать это?

Answer 1

Вы можете использовать importlib модуль: https://docs.python.org/3/library/importlib.html

import os 
import importlib 

for student_dir in os.listdir(): 
    if os.path.isdir(student_dir): 
     # here you may add an additional check for not desired folders like 'lib', 'settings', etc 
     if not os.path.exists(os.path.join(student_dir, 'answers.py')): 
      print("Error: file answers.py is not found in %s" % (student_dir)) 
      continue 
     student_module = importlib.import_module("%s.answers" % student_dir) 
     try: 
      student_module.run_test() 
      print("%s: OK" % student_dir) 
     except Exception as ee: 
      print("Error for %s" % student_dir)