Maneja una lista o entero simple como argumento

Question

Una función debe seleccionar filas en una tabla según el nombre de la fila (columna 2 en este caso). Debería poder tomar un solo nombre o una lista de nombres como argumentos y manejarlos correctamente.

Esto es lo que tengo ahora, pero lo ideal sería que no habría este código duplicado y algo así como las excepciones son utilizados de forma inteligente para elegir el camino correcto para manejar el argumento de entrada:

def select_rows(to_select): 
    # For a list 
    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in to_select: 
      table.selectRow(row) 
    # For a single integer 
    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() == to_select: 
      table.selectRow(row) 

Answer 1

En realidad estoy de acuerdo con Andrew Hare anterior, solo pase una lista con un solo elemento.

Pero si realmente debe aceptar una lista no, ¿qué tal si la convierte en una lista en ese caso?

def select_rows(to_select): 
    if type(to_select) is not list: to_select = [ to_select ] 

    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in to_select: 
      table.selectRow(row) 

La penalización de rendimiento para hacer 'en' en una lista de un solo artículo no es probable que sea alta :-) Pero eso sí señala otra cosa es posible que desee considerar hacer si su 'to_select La lista puede ser larga: considere enviarla a un conjunto para que las búsquedas sean más eficientes.

def select_rows(to_select): 
    if type(to_select) is list: to_select = set(to_select) 
    elif type(to_select) is not set: to_select = set([to_select]) 

    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in to_select: 
      table.selectRow(row) 

----- N

Answer 2

Haría sólo esto:

def select_rows(to_select): 
    # For a list 
    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in to_select: 
      table.selectRow(row) 

y esperamos que el argumento siempre sea una lista, incluso si es solo una lista de un elemento.

Recuerde:

Es más fácil pedir perdón que permiso.

Answer 3

Se podría volver a definir su función para tomar cualquier número de argumentos, así:

def select_rows(*arguments): 
    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in arguments: 
      table.selectRow(row) 

entonces puede pasar un solo argumento así:

select_rows('abc') 

múltiples argumentos como este:

select_rows('abc', 'def') 

Y si ya tiene una lis t:

items = ['abc', 'def'] 
select_rows(*items) 

Answer 4

me gustaría ir junto con la versión de Sharkey, pero el uso de un poco más de pato escribiendo:

def select_rows(to_select): 
    try: 
     len(to_select) 
    except TypeError: 
     to_select = [to_select] 

    for row in range(0, table.numRows()): 
     if _table.item(row, 1).text() in to_select: 
      table.selectRow(row) 

Esto tiene la ventaja de trabajar con cualquier objeto que admite el operador in. Además, la versión anterior, si se le da una tupla u otra secuencia, simplemente la incluiría en una lista. La desventaja es que hay una cierta penalización de rendimiento por usar el manejo de excepciones.