t_euler_problem.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
Usage: ./t_euler_problem.py [problem_id*]
"""
import sys
import re                       # regex
import importlib
from os import listdir
from os.path import isdir, join

target_dir = "./"

def find_and_figure_out_problem():
    t_exec_flag = False
    for argv_index in range(1, len(sys.argv)):
        # print sys.argv[argv_index]
        problem_str = sys.argv[argv_index]
        problem_id = abs(int(problem_str))
        problem_str = str(problem_id) # format begin with 00*
        find_str   = problem_str
        if problem_id < 100:
            find_str = "0" + problem_str
        if problem_id < 10:
            find_str = "0" + find_str
        dir_list = [f for f in listdir(target_dir) if isdir(join(target_dir, f))]
        for dir_name in dir_list:
            pattern = re.compile(r'^' + find_str)
            match = pattern.match(dir_name)
            if match:
                # print find_str
                t_exec_flag = True
                print dir_name
                mo = importlib.import_module(dir_name + '.' + dir_name)
                mo.exec_main()
        if not t_exec_flag:
            print "Not solved."

def exec_main():
    find_and_figure_out_problem()

if __name__ == '__main__':
    exec_main()