from django import forms

import account.forms

class SignupForm(account.forms.SignupForm):

    first_name = forms.CharField()

    last_name = forms.CharField()

    email_confirm = forms.EmailField(label="Confirm Email")

    def __init__(self, *args, **kwargs):

        super(SignupForm, self).__init__(*args, **kwargs)

def reorder_fields(fields, order):

    """Reorder form fields by order, removing items not in order.

    >>> reorder_fields(

    ...     OrderedDict([('a', 1), ('b', 2), ('c', 3)]),

    ...     ['b', 'c', 'a'])

    OrderedDict([('b', 2), ('c', 3), ('a', 1)])

    """

    for key, v in fields.items():

        if key not in order:

            del fields[key]

        # fields is SortedDict

        fields.keyOrder.sort(key=lambda k: order.index(k[0]))

        return fields

    else:

        # fields is OrderedDict

        return OrderedDict(sorted(fields.items(), key=lambda k: order.index(k[0])))