mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 05:22:16 +00:00
136 lines
5.2 KiB
Python
136 lines
5.2 KiB
Python
from RIGS import models, forms
|
|
from django_ical.views import ICalFeed
|
|
from django.db.models import Q
|
|
from django.core.urlresolvers import reverse_lazy, reverse, NoReverseMatch
|
|
from django.utils import timezone
|
|
from django.conf import settings
|
|
|
|
import datetime, pytz
|
|
|
|
class CalendarICS(ICalFeed):
|
|
"""
|
|
A simple event calender
|
|
"""
|
|
#Metadata which is passed on to clients
|
|
product_id = 'RIGS'
|
|
title = 'RIGS Calendar'
|
|
timezone = settings.TIME_ZONE
|
|
file_name = "rigs.ics"
|
|
|
|
def get(self, *args, **kwargs):
|
|
timezone.activate(timezone.UTC)
|
|
return super(CalendarICS, self).get(*args, **kwargs)
|
|
|
|
|
|
def items(self):
|
|
#include events from up to 1 year ago
|
|
start = datetime.datetime.now() - datetime.timedelta(days=365)
|
|
filter = Q(start_date__gte=start) & ~Q(status=models.Event.CANCELLED)
|
|
|
|
return models.Event.objects.filter(filter).order_by('-start_date').select_related('person', 'organisation', 'venue', 'mic')
|
|
|
|
def item_title(self, item):
|
|
title = ''
|
|
|
|
# Prefix title with status (if it's a critical status)
|
|
if item.cancelled:
|
|
title += 'CANCELLED: '
|
|
|
|
if not item.is_rig:
|
|
title += 'NON-RIG: '
|
|
|
|
if item.dry_hire:
|
|
title += 'DRY HIRE: '
|
|
|
|
# Add the rig name
|
|
title += item.name
|
|
|
|
# Add the status
|
|
title += ' ('+str(item.get_status_display())+')'
|
|
|
|
return title
|
|
|
|
def item_start_datetime(self, item):
|
|
#set start date to the earliest defined time for the event
|
|
if item.meet_at:
|
|
startDateTime = item.meet_at
|
|
elif item.access_at:
|
|
startDateTime = item.access_at
|
|
elif item.has_start_time:
|
|
startDateTime = datetime.datetime.combine(item.start_date,item.start_time)
|
|
tz = pytz.timezone(settings.TIME_ZONE)
|
|
startDateTime = tz.normalize(tz.localize(startDateTime)).astimezone(pytz.timezone(self.timezone))
|
|
else:
|
|
startDateTime = item.start_date
|
|
|
|
return startDateTime
|
|
|
|
def item_end_datetime(self, item):
|
|
# Assume end is same as start
|
|
endDateTime = item.start_date
|
|
|
|
# If end date defined then use it
|
|
if item.end_date:
|
|
endDateTime = item.end_date
|
|
|
|
if item.has_start_time and item.has_end_time: # don't allow an event with specific end but no specific start
|
|
endDateTime = datetime.datetime.combine(endDateTime,item.end_time)
|
|
tz = pytz.timezone(settings.TIME_ZONE)
|
|
endDateTime = tz.normalize(tz.localize(endDateTime)).astimezone(pytz.timezone(self.timezone))
|
|
elif item.has_end_time: # if there's a start time specified then an end time should also be specified
|
|
endDateTime = datetime.datetime.combine(endDateTime+datetime.timedelta(days=1),datetime.time(00, 00))
|
|
#elif item.end_time: # end time but no start time - this is weird - don't think ICS will like it so ignoring
|
|
# do nothing
|
|
|
|
return endDateTime
|
|
|
|
def item_location(self,item):
|
|
return item.venue
|
|
|
|
def item_description(self, item):
|
|
# Create a nice information-rich description
|
|
# note: only making use of information available to "non-keyholders"
|
|
|
|
desc = 'Rig ID = '+str(item.pk)+'\n'
|
|
desc += 'Event = ' + item.name + '\n'
|
|
desc += 'Venue = ' + (item.venue.name if item.venue else '---') + '\n'
|
|
if item.is_rig and item.person:
|
|
desc += 'Client = ' + item.person.name + ( (' for '+item.organisation.name) if item.organisation else '') + '\n'
|
|
desc += 'Status = ' + str(item.get_status_display()) + '\n'
|
|
desc += 'MIC = ' + (item.mic.name if item.mic else '---') + '\n'
|
|
|
|
|
|
desc += '\n'
|
|
if item.meet_at:
|
|
desc += 'Crew Meet = ' + item.meet_at.strftime('%Y-%m-%d %H:%M') + (('('+item.meet_info+')') if item.meet_info else '---') + '\n'
|
|
if item.access_at:
|
|
desc += 'Access At = ' + item.access_at.strftime('%Y-%m-%d %H:%M') + '\n'
|
|
if item.start_date:
|
|
desc += 'Event Start = ' + item.start_date.strftime('%Y-%m-%d') + ((' '+item.start_time.strftime('%H:%M')) if item.has_start_time else '') + '\n'
|
|
if item.end_date:
|
|
desc += 'Event End = ' + item.end_date.strftime('%Y-%m-%d') + ((' '+item.end_time.strftime('%H:%M')) if item.has_end_time else '') + '\n'
|
|
|
|
desc += '\n'
|
|
if item.description:
|
|
desc += 'Event Description:\n'+item.description+'\n\n'
|
|
if item.notes:
|
|
desc += 'Notes:\n'+item.notes+'\n\n'
|
|
|
|
base_url = "https://pyrigs.nottinghamtec.co.uk"
|
|
desc += 'URL = '+base_url+str(item.get_absolute_url())
|
|
|
|
return desc
|
|
|
|
def item_link(self, item):
|
|
# Make a link to the event in the web interface
|
|
# base_url = "https://pyrigs.nottinghamtec.co.uk"
|
|
return item.get_absolute_url()
|
|
|
|
# def item_created(self, item): #TODO - Implement created date-time (using django-reversion?) - not really necessary though
|
|
# return ''
|
|
|
|
def item_updated(self, item): # some ical clients will display this
|
|
return item.last_edited_at
|
|
|
|
def item_guid(self, item): # use the rig-id as the ical unique event identifier
|
|
return item.pk |