¿Patrón elegante para args de palabra clave mutuamente excluyentes?

Question

A veces en mi código tengo una función que puede tomar un argumento de una de dos maneras. Algo así como:

def func(objname=None, objtype=None): 
    if objname is not None and objtype is not None: 
     raise ValueError("only 1 of the ways at a time") 
    if objname is not None: 
     obj = getObjByName(objname) 
    elif objtype is not None: 
     obj = getObjByType(objtype) 
    else: 
     raise ValueError("not given any of the ways") 

    doStuffWithObj(obj) 

¿Hay alguna forma más elegante de hacer esto? ¿Qué pasa si el arg puede venir en una de tres maneras? Si los tipos son distintos, podría hacer:

def func(objnameOrType): 
    if type(objnameOrType) is str: 
     getObjByName(objnameOrType) 
    elif type(objnameOrType) is type: 
     getObjByType(objnameOrType) 
    else: 
     raise ValueError("unk arg type: %s" % type(objnameOrType)) 

Pero, ¿y si no lo son? Esta alternativa parece tonta:

def func(objnameOrType, isName=True): 
    if isName: 
     getObjByName(objnameOrType) 
    else: 
     getObjByType(objnameOrType) 

causa, entonces usted tiene que llamar así func(mytype, isName=False) lo cual es raro.

Answer 1

¿Y si uso algo así como un patrón de comando de envío:

def funct(objnameOrType): 
    dispatcher = {str: getObjByName, 
       type1: getObjByType1, 
       type2: getObjByType2} 
    t = type(objnameOrType) 
    obj = dispatcher[t](objnameOrType) 
    doStuffWithObj(obj) 

donde type1, type2, etc, son tipos reales Python (por ejemplo, int, float, etc).

Answer 2

Una forma para que sea un poco más corto es

def func(objname=None, objtype=None): 
    if [objname, objtype].count(None) != 1: 
     raise TypeError("Exactly 1 of the ways must be used.") 
    if objname is not None: 
     obj = getObjByName(objname) 
    else: 
     obj = getObjByType(objtype) 

Todavía no he decidido si iba a llamar a este "elegante".

Tenga en cuenta que debe generar un TypeError si se proporcionó una cantidad incorrecta de argumentos, no un ValueError.

Answer 3

suena como tiene que ir a https://codereview.stackexchange.com/

De todos modos, manteniendo la misma interfaz, que puede probar

arg_parsers = { 
    'objname': getObjByName, 
    'objtype': getObjByType, 
    ... 
} 
def func(**kwargs): 
    assert len(kwargs) == 1 # replace this with your favorite exception 
    (argtypename, argval) = next(kwargs.items()) 
    obj = arg_parsers[argtypename](argval) 
    doStuffWithObj(obj) 

o simplemente crear 2 funciones?

def funcByName(name): ... 
def funcByType(type_): ... 

Answer 4

Parece que usted está buscando function overloading, que no está implementado en Python 2. En Python 2, la solución es casi tan bueno como se puede esperar obtener.

Probablemente se podría pasar por alto la adicional problema argumento al permitir que su función para procesar varios objetos y devolver un generador:

import types 

all_types = set([getattr(types, t) for t in dir(types) if t.endswith('Type')]) 

def func(*args): 
    for arg in args: 
     if arg in all_types: 
      yield getObjByType(arg) 
     else: 
      yield getObjByName(arg) 

prueba:

>>> getObjByName = lambda a: {'Name': a} 
>>> getObjByType = lambda a: {'Type': a} 
>>> list(func('IntType')) 
[{'Name': 'IntType'}] 
>>> list(func(types.IntType)) 
[{'Type': <type 'int'>}] 

Answer 5

Por lo que vale, el mismo género de cosas suceden en las bibliotecas estándar; ver, por ejemplo, el comienzo de GzipFile en gzip.py (mostrado aquí con docstrings retirados):

class GzipFile: 
    myfileobj = None 
    max_read_chunk = 10 * 1024 * 1024 # 10Mb 
    def __init__(self, filename=None, mode=None, 
       compresslevel=9, fileobj=None): 
     if mode and 'b' not in mode: 
      mode += 'b' 
     if fileobj is None: 
      fileobj = self.myfileobj = __builtin__.open(filename, mode or 'rb') 
     if filename is None: 
      if hasattr(fileobj, 'name'): filename = fileobj.name 
      else: filename = '' 
     if mode is None: 
      if hasattr(fileobj, 'mode'): mode = fileobj.mode 
      else: mode = 'rb' 

Por supuesto, esto acepta tanto filename y fileobj palabras clave y define un comportamiento en particular en el caso de que recibe ambos; pero el enfoque general parece bastante idéntico.