114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
# Copyright (c) 2018, Frappe and Contributors
|
|
# License: GNU General Public License v3. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
import frappe
|
|
|
|
def execute():
|
|
""" Generates leave ledger entries for leave allocation/application/encashment
|
|
for last allocation """
|
|
frappe.reload_doc("HR","doctype", "Leave Ledger Entry")
|
|
if frappe.db.a_row_exists("Leave Ledger Entry"):
|
|
return
|
|
|
|
allocation_list = get_allocation_records()
|
|
generate_allocation_ledger_entries(allocation_list)
|
|
generate_application_leave_ledger_entries(allocation_list)
|
|
generate_encashment_leave_ledger_entries(allocation_list)
|
|
|
|
def generate_allocation_ledger_entries(allocation_list):
|
|
''' fix ledger entries for missing leave allocation transaction '''
|
|
from erpnext.hr.doctype.leave_allocation.leave_allocation import LeaveAllocation
|
|
|
|
for allocation in allocation_list:
|
|
if not frappe.db.exists("Leave Ledger Entry", {'transaction_type': 'Leave Allocation', 'transaction_name': allocation.name}):
|
|
allocation.update(dict(doctype="Leave Allocation"))
|
|
leave_allocation = LeaveAllocation(allocation)
|
|
leave_allocation.create_leave_ledger_entry()
|
|
|
|
def generate_application_leave_ledger_entries(allocation_list):
|
|
''' fix ledger entries for missing leave application transaction '''
|
|
from erpnext.hr.doctype.leave_application.leave_application import LeaveApplication
|
|
|
|
leave_applications = get_leaves_application_records(allocation_list)
|
|
|
|
for record in leave_applications:
|
|
if not frappe.db.exists("Leave Ledger Entry", {'transaction_type': 'Leave Application', 'transaction_name': record.name}):
|
|
record.update(dict(doctype="Leave Application"))
|
|
leave_application = LeaveApplication(record)
|
|
leave_application.create_leave_ledger_entry()
|
|
|
|
def generate_encashment_leave_ledger_entries(allocation_list):
|
|
''' fix ledger entries for missing leave encashment transaction '''
|
|
from erpnext.hr.doctype.leave_encashment.leave_encashment import LeaveEncashment
|
|
|
|
leave_encashments = get_leave_encashment_records(allocation_list)
|
|
|
|
for record in leave_encashments:
|
|
if not frappe.db.exists("Leave Ledger Entry", {'transaction_type': 'Leave Encashment', 'transaction_name': record.name}):
|
|
record.update(dict(doctype="Leave Encashment"))
|
|
leave_encashment = LeaveEncashment(record)
|
|
leave_encashment.create_leave_ledger_entry()
|
|
|
|
def get_allocation_records():
|
|
return frappe.db.sql("""
|
|
WITH allocation_values AS (
|
|
SELECT
|
|
DISTINCT name,
|
|
employee,
|
|
leave_type,
|
|
new_leaves_allocated,
|
|
carry_forwarded_leaves,
|
|
from_date,
|
|
to_date,
|
|
carry_forward,
|
|
RANK() OVER(
|
|
PARTITION BY employee, leave_type
|
|
ORDER BY to_date DESC
|
|
) as allocation
|
|
FROM `tabLeave Allocation`
|
|
)
|
|
SELECT
|
|
*
|
|
FROM
|
|
`allocation_values`
|
|
WHERE
|
|
allocation=1
|
|
""", as_dict=1)
|
|
|
|
def get_leaves_application_records(allocation_list):
|
|
leave_applications = []
|
|
for allocation in allocation_list:
|
|
leave_applications += frappe.db.sql("""
|
|
SELECT
|
|
DISTINCT name,
|
|
employee,
|
|
leave_type,
|
|
total_leave_days,
|
|
from_date,
|
|
to_date
|
|
FROM `tabLeave Application`
|
|
WHERE
|
|
from_date >= %s
|
|
AND leave_type = %s
|
|
AND employee = %s
|
|
""", (allocation.from_date, allocation.leave_type, allocation.employee), as_dict=1)
|
|
return leave_applications
|
|
|
|
def get_leave_encashment_records(allocation_list):
|
|
leave_encashments = []
|
|
for allocation in allocation_list:
|
|
leave_encashments += frappe.db.sql("""
|
|
SELECT
|
|
DISTINCT name,
|
|
employee,
|
|
leave_type,
|
|
encashable_days,
|
|
encashment_date
|
|
FROM `tabLeave Encashment`
|
|
WHERE
|
|
leave_type = %s
|
|
AND employee = %s
|
|
AND encashment_date >= %s
|
|
""", (allocation.leave_type, allocation.employee, allocation.from_date), as_dict=1)
|
|
return leave_encashments |