|
from django import forms
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from .models import Announce, Category, Broadcast, channel_choices
|
|
|
|
|
|
class AnnounceForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
model = Announce
|
|
fields = '__all__'
|
|
widgets = {
|
|
'publication_time': forms.TextInput(attrs={'class': 'datetimepicker',
|
|
'readonly': True}),
|
|
'expiration_time': forms.TextInput(attrs={'class': 'datetimepicker',
|
|
'readonly': True})
|
|
}
|
|
|
|
def save(self, *args, **kwargs):
|
|
instance = super(AnnounceForm, self).save(*args, **kwargs)
|
|
if instance:
|
|
Broadcast.objects.get_or_create(announce=instance)
|
|
return instance
|
|
|
|
|
|
class CategoryForm(forms.ModelForm):
|
|
class Meta:
|
|
fields = ('name', )
|
|
model = Category
|