Obsługa jako argument listy lub pojedynczej liczby całkowitej

Question

Funkcja powinna wybierać wiersze w tabeli na podstawie nazwy wiersza (kolumna 2 w tym przypadku). Powinien być w stanie wziąć jedną nazwę lub listę nazw jako argumenty i poprawnie nimi zarządzać.

To jest to, co mam teraz, ale idealnie nie byłoby to kod duplikowane i coś wyjątkami byłyby wykorzystywane w sposób inteligentny, aby wybrać odpowiedni sposób obsłużyć argument wejściowy:

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

Właściwie zgadzam się z Andrew Hare powyżej, wystarczy podać listę z jednym elementem.

Ale jeśli naprawdę musisz zaakceptować listę nieistniejącą, to może zamieniasz ją w listę?

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) 

Kara za wydajność robi „w” na liście pojedynczego elementu nie może być wysoka :-) Ale to ma podkreślić jedną rzecz, może warto rozważyć, jeśli robi „to_select "lista może być długa: rozważ rzucenie jej do zestawu, aby wyszukiwanie było wydajniejsze.

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

zrobiłbym tylko:

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) 

i oczekuj, że argument będzie zawsze listą - nawet jeśli jest to tylko lista jednego elementu.

Pamiętaj:

Łatwiej prosić o przebaczenie niż pozwolenie.

Answer 3

Można przedefiniować swoją funkcję podjąć dowolną liczbę argumentów, tak:

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

Następnie można przekazać jeden argument takiego:

select_rows('abc') 

wielu argumentów, takich jak ten:

select_rows('abc', 'def') 

A jeśli już masz t:

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

Answer 4

pójdę wraz z wersją Sharkey, ale użyć trochę więcej kaczka wpisując:

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) 

ta ma tę zaletę w pracy z dowolnego obiektu który obsługuje operator in. Również poprzednia wersja, jeśli zostanie podana krotka lub inna sekwencja, po prostu zapakuje ją na listę. Wadą jest to, że za stosowanie wyjątków jest ograniczona wydajność.