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

fix in holiday list

Browse Source
This commit is contained in:
Anand Doshi 2013-01-08 18:14:39 +05:30
parent cfd185e5ac
commit 89230cbc4b
1 changed files with 55 additions and 92 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

133
hr/doctype/holiday_list/holiday_list.py
View File

@ -8,11 +8,11 @@
# #
# This program is distributed in the hope that it will be useful, # This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of # but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details. # GNU General Public License for more details.
# #
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
from __future__ import unicode_literals from __future__ import unicode_literals
import webnotes import webnotes
@ -27,100 +27,63 @@ sql = webnotes.conn.sql
import datetime import datetime
class DocType: class DocType:
def __init__(self,doc,doclist=[]): def __init__(self,doc,doclist=[]):
self.doc = doc self.doc = doc
self.doclist = doclist self.doclist = doclist
# --------- def autoname(self):
# autoname self.doc.name = make_autoname(self.doc.fiscal_year +"/"+ self.doc.holiday_list_name+"/.###")
# ---------
def autoname(self):
self.doc.name = make_autoname(self.doc.fiscal_year +"/"+ self.doc.holiday_list_name+"/.###")
def validate(self):
self.update_default_holiday_list()
# *************************************************** utilities *********************************************** def get_weekly_off_dates(self):
# ---------------- self.validate_values()
# validate values yr_start_date, yr_end_date = self.get_fy_start_end_dates()
# ---------------- date_list = self.get_weekly_off_date_list(yr_start_date, yr_end_date)
def validate_values(self): for d in date_list:
if not self.doc.fiscal_year: ch = addchild(self.doc, 'holiday_list_details', 'Holiday', self.doclist)
msgprint("Please select Fiscal Year") ch.description = self.doc.weekly_off
raise Exception ch.holiday_date = d
if not self.doc.weekly_off:
msgprint("Please select weekly off day")
raise Exception
def validate_values(self):
if not self.doc.fiscal_year:
msgprint("Please select Fiscal Year")
raise Exception
if not self.doc.weekly_off:
msgprint("Please select weekly off day")
raise Exception
# ------------------------------------ def get_fy_start_end_dates(self):
# get fiscal year start and end dates return webnotes.conn.sql("""select year_start_date,
# ------------------------------------ subdate(adddate(year_start_date, interval 1 year), interval 1 day)
def get_fy_start_end_dates(self): as year_end_date
st_date = sql("select year_start_date from `tabFiscal Year` where name = '%s'" %(self.doc.fiscal_year)) from `tabFiscal Year`
st_date = st_date and st_date[0][0].strftime('%Y-%m-%d') or '' where name=%s""", (self.doc.fiscal_year,))[0]
ed_date = add_days(add_years(st_date,1), -1)
return st_date, ed_date
# ------------------------- def get_weekly_off_date_list(self, year_start_date, year_end_date):
# get weekly off date list from webnotes.utils import getdate
# ------------------------- year_start_date, year_end_date = getdate(year_start_date), getdate(year_end_date)
def get_weekly_off_date_list(self, yr_start_date, yr_end_date):
days_dict, dt_list, lst_st = {'Monday':0,'Tuesday':1,'Wednesday':2,'Thursday':3,'Friday':4,'Saturday':5,'Sunday':6}, [], ''
w = cint(days_dict[self.doc.weekly_off]) # Weekly Off Day No. from dateutil import relativedelta
from datetime import timedelta
import calendar
st_dt_weekday = getdate(yr_start_date).weekday() # Year Start Date weekday() date_list = []
weekday = getattr(calendar, (self.doc.weekly_off).upper())
reference_date = year_start_date + relativedelta.relativedelta(weekday=weekday)
if w == st_dt_weekday: # Get Start Date while reference_date <= year_end_date:
lst_st = yr_start_date date_list.append(reference_date)
dt_list.append(lst_st) reference_date += timedelta(days=7)
elif w > st_dt_weekday:
lst_st = add_days(yr_start_date,w - st_dt_weekday)
dt_list.append(lst_st)
else:
lst_st = add_days(yr_start_date,6 - st_dt_weekday + 1)
dt_list.append(lst_st)
while getdate(lst_st) < getdate(yr_end_date): # Get list of dates return date_list
lst_st = add_days(lst_st,7)
if getdate(lst_st) > getdate(yr_end_date):
break
dt_list.append(lst_st)
return dt_list def clear_table(self):
self.doclist = self.doc.clear_table(self.doclist, 'holiday_list_details')
# --------------------- def update_default_holiday_list(self):
# get weekly off dates webnotes.conn.sql("""update `tabHoliday List` set is_default = 0
# --------------------- where ifnull(is_default, 0) = 1 and fiscal_year = %s""", (self.doc.fiscal_year,))
def get_weekly_off_dates(self):
self.validate_values()
yr_start_date, yr_end_date = self.get_fy_start_end_dates()
date_list = self.get_weekly_off_date_list(yr_start_date, yr_end_date)
for d in date_list:
ch = addchild(self.doc, 'holiday_list_details', 'Holiday', self.doclist)
ch.description = self.doc.weekly_off
ch.holiday_date = d
# ------------
# clear table
# ------------
def clear_table(self):
self.doclist = self.doc.clear_table(self.doclist,'holiday_list_details')
# ***************************************** validate *************************************************
# ---------------------------
# check default holiday list
# ---------------------------
def update_default_holiday_list(self):
sql("update `tabHoliday List` set is_default = 0 where ifnull(is_default, 0) = 1 and fiscal_year = '%s'" % (self.doc.fiscal_year))
# ---------
# validate
# ---------
def validate(self):
self.update_default_holiday_list()