class CurrentlyOpenView(TemplateView):
model = get_premises_model()
template_name = "openinghours/index.html"
def get_context_data(self, **kwargs):
context = super(CurrentlyOpenView, self).get_context_data(**kwargs)
context['location'] = self.model.objects.first()
return context
def opening_hours(location=None, concise=False):
"""
Creates a rendered listing of hours.
"""
template_name = 'openinghours/opening_hours_list.html'
days = [] # [{'hours': '9:00am to 5:00pm', 'name': u'Monday'}, {'hours...
# Without `location`, choose the first company.
if location:
ohrs = OpeningHours.objects.filter(company=location)
else:
try:
Location = utils.get_premises_model()
ohrs = Location.objects.first().openinghours_set.all()
except AttributeError:
raise Exception("You must define some opening hours"
" to use the opening hours tags.")
ohrs.order_by('weekday', 'from_hour')
for o in ohrs:
days.append({
'day_number': o.weekday,
'name': o.get_weekday_display(),
'from_hour': o.from_hour,
'to_hour': o.to_hour,
'hours': '%s%s to %s%s' % (
o.from_hour.strftime('%I:%M').lstrip('0'),
o.from_hour.strftime('%p').lower(),
o.to_hour.strftime('%I:%M').lstrip('0'),
o.to_hour.strftime('%p').lower()
)
})
open_days = [o.weekday for o in ohrs]
for day_number, day_name in WEEKDAYS:
if day_number not in open_days:
days.append({
'day_number': day_number,
'name': day_name,
'hours': 'Closed'
})
days = sorted(days, key=lambda k: k['day_number'])
if concise:
# [{'hours': '9:00am to 5:00pm', 'day_names': u'Monday to Friday'},
# {'hours':...
template_name = 'openinghours/opening_hours_list_concise.html'
concise_days = []
current_set = {}
for day in days:
if 'hours' not in current_set.keys():
current_set = {'day_names': [day['name']],
'hours': day['hours']}
elif day['hours'] != current_set['hours']:
concise_days.append(current_set)
current_set = {'day_names': [day['name']],
'hours': day['hours']}
else:
current_set['day_names'].append(day['name'])
concise_days.append(current_set)
for day_set in concise_days:
if len(day_set['day_names']) > 2:
day_set['day_names'] = '%s to %s' % (day_set['day_names'][0],
day_set['day_names'][-1])
elif len(day_set['day_names']) > 1:
day_set['day_names'] = '%s and %s' % (day_set['day_names'][0],
day_set['day_names'][-1])
else:
day_set['day_names'] = '%s' % day_set['day_names'][0]
days = concise_days
template = get_template(template_name)
return template.render(Context({'days': days}))
class OpeningHoursEditView(DetailView):
"""Powers editing UI supporting up to 2 time slots (sets) per day.
Models still support more slots via shell or admin UI.
This UI will delete and not recreate anything above 2 daily slots.
Inspired by Google local opening hours UI and earlier works.
"""
model = get_premises_model()
template_name = "openinghours/edit_base.html"
def form_prefix(self, day_n, slot_n):
"""Form prefix made up of day number and slot number.
- day number 1-7 for Monday to Sunday
- slot 1-2 typically morning and afternoon
"""
return "day%s_%s" % (day_n, slot_n)
def post(self, request, pk):
""" Clean the data and save opening hours in the database.
Old opening hours are purged before new ones are saved.
"""
location = self.get_object()
# open days, disabled widget data won't make it into request.POST
present_prefixes = [x.split('-')[0] for x in request.POST.keys()]
day_forms = OrderedDict()
for day_no, day_name in WEEKDAYS:
for slot_no in (1, 2):
prefix = self.form_prefix(day_no, slot_no)
# skip closed day as it would be invalid form due to no data
if prefix not in present_prefixes:
continue
day_forms[prefix] = (day_no, Slot(request.POST, prefix=prefix))
if all([day_form[1].is_valid()
for pre, day_form in day_forms.items()]):
OpeningHours.objects.filter(company=location).delete()
for prefix, day_form in day_forms.items():
day, form = day_form
opens, shuts = [
str_to_time(form.cleaned_data[x])
for x in ('opens', 'shuts')
]
if opens != shuts:
OpeningHours(from_hour=opens,
to_hour=shuts,
company=location,
weekday=day).save()
return redirect(request.path_info)
def get(self, request, pk):
""" Initialize the editing form
1. Build opening_hours, a lookup dictionary to populate the form
slots: keys are day numbers, values are lists of opening
hours for that day.
2. Build days, a list of days with 2 slot forms each.
3. Build form initials for the 2 slots padding/trimming
opening_hours to end up with exactly 2 slots even if it's
just None values.
"""
location = self.get_object()
two_sets = False
closed = None
opening_hours = {}
for o in OpeningHours.objects.filter(company=location):
opening_hours.setdefault(o.weekday, []).append(o)
days = []
for day_no, day_name in WEEKDAYS:
if day_no not in opening_hours.keys():
if opening_hours:
closed = True
ini1, ini2 = [None, None]
else:
closed = False
ini = [{
'opens': time_to_str(oh.from_hour),
'shuts': time_to_str(oh.to_hour)
} for oh in opening_hours[day_no]]
ini += [None] * (2 - len(ini[:2])) # pad
ini1, ini2 = ini[:2] # trim
if ini2:
two_sets = True
days.append({
'name':
day_name,
'number':
day_no,
'slot1':
Slot(prefix=self.form_prefix(day_no, 1), initial=ini1),
'slot2':
Slot(prefix=self.form_prefix(day_no, 2), initial=ini2),
'closed':
closed
})
return render(request, self.template_name, {
'days': days,
'two_sets': two_sets,
'location': location,
})
def opening_hours(location=None, concise=False):
"""
Creates a rendered listing of hours.
"""
template_name = 'openinghours/opening_hours_list.html'
days = [] # [{'hours': '9:00am to 5:00pm', 'name': u'Monday'}, {'hours...
# Without `location`, choose the first company.
if location:
ohrs = OpeningHours.objects.filter(company=location)
else:
try:
Location = utils.get_premises_model()
ohrs = Location.objects.first().openinghours_set.all()
except AttributeError:
raise Exception("You must define some opening hours"
" to use the opening hours tags.")
ohrs.order_by('weekday', 'from_hour')
for o in ohrs:
days.append({
'day_number':
o.weekday,
'name':
o.get_weekday_display(),
'from_hour':
o.from_hour,
'to_hour':
o.to_hour,
'hours':
'%s%s to %s%s' % (o.from_hour.strftime('%I:%M').lstrip('0'),
o.from_hour.strftime('%p').lower(),
o.to_hour.strftime('%I:%M').lstrip('0'),
o.to_hour.strftime('%p').lower())
})
open_days = [o.weekday for o in ohrs]
for day_number, day_name in WEEKDAYS:
if day_number not in open_days:
days.append({
'day_number': day_number,
'name': day_name,
'hours': 'Closed'
})
days = sorted(days, key=lambda k: k['day_number'])
if concise:
# [{'hours': '9:00am to 5:00pm', 'day_names': u'Monday to Friday'},
# {'hours':...
template_name = 'openinghours/opening_hours_list_concise.html'
concise_days = []
current_set = {}
for day in days:
if 'hours' not in current_set.keys():
current_set = {
'day_names': [day['name']],
'hours': day['hours']
}
elif day['hours'] != current_set['hours']:
concise_days.append(current_set)
current_set = {
'day_names': [day['name']],
'hours': day['hours']
}
else:
current_set['day_names'].append(day['name'])
concise_days.append(current_set)
for day_set in concise_days:
if len(day_set['day_names']) > 2:
day_set['day_names'] = '%s to %s' % (day_set['day_names'][0],
day_set['day_names'][-1])
elif len(day_set['day_names']) > 1:
day_set['day_names'] = '%s and %s' % (day_set['day_names'][0],
day_set['day_names'][-1])
else:
day_set['day_names'] = '%s' % day_set['day_names'][0]
days = concise_days
template = get_template(template_name)
return template.render(Context({'days': days}))
class OpeningHoursEditView(DetailView, UpdateView):
"""
Powers editing UI supporting up to 2 time slots (sets) per day.
Models still support more slots via shell or admin UI.
This UI will delete and not recreate anything above 2 daily slots.
Inspired by Google local opening hours UI and earlier works.
"""
model = get_premises_model()
template_name = "openinghours/edit_base.html"
include_form_actions = True
def form_prefix(self, day_n, slot_n):
"""Form prefix made up of day number and slot number.
- day number 1-7 for Monday to Sunday
- slot 1-2 typically morning and afternoon
"""
return "day%s_%s" % (day_n, slot_n)
def get_success_url(self):
"""
Returns the supplied success URL.
"""
if not self.success_url:
self.success_url = self.request.path_info
if self.success_url:
# Forcing possible reverse_lazy evaluation
url = force_text(self.success_url)
else:
url = self.request.path_info
return url
def get_context_data(self, **kwargs):
"""
Initialize the editing form
1. Build opening_hours, a lookup dictionary to populate the form
slots: keys are day numbers, values are lists of opening
hours for that day.
2. Build days, a list of days with 2 slot forms each.
3. Build form initials for the 2 slots padding/trimming
opening_hours to end up with exactly 2 slots even if it's
just None values.
"""
self.object = self.get_object()
context = super(OpeningHoursEditView, self).get_context_data(**kwargs)
two_sets = False
closed = None
opening_hours = {}
for o in OpeningHours.objects.filter(company=self.object):
opening_hours.setdefault(o.weekday, []).append(o)
days = []
for day_no, day_name in WEEKDAYS:
if day_no not in opening_hours.keys():
closed = True
ini1, ini2 = [None, None]
else:
closed = False
ini = [{'opens': time_to_str(oh.from_hour),
'shuts': time_to_str(oh.to_hour)}
for oh in opening_hours[day_no]]
ini += [None] * (2 - len(ini[:2])) # pad
ini1, ini2 = ini[:2] # trim
if ini2:
two_sets = True
days.append({
'name': day_name,
'number': day_no,
'slot1': Slot(prefix=self.form_prefix(day_no, 1), initial=ini1),
'slot2': Slot(prefix=self.form_prefix(day_no, 2), initial=ini2),
'closed': closed
})
context['days'] = days
context['two_sets'] = two_sets
context['location'] = self.object
context['include_form_actions'] = self.include_form_actions
closing_rules_queryset = ClosingRules.objects.filter(company=self.object.pk)
extra = 2
if closing_rules_queryset.count() == 0:
extra = 3
formset = modelformset_factory(ClosingRules, ClosingRulesForm, extra=extra)
context['formset'] = formset(queryset=closing_rules_queryset)
return context
def get(self, request, *args, **kwargs):
context = self.get_context_data(object=self.object)
return self.render_to_response(context)
def process_post(self, request):
"""
Clean the data and save opening hours in the database.
Old opening hours are purged before new ones are saved.
"""
location = self.get_object()
# open days, disabled widget data won't make it into request.POST
present_prefixes = [x.split('-')[0] for x in request.POST.keys()]
day_forms = OrderedDict()
for day_no, day_name in WEEKDAYS:
for slot_no in (1, 2):
prefix = self.form_prefix(day_no, slot_no)
# skip closed day as it would be invalid form due to no data
if prefix not in present_prefixes:
continue
day_forms[prefix] = (day_no, Slot(request.POST, prefix=prefix))
if all([day_form[1].is_valid() for pre, day_form in day_forms.items()]):
OpeningHours.objects.filter(company=location).delete()
for prefix, day_form in day_forms.items():
day, form = day_form
opens, shuts = [str_to_time(form.cleaned_data[x])
for x in ('opens', 'shuts')]
if opens != shuts:
OpeningHours(from_hour=opens, to_hour=shuts,
company=location, weekday=day).save()
# assume that forms are validated
ClosingRulesFormSet = modelformset_factory(ClosingRules, ClosingRulesForm)
formset = ClosingRulesFormSet(request.POST)
for form in formset.forms:
if form.is_valid():
closing_rule = form.save(commit=False)
closing_rule.start = construct_tz_aware_time(form.cleaned_data['start_date'],
form.cleaned_data['start_time'],
location.timezone)
closing_rule.end = construct_tz_aware_time(form.cleaned_data['end_date'],
form.cleaned_data['end_time'],
location.timezone)
closing_rule.company = location
closing_rule.save()
elif form.is_bound:
ClosingRules.objects.filter(pk=form.instance.pk).delete()
def post(self, request, *args, **kwargs):
self.process_post(request)
success_url = self.get_success_url()
return HttpResponseRedirect(success_url)