import types

from decorator import decorator

from .user import vUser

from .schema import Offer

from .dbdefer import  dbdefer

from .pool import runInDBPool

from ..error import ValidateException

@dbdefer

def vRoute(session, value, field, typ, allowString=True, allowList=True):

    str_ = False

    ret = []

    if type(value) is types.StringType:

        if not allowString:

            raise ValidateException(field=field,msg='%s must be a list of routes.'%field)

        str_=True

        value=[value]

    elif not allowList:

        raise ValidateException(field=field,msg='%s must be a route - No list of routes.'%field)

    routes = [o.name for o in Offer.routes(session,typ)]

    providers =  [o.provider for o in Offer.providers(session,typ)]

    for v in value:

        if v not in routes and v not in providers and v != "default":

            raise ValidateException(field=field,msg='Route %s is not valid.'%v)

        if v not in ret:

            ret.append(v)

    if str_:

        return ret[0]

    return ret

@dbdefer

def vTyp(value,field, session):

    for typ in Offer.typs(session):

        if value == typ[0]:

            break

    else:

        raise ValidateException(field=field,msg='Typ %s is not valid.'%value)

    return value