Shiloh/brotherton-erpnext
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

timezone manipulation using pytz

Browse Source
This commit is contained in:
Pranav Nachanekar 2019-10-03 11:58:02 +05:30
parent 1dcedb5054
commit 93670fedda
1 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
erpnext/www/book-appointment/index.py
View File

@ -1,6 +1,7 @@
import frappe import frappe
import datetime import datetime
import json import json
import pytz
WEEKDAYS = ["Monday", "Tuesday", "Wednesday", WEEKDAYS = ["Monday", "Tuesday", "Wednesday",
@ -24,20 +25,26 @@ def get_holiday_list(holiday_list_name):
@frappe.whitelist(allow_guest=True) @frappe.whitelist(allow_guest=True)
def get_timezones(): def get_timezones():
timezones = frappe.get_list('Timezone', fields='*') timezones = frappe.get_list('Timezone', fields='*')
return timezones return pytz.all_timezones
@frappe.whitelist(allow_guest=True) @frappe.whitelist(allow_guest=True)
def get_appointment_slots(date, timezone): def get_appointment_slots(date, timezone):
timezone = int(timezone) import pytz
guest_timezone = pytz.timezone(timezone)
format_string = '%Y-%m-%d %H:%M:%S' format_string = '%Y-%m-%d %H:%M:%S'
query_start_time = datetime.datetime.strptime( query_start_time = datetime.datetime.strptime(
date + ' 00:00:00', format_string) date + ' 00:00:00', format_string)
query_end_time = datetime.datetime.strptime( query_end_time = datetime.datetime.strptime(
date + ' 23:59:59', format_string) date + ' 23:59:59', format_string)
query_start_time = _convert_to_ist(query_start_time, timezone) local_timezone = frappe.utils.get_time_zone()
query_end_time = _convert_to_ist(query_end_time, timezone) local_timezone = pytz.timezone(local_timezone)
query_start_time = guest_timezone.localize(query_start_time)
query_end_time = guest_timezone.localize(query_end_time)
query_start_time = query_start_time.astimezone(local_timezone)
query_end_time = query_end_time.astimezone(local_timezone)
now = datetime.datetime.now() now = datetime.datetime.now()
# now = local_timezone.localize(now)
# Database queries # Database queries
settings = frappe.get_doc('Appointment Booking Settings') settings = frappe.get_doc('Appointment Booking Settings')
holiday_list = frappe.get_doc('Holiday List', settings.holiday_list) holiday_list = frappe.get_doc('Holiday List', settings.holiday_list)
@ -47,18 +54,22 @@ def get_appointment_slots(date, timezone):
# Filter timeslots based on date # Filter timeslots based on date
converted_timeslots = [] converted_timeslots = []
for timeslot in timeslots: for timeslot in timeslots:
timeslot = local_timezone.localize(timeslot)
print(timeslot)
timeslot = timeslot.astimezone(guest_timezone)
timeslot = timeslot.replace(tzinfo=None)
# Check if holiday # Check if holiday
if _is_holiday(timeslot.date(), holiday_list): if _is_holiday(timeslot.date(), holiday_list):
converted_timeslots.append( converted_timeslots.append(
dict(time=_convert_to_tz(timeslot, timezone), availability=False)) dict(time=timeslot, availability=False))
continue continue
# Check availability # Check availability
if check_availabilty(timeslot, settings) and timeslot >= now: if check_availabilty(timeslot, settings) and timeslot >= now:
converted_timeslots.append( converted_timeslots.append(
dict(time=_convert_to_tz(timeslot, timezone), availability=True)) dict(time=timeslot, availability=True))
else: else:
converted_timeslots.append( converted_timeslots.append(
dict(time=_convert_to_tz(timeslot, timezone), availability=False)) dict(time=timeslot, availability=False))
date_required = datetime.datetime.strptime( date_required = datetime.datetime.strptime(
date + ' 00:00:00', format_string).date() date + ' 00:00:00', format_string).date()
converted_timeslots = filter_timeslots(date_required, converted_timeslots) converted_timeslots = filter_timeslots(date_required, converted_timeslots)
@ -133,18 +144,4 @@ def _deltatime_to_datetime(date, deltatime):
def _datetime_to_deltatime(date_time): def _datetime_to_deltatime(date_time):
midnight = datetime.datetime.combine(date_time.date(), datetime.time.min) midnight = datetime.datetime.combine(date_time.date(), datetime.time.min)
return (date_time-midnight) return (date_time-midnight)
def _convert_to_ist(datetime_object, timezone):
offset = datetime.timedelta(minutes=timezone)
datetime_object = datetime_object + offset
offset = datetime.timedelta(minutes=-330)
datetime_object = datetime_object - offset
return datetime_object
def _convert_to_tz(datetime_object, timezone):
offset = datetime.timedelta(minutes=timezone)
datetime_object = datetime_object - offset
offset = datetime.timedelta(minutes=-330)
datetime_object = datetime_object + offset
return datetime_object