PamGram/metadata/pool/pool.py

from metadata.pool.pool_1 import POOL_1
from metadata.pool.pool_2 import POOL_2
from metadata.pool.pool_11 import POOL_11
from metadata.pool.pool_12 import POOL_12


def get_pool_by_id(pool_type):
    if pool_type == 1:
        return POOL_1
    elif pool_type == 2:
        return POOL_2
    if pool_type == 11:
        return POOL_11
    if pool_type == 12:
        return POOL_12
    return None