forked from gergely/calendar-social
160 lines
4.2 KiB
Python
160 lines
4.2 KiB
Python
from datetime import datetime, timedelta
|
|
from functools import wraps
|
|
|
|
from flask_babelex import lazy_gettext as _
|
|
|
|
from . import CalendarSystem
|
|
|
|
|
|
def to_timestamp(func):
|
|
@wraps(func)
|
|
def decorator(*args, **kwargs):
|
|
return func(*args, **kwargs).timestamp()
|
|
|
|
return decorator
|
|
|
|
|
|
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'),
|
|
)
|
|
|
|
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
|
|
|
|
@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:
|
|
ts = self.prev_year.replace(month=12)
|
|
else:
|
|
ts = self.timestamp.replace(month=self.timestamp.month - 1)
|
|
|
|
return self.month_names[ts.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:
|
|
ts = self.prev_year.replace(month=1)
|
|
else:
|
|
ts = self.timestamp.replace(month=self.timestamp.month + 1)
|
|
|
|
return self.month_names[ts.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
|
|
|
|
def day_events(self, date, user=None):
|
|
from ..models import Event
|
|
|
|
events = Event.query
|
|
|
|
if user:
|
|
events = events.filter(Event.user == user)
|
|
|
|
start_timestamp = date.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
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)))
|
|
|
|
return events
|