style: Broken into smaller function

This commit is contained in:
Anurag Mishra
2020-10-27 15:44:43 +05:30
parent 4d6c3c9449
commit 46e1c09b45
5 changed files with 89 additions and 93 deletions

View File

@@ -7,6 +7,7 @@ import frappe, os, json
from frappe.custom.doctype.custom_field.custom_field import create_custom_fields
from frappe.permissions import add_permission, update_permission_property
from erpnext.setup.setup_wizard.operations.taxes_setup import create_sales_tax
from erpnext.payroll.doctype.gratuity_rule.gratuity_rule import get_gratuity_rule
def setup(company=None, patch=True):
make_custom_fields()
@@ -161,104 +162,91 @@ def add_permissions():
update_permission_property(doctype, role, 0, 'create', 1)
def create_gratuity_rule():
# Standard Gratuity Rules for UAE
rule_1 = rule_2 = rule_3 = None
# Rule Under Limited Contract
slabs = get_slab_for_limited_contract()
if not frappe.db.exists("Gratuity Rule", "Rule Under Limited Contract (UAE)"):
rule_1 = frappe.new_doc("Gratuity Rule")
rule_1.name = "Rule Under Limited Contract (UAE)"
rule_1.calculate_gratuity_amount_based_on = "Sum of all previous slabs"
rule_1.work_experience_calculation_method = "Take Exact Completed Years"
rule_1.minimum_year_for_gratuity = 1
rule_1.append("gratuity_rule_slabs", {
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
})
rule_1.append("gratuity_rule_slabs", {
"from_year": 1,
"to_year":5,
"fraction_of_applicable_earnings": 21/30
})
rule_1.append("gratuity_rule_slabs", {
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": 1
})
rule_1 = get_gratuity_rule("Rule Under Limited Contract (UAE)", slabs, calculate_gratuity_amount_based_on="Sum of all previous slabs")
# Rule Under Unlimited Contract on termination
slabs = get_slab_for_unlimited_contract_on_termination()
if not frappe.db.exists("Gratuity Rule", "Rule Under Unlimited Contract on termination (UAE)"):
rule_2 = frappe.new_doc("Gratuity Rule")
rule_2.name = "Rule Under Unlimited Contract on termination (UAE)"
rule_2.calculate_gratuity_amount_based_on = "Current Slab"
rule_2.work_experience_calculation_method = "Take Exact Completed Years"
rule_2.minimum_year_for_gratuity = 1
rule_2 = get_gratuity_rule("Rule Under Unlimited Contract on termination (UAE)", slabs)
rule_2.append("gratuity_rule_slabs", {
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
})
rule_2.append("gratuity_rule_slabs", {
"from_year": 1,
"to_year":5,
"fraction_of_applicable_earnings": 21/30
})
rule_2.append("gratuity_rule_slabs", {
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": 1
})
# Rule Under Unlimited Contract
# Rule Under Unlimited Contract on resignation
slabs = get_slab_for_unlimited_contract_on_resignation()
if not frappe.db.exists("Gratuity Rule", "Rule Under Unlimited Contract on resignation (UAE)"):
rule_3 = frappe.new_doc("Gratuity Rule")
rule_3.name = "Rule Under Unlimited Contract on resignation (UAE)"
rule_3.calculate_gratuity_amount_based_on = "Current Slab"
rule_3.work_experience_calculation_method = "Take Exact Completed Years"
rule_3.minimum_year_for_gratuity = 1
rule_3 = get_gratuity_rule("Rule Under Unlimited Contract on resignation (UAE)", slabs)
rule_3.append("gratuity_rule_slabs", {
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
})
fraction_of_applicable_earnings = 1/3 * 21/30
rule_3.append("gratuity_rule_slabs", {
"from_year": 1,
"to_year":3,
"fraction_of_applicable_earnings": fraction_of_applicable_earnings
})
fraction_of_applicable_earnings = 2/3 * 21/30
rule_3.append("gratuity_rule_slabs", {
"from_year": 3,
"to_year":5,
"fraction_of_applicable_earnings": fraction_of_applicable_earnings
})
fraction_of_applicable_earnings = 21/30
rule_3.append("gratuity_rule_slabs", {
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": fraction_of_applicable_earnings
})
#for applicable salary component user need to set this by its own
#for applicable salary component user need to set this by its own
if rule_1:
rule_1.flags.ignore_mandatory = True
rule_2.flags.ignore_mandatory = True
rule_3.flags.ignore_mandatory = True
rule_1.save()
if rule_2:
rule_2.flags.ignore_mandatory = True
rule_2.save()
if rule_3:
rule_3.flags.ignore_mandatory = True
rule_3.save()
def get_slab_for_limited_contract():
return [{
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
},
{
"from_year": 1,
"to_year":5,
"fraction_of_applicable_earnings": 21/30
},
{
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": 1
}]
def get_slab_for_unlimited_contract_on_termination():
return [{
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
},
{
"from_year": 1,
"to_year":5,
"fraction_of_applicable_earnings": 21/30
},
{
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": 1
}]
def get_slab_for_unlimited_contract_on_resignation():
fraction_1 = 1/3 * 21/30
fraction_2 = 2/3 * 21/30
fraction_3 = 21/30
return [{
"from_year": 0,
"to_year":1,
"fraction_of_applicable_earnings": 0
},
{
"from_year": 1,
"to_year":3,
"fraction_of_applicable_earnings": fraction_1
},
{
"from_year": 3,
"to_year":5,
"fraction_of_applicable_earnings": fraction_2
},
{
"from_year": 5,
"to_year":0,
"fraction_of_applicable_earnings": fraction_3
}]