126 lines
4.2 KiB
Python
126 lines
4.2 KiB
Python
from django.db import models
|
|
from django.utils.translation import ugettext as _
|
|
from django.contrib.auth.models import User, Permission
|
|
|
|
|
|
from django.db.models import Q
|
|
|
|
from datetime import datetime
|
|
from location_field.models import PlainLocationField
|
|
|
|
|
|
class Event ( models.Model ):
|
|
|
|
EVENT_TYPES = (
|
|
( 'Reh', _('Rehearsal') ),
|
|
( 'Conc', _('Concert') ),
|
|
( 'Party', _('Party') ),
|
|
( 'Travel', _('Travel') ),
|
|
( 'Option', _('Option') ),
|
|
)
|
|
|
|
type = models.CharField( max_length=6, choices=EVENT_TYPES, default='Reh', verbose_name= _("type") )
|
|
short_desc = models.CharField( null=True, max_length=100, blank = True, verbose_name= _("Short Description") )
|
|
location = models.TextField( blank=True, verbose_name=_("location") )
|
|
map_location = PlainLocationField(blank=True, based_field = location, zoom=7, verbose_name=_("Location on map") )
|
|
desc = models.TextField( blank=True, verbose_name=_("description") )
|
|
|
|
date = models.DateField( verbose_name= _("date") )
|
|
time = models.TimeField( null=True, blank=True, verbose_name = _("time") )
|
|
meeting_time = models.TimeField( null=True, blank=True, verbose_name = _("meeting_time") )
|
|
|
|
end_date = models.DateField( null=True, blank = True, verbose_name = _("End Date") )
|
|
|
|
participants = models.ManyToManyField( User, through='EventParticipation', verbose_name=_("participants") )
|
|
|
|
def __unicode__(self):
|
|
return self.title
|
|
|
|
|
|
def save(self, *args, **kwargs):
|
|
# Call the "real" save() method
|
|
super(Event, self).save(*args, **kwargs)
|
|
|
|
# Create a "Don't Know" participation for each Musician
|
|
for u in User.objects.all():
|
|
if not u in self.participants.all():
|
|
EventParticipation.objects.create( event=self, user = u, status='?', comment = '' )
|
|
|
|
@property
|
|
def title(self):
|
|
res = self.get_type_display()
|
|
if ( self.short_desc ):
|
|
res += " (" + self.short_desc + ") "
|
|
|
|
return res
|
|
|
|
@property
|
|
def displaytime(self):
|
|
if self.meeting_time is None or self.meeting_time == "":
|
|
return self.time
|
|
else:
|
|
return self.meeting_time
|
|
|
|
|
|
|
|
|
|
class EventParticipation( models.Model ):
|
|
OPTIONS = ( ('?' , _('?' )),
|
|
('Yes', _('Yes')),
|
|
('No' , _('No' ))
|
|
)
|
|
|
|
event = models.ForeignKey( Event, verbose_name=_("event") )
|
|
user = models.ForeignKey( User, verbose_name=_("user") )
|
|
status = models.CharField ( max_length=3, choices = OPTIONS, default='?', verbose_name=_("status") )
|
|
comment = models.CharField ( max_length=64, blank=True, verbose_name=_("comment") )
|
|
|
|
def get_username(self):
|
|
return self.user.username
|
|
|
|
@staticmethod
|
|
def hasUserSetParticipationForAllEvents( user ):
|
|
if not EventParticipation.isMember(user):
|
|
return True
|
|
|
|
futurePart = EventParticipation.objects.filter( event__date__gte = datetime.now() )
|
|
|
|
maybeObjects = futurePart.filter( user = user ).filter( status = '?' )
|
|
if len( maybeObjects ) > 0:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
@staticmethod
|
|
def isMember( user ):
|
|
return user.has_perm('eventplanner.member')
|
|
|
|
@staticmethod
|
|
def isAdmin( user ):
|
|
return user.has_perm('eventplanner.admin')
|
|
|
|
@staticmethod
|
|
def members():
|
|
perm = Permission.objects.get( codename='member' )
|
|
f = User.objects.filter(Q(groups__permissions=perm) | Q(user_permissions=perm) ).distinct()
|
|
return f.order_by('musician__position')
|
|
|
|
|
|
@staticmethod
|
|
def get_or_create( user , event ):
|
|
try:
|
|
result = EventParticipation.objects.get( event = event, user = user )
|
|
except EventParticipation.DoesNotExist:
|
|
result = EventParticipation.objects.create( event = event, user = user, status='?', comment = '' )
|
|
|
|
return result
|
|
|
|
class Meta:
|
|
unique_together = ("event", "user")
|
|
permissions = (
|
|
("admin", _("Admin") ),
|
|
("member", _("Member") ),
|
|
)
|
|
|
|
|
|
|