PaiGram/metadata/pool/pool.py

from metadata.pool.pool_100 import POOL_100
from metadata.pool.pool_200 import POOL_200
from metadata.pool.pool_301 import POOL_301
from metadata.pool.pool_302 import POOL_302
from metadata.pool.pool_500 import POOL_500


def get_pool_by_id(pool_type):
    if pool_type == 100:
        return POOL_100
    if pool_type == 200:
        return POOL_200
    if pool_type in [301, 400]:
        return POOL_301
    if pool_type == 302:
        return POOL_302
    if pool_type == 500:
        return POOL_500
    return None