2016-03-01 27 views
5

Se ho una classe come:Fai namedtuple accettare kwargs

class Person(object): 
    def __init__(self, name, **kwargs): 
     self.name = name 

p = Person(name='joe', age=25) # age is ignored 

params extra vengono ignorati. Ma se ho un namedtuple, vado a prendere `argomento chiave inaspettato:

from collections import namedtuple 

Person = namedtuple('Person', 'name') 
p = Person(name='joe', age=25) 

# Traceback (most recent call last): 
# File "python", line 1, in <module> 
# TypeError: __new__() got an unexpected keyword argument 'age' 

Come posso fare namedtuple accettare kwargs così posso passare argomenti extra in modo sicuro?

+1

Non puoi farlo. –

+0

Gli indizi nel nome. –

+0

@Alex, che cosa ha a che fare con la mia domanda? – norbertpy

risposta

7

La seguente sessione nell'interprete mostra una p soluzione OSSIBILI per fissare il vostro problema:

Python 3.5.0 (v3.5.0:374f501f4567, Sep 13 2015, 02:27:37) [MSC v.1900 64 bit (AMD64)] on win32 
Type "copyright", "credits" or "license()" for more information. 
>>> import collections 
>>> class Person(collections.namedtuple('base', 'name')): 
    __slots__ =() 
    def __new__(cls, *args, **kwargs): 
     for key in tuple(kwargs): 
      if key not in cls._fields: 
       del kwargs[key] 
     return super().__new__(cls, *args, **kwargs) 


>>> p = Person(name='joe', age=25) 
>>> p 
Person(name='joe') 
>>> 

Alternativa:

Dal momento che si preferisce avere una soluzione più semplice, si potrebbe trovare il prossimo programma più a proprio piacimento:

#! /usr/bin/env python3 
import collections 


def main(): 
    Person = namedtuple('Person', 'name') 
    p = Person(name='joe', age=25) 
    print(p) 


def namedtuple(typename, field_names, verbose=False, rename=False): 
    base = collections.namedtuple('Base', field_names, verbose, rename) 
    return type(typename, (base,), { 
     '__slots__':(), 
     '__new__': lambda cls, *args, **kwargs: base.__new__(cls, *args, **{ 
      key: value for key, value in kwargs.items() 
      if key in base._fields})}) 


if __name__ == '__main__': 
    main() 
+0

Bene, se voglio scrivere un metodo '__new__' per ogni classe di cui ho bisogno, allora farò solo una classe come il mio primo esempio. Ma grazie per la tua risposta. – norbertpy

+0

@norbertpy Ho aggiunto una nuova implementazione di 'namedtuple' per risolvere automaticamente questo problema. Fa la stessa cosa del primo esempio, ma gestisce la modifica a '__new__' in modo da evitare di scrivere il codice da soli. –

+0

Non creare sottoclassi di namedtuple senza '__slots__'. Spreca memoria. – ppperry

6

Non è bello:

p = Person(*(dict(name='joe', age=25)[k] for k in Person._fields)) 
2

E 'possibile avvolgere il costruttore Person classe di ignorare gli argomenti che non sono definiti come campi della Person namedtuple:

from collections import namedtuple 
Person = namedtuple('Person', 'name') 

def make_person(*args, **kwargs): 
    person_args = {} 

    # process positional args 
    if len(args) > len(Person._fields): 
     msg = "Person() takes %d positional arguments but %d were given" % (len(Person._fields), len(args)) 
     raise TypeError(msg) 
    for arg_name, arg_value in zip(Person._fields, args): 
     person_args[arg_name] = arg_value 

    # process keyword args 
    for arg_name, arg_value in kwargs.items(): 
     try: 
      i = Person._fields.index(arg_name) 
     except ValueError: 
      pass # ignore arguments not defined as Person fields 
     else: 
      if arg_name in person_args: 
       msg = "make_person() got multiple values for argument " + repr(arg_name) 
       raise TypeError(msg) 
      person_args[arg_name] = arg_value 

    if len(person_args) != len(Person._fields): 
     msg = "Person() requires additional arguments: " 
     msg += ", ".join([repr(x) for x in Person._fields if x not in person_args]) 
     raise TypeError(msg) 
    return Person(*[person_args[x] for x in Person._fields]) 

Tutto ciò premesso:

>>> make_person('a') 
Person(name='a') 
>>> make_person('a', b='b') 
Person(name='a') 
>>> make_person('a', name='b') 
TypeError: make_person() got multiple values for argument 'name' 
>>> make_person(b='b') 
TypeError: Person() requires additional arguments: 'name' 
>>> make_person(1, 2) 
TypeError: Person() takes 1 positional arguments but 2 were given 
+0

A differenza della risposta di Paolo, questa risposta supporta la combinazione di argomenti posizionali e parole chiave in modo corretto e gestisce correttamente tutti i casi limite con messaggi di errore informativi. – taleinat