erpnext/regional/saudi_arabia/setup.py

# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
# License: GNU General Public License v3. See license.txt

import frappe
from frappe.permissions import add_permission, update_permission_property
from erpnext.regional.saudi_arabia.wizard.operations.setup_ksa_vat_setting import create_ksa_vat_setting
from frappe.custom.doctype.custom_field.custom_field import create_custom_fields

def setup(company=None, patch=True):
	make_custom_fields()
	add_print_formats()
	add_permissions()
	make_custom_fields()

def make_custom_fields():
	is_zero_rated = dict(fieldname='is_zero_rated', label='Is Zero Rated',
		fieldtype='Check', fetch_from='item_code.is_zero_rated', insert_after='description',
		print_hide=1)
	
	is_exempt = dict(fieldname='is_exempt', label='Is Exempt',
		fieldtype='Check', fetch_from='item_code.is_exempt', insert_after='is_zero_rated',
		print_hide=1)

	purchase_invoice_fields = [
			dict(fieldname='company_trn', label='Company TRN',
				fieldtype='Read Only', insert_after='shipping_address',
				fetch_from='company.tax_id', print_hide=1),
			dict(fieldname='supplier_name_in_arabic', label='Supplier Name in Arabic',
				fieldtype='Read Only', insert_after='supplier_name',
				fetch_from='supplier.supplier_name_in_arabic', print_hide=1)
		]

	sales_invoice_fields = [
			dict(fieldname='company_trn', label='Company TRN',
				fieldtype='Read Only', insert_after='company_address',
				fetch_from='company.tax_id', print_hide=1),
			dict(fieldname='customer_name_in_arabic', label='Customer Name in Arabic',
				fieldtype='Read Only', insert_after='customer_name',
				fetch_from='customer.customer_name_in_arabic', print_hide=1)
		]

	custom_fields = {
		'Item': [is_zero_rated, is_exempt],
		'Customer': [
			dict(fieldname='customer_name_in_arabic', label='Customer Name in Arabic',
				fieldtype='Data', insert_after='customer_name'),
		],
		'Supplier': [
			dict(fieldname='supplier_name_in_arabic', label='Supplier Name in Arabic',
				fieldtype='Data', insert_after='supplier_name'),
		],
		'Purchase Invoice': purchase_invoice_fields,
		'Purchase Order': purchase_invoice_fields,
		'Purchase Receipt': purchase_invoice_fields,
		'Sales Invoice': sales_invoice_fields,
		'POS Invoice': sales_invoice_fields,
		'Sales Order': sales_invoice_fields,
		'Delivery Note': sales_invoice_fields,
		'Sales Invoice Item': [is_zero_rated, is_exempt],
		'POS Invoice Item': [is_zero_rated, is_exempt],
		'Purchase Invoice Item': [is_zero_rated, is_exempt],
		'Sales Order Item': [is_zero_rated, is_exempt],
		'Delivery Note Item': [is_zero_rated, is_exempt],
		'Quotation Item': [is_zero_rated, is_exempt],
		'Purchase Order Item': [is_zero_rated, is_exempt],
		'Purchase Receipt Item': [is_zero_rated, is_exempt],
		'Supplier Quotation Item': [is_zero_rated, is_exempt],
	}

	create_custom_fields(custom_fields)

def add_print_formats():
	frappe.reload_doc("regional", "print_format", "detailed_tax_invoice", force=True)
	frappe.reload_doc("regional", "print_format", "simplified_tax_invoice", force=True)
	frappe.reload_doc("regional", "print_format", "tax_invoice", force=True)
	frappe.reload_doc("regional", "print_format", "ksa_vat_invoice", force=True)
	frappe.reload_doc("regional", "print_format", "ksa_pos_invoice", force=True)

	for d in ('Simplified Tax Invoice', 'Detailed Tax Invoice', 'Tax Invoice', 'KSA VAT Invoice', 'KSA POS Invoice'):
		frappe.db.set_value("Print Format", d, "disabled", 0)

def add_permissions():
	"""Add Permissions for KSA VAT Setting."""
	add_permission('KSA VAT Setting', 'All', 0)
	for role in ('Accounts Manager', 'Accounts User', 'System Manager'):
		add_permission('KSA VAT Setting', role, 0)
		update_permission_property('KSA VAT Setting', role, 0, 'write', 1)
		update_permission_property('KSA VAT Setting', role, 0, 'create', 1)

	"""Enable KSA VAT Report"""
	frappe.db.set_value('Report', 'KSA VAT', 'disabled', 0)

def make_custom_fields():
	"""Create Custom fields
	- QR code Image file
	- Company Name in Arabic
	- Address in Arabic
	"""
	custom_fields = {
		'Sales Invoice': [
			dict(
				fieldname='ksa_einv_qr',
				label='KSA E-Invoicing QR',
				fieldtype='Attach Image',
				read_only=1, no_copy=1, hidden=1
			)
		],
		'POS Invoice': [
			dict(
				fieldname='ksa_einv_qr',
				label='KSA E-Invoicing QR',
				fieldtype='Attach Image',
				read_only=1, no_copy=1, hidden=1
			)
		],
		'Address': [
			dict(
				fieldname='address_in_arabic',
				label='Address in Arabic',
				fieldtype='Data',
				insert_after='address_line2'
			)
		],
		'Company': [
			dict(
				fieldname='company_name_in_arabic',
				label='Company Name In Arabic',
				fieldtype='Data',
				insert_after='company_name'
			)
		]
	}

	create_custom_fields(custom_fields, update=True)

def update_regional_tax_settings(country, company):
	create_ksa_vat_setting(company)