calendar-social/calsocial/calendar_system/gregorian.py

162 lines
4.3 KiB
Python
Raw Normal View History

2018-06-25 07:01:13 +00:00
from datetime import datetime, timedelta
from functools import wraps
from flask_babelex import lazy_gettext as _
2018-06-25 07:01:13 +00:00
from . import CalendarSystem
def to_timestamp(func):
@wraps(func)
def decorator(*args, **kwargs):
return func(*args, **kwargs).timestamp()
return decorator
2018-06-25 07:01:13 +00:00
class GregorianCalendar(CalendarSystem):
__name__ = _('Gregorian')
__has_months__ = True
START_DAY = 0
END_DAY = 7
month_names = (
_('January'),
_('February'),
_('March'),
_('April'),
_('May'),
_('June'),
_('July'),
_('August'),
_('September'),
_('October'),
_('November'),
_('December'),
)
2018-06-25 07:01:13 +00:00
day_names = (
_('Monday'),
_('Tuesday'),
_('Wednesday'),
_('Thursday'),
_('Friday'),
_('Saturday'),
_('Sunday'),
)
def __init__(self, timestamp):
self.timestamp = datetime.utcfromtimestamp(timestamp)
@property
def month(self):
return self.timestamp.strftime("%B, %Y")
@property
def days(self):
day_list = []
start_day = self.timestamp.replace(day=1)
while start_day.weekday() > self.START_DAY:
start_day -= timedelta(days=1)
day_list.append(start_day)
current_day = start_day
while current_day.weekday() < self.END_DAY and current_day.month <= self.timestamp.month:
current_day += timedelta(days=1)
day_list.append(current_day)
return day_list
2018-06-29 14:14:35 +00:00
@property
@to_timestamp
def prev_year(self):
return self.timestamp.replace(year=self.timestamp.year - 1)
@property
def prev_year_year(self):
return self.timestamp.replace(year=self.timestamp.year - 1).year
@property
@to_timestamp
def prev_month(self):
if self.timestamp.month == 1:
return self.prev_year.replace(month=12)
return self.timestamp.replace(month=self.timestamp.month - 1)
@property
def prev_month_name(self):
if self.timestamp.month == 1:
timestamp = self.prev_year.replace(month=12)
else:
timestamp = self.timestamp.replace(month=self.timestamp.month - 1)
return self.month_names[timestamp.month - 1]
@property
@to_timestamp
def next_month(self):
if self.timestamp.month == 12:
return self.next_year.replace(month=1)
return self.timestamp.replace(month=self.timestamp.month + 1)
@property
def next_month_name(self):
if self.timestamp.month == 12:
timestamp = self.prev_year.replace(month=1)
else:
timestamp = self.timestamp.replace(month=self.timestamp.month + 1)
return self.month_names[timestamp.month - 1]
@property
@to_timestamp
def next_year(self):
return self.timestamp.replace(year=self.timestamp.year + 1)
@property
def next_year_year(self):
return self.timestamp.replace(year=self.timestamp.year + 1).year
@property
def has_today(self):
now = datetime.utcnow()
month_start_timestamp = self.timestamp.replace(day=1,
hour=0,
minute=0,
second=0,
microsecond=0)
if self.timestamp.month == 12:
next_month = 1
else:
next_month = self.timestamp.month + 1
month_end_timestamp = month_start_timestamp.replace(month=next_month)
return now >= month_start_timestamp and now < month_end_timestamp
@staticmethod
def day_events(date, user=None):
from ..models import Event
2018-06-29 14:14:35 +00:00
events = Event.query
if user:
events = events.filter(Event.user == user)
start_timestamp = date.replace(hour=0, minute=0, second=0, microsecond=0)
2018-06-29 14:14:35 +00:00
end_timestamp = start_timestamp + timedelta(days=1)
events = events.filter(((Event.start_time >= start_timestamp) & (Event.start_time < end_timestamp)) |
((Event.end_time >= start_timestamp) & (Event.end_time < end_timestamp))) \
.order_by('start_time', 'end_time')
2018-06-29 14:14:35 +00:00
return events