from django import forms


class UserinfoForm(forms.Form):

    industry = forms.CharField(required=False, max_length=64)
    job_title = forms.CharField(required=False, max_length=64)
    tidb_relation = forms.CharField(required=False, max_length=64)
    newsql_usage_stage = forms.CharField(required=False, max_length=64)
    want_try_tidb_version = forms.CharField(required=False, max_length=64)
    tried_tidb_version = forms.CharField(required=False, max_length=64)
    accept_edm = forms.BooleanField(required=False)

    def __init__(self, user, *args, **kwargs):
        self.user = user
        super(UserinfoForm, self).__init__(*args, **kwargs)

    def save(self):
        data = self.cleaned_data
        self.user.industry = data.get('industry')
        self.user.job_title = data.get('job_title')
        self.user.tidb_relation = data.get('tidb_relation')
        self.user.newsql_usage_stage = data.get('newsql_usage_stage')
        self.user.want_try_tidb_version = data.get('want_try_tidb_version')
        self.user.tried_tidb_version = data.get('tried_tidb_version')
        self.user.accept_edm = data.get('accept_edm')
        self.user.save()