示例#1
0
文件: forms.py 项目: wbrefvem/worklog
    def __init__(self, *args, **kwargs):
        user = kwargs.pop("user")
        super(WorkItemForm,self).__init__(*args,**kwargs)

        queryset = Job.get_jobs_open_on(datetime.date.today())

        queryset = queryset.filter(Q(available_all_users=True) | Q(users__id=user.id)).distinct()
        queryset = queryset.order_by('name')
        self.fields["job"].queryset = queryset

        self.fields["hours"].widget.attrs['class'] = 'form-control'
        self.fields["text"].widget.attrs['class'] = 'form-control'

        self.fields["hours"].widget.attrs['placeholder'] = 'Hours Worked'
        self.fields["text"].widget.attrs['placeholder'] = 'Work Description'

        self.fields["text"].widget.attrs['rows'] = '6'

        if args: 
            data = args[0]
        else:
            data = kwargs.get('data')

        if data:
            repo_id = data.get('repo')
            if repo_id:
                repo = Repo.objects.get(github_id=repo_id)
                self.fields["issue"].queryset = Issue.objects.filter(repo=repo)