jcloude/press/api/central.py

# Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors
# For license information, please see license.txt

import jingrow
from frappe.core.utils import find
from frappe.geo.country_info import get_country_timezone_info

from press.api.account import get_account_request_from_key
from press.press.doctype.site.erpnext_site import (
	ERPNextSite,
	get_erpnext_domain,
	get_erpnext_plan,
)
from press.press.doctype.site.pool import get as get_pooled_site
from press.press.doctype.team.team import Team


@frappe.whitelist(allow_guest=True)
def account_request(
	subdomain, email, first_name, last_name, phone_number, country, url_args=None
):
	email = email.strip().lower()
	frappe.utils.validate_email_address(email, True)

	if not check_subdomain_availability(subdomain):
		frappe.throw(f"Subdomain {subdomain} is already taken")

	all_countries = frappe.db.get_all("Country", pluck="name")
	country = find(all_countries, lambda x: x.lower() == country.lower())
	if not country:
		frappe.throw("Country filed should be a valid country name")

	account_request = frappe.get_pg(
		{
			"doctype": "Account Request",
			"erpnext": True,
			"subdomain": subdomain,
			"email": email,
			"role": "Press Admin",
			"first_name": first_name,
			"last_name": last_name,
			"phone_number": phone_number,
			"country": country,
			"url_args": url_args,
			"send_email": True,
		}
	).insert(ignore_permissions=True)

	current_user = frappe.session.user
	current_session_data = frappe.session.data
	frappe.set_user("Administrator")

	try:
		pooled_site = get_pooled_site()
		if pooled_site:
			# Rename a standby site
			ERPNextSite(site=pooled_site).rename_pooled_site(account_request)
		else:
			# Create a new site if pooled sites aren't available
			site = ERPNextSite(account_request=account_request).insert(ignore_permissions=True)
			site.create_subscription(get_erpnext_plan())
	finally:
		frappe.set_user(current_user)
		frappe.session.data = current_session_data


@frappe.whitelist(allow_guest=True)
def setup_account(key, business_data=None):
	account_request = get_account_request_from_key(key)
	if not account_request:
		frappe.throw("Invalid or Expired Key")

	frappe.set_user("Administrator")

	if business_data:
		business_data = frappe.parse_json(business_data)

	if isinstance(business_data, dict):
		business_data = {
			key: business_data.get(key)
			for key in [
				"company",
				"no_of_employees",
				"industry",
				"no_of_users",
				"designation",
				"referral_source",
				"agreed_to_partner_consent",
			]
		}

	account_request.update(business_data)
	account_request.save(ignore_permissions=True)

	email = account_request.email

	if not frappe.db.exists("Team", email):
		team_pg = Team.create_new(
			account_request,
			account_request.first_name,
			account_request.last_name,
			country=account_request.country,
			via_erpnext=True,
		)
	else:
		team_pg = frappe.get_pg("Team", email)

	site_name = frappe.db.get_value("Site", {"account_request": account_request.name})
	site = frappe.get_pg("Site", site_name)
	site.team = team_pg.name
	site.save()

	subscription = site.subscription
	if subscription:
		subscription.team = team_pg.name
		subscription.save()

	frappe.set_user(team_pg.user)
	frappe.local.login_manager.login_as(team_pg.user)

	return site.name


@frappe.whitelist(allow_guest=True)
def check_subdomain_availability(subdomain):
	exists = bool(
		frappe.db.exists(
			"Site",
			{
				"subdomain": subdomain,
				"domain": get_erpnext_domain(),
				"status": ("!=", "Archived"),
			},
		)
	)
	if exists:
		return False

	return True


@frappe.whitelist(allow_guest=True)
def options_for_regional_data(key):
	account_request = get_account_request_from_key(key)
	if not account_request:
		frappe.throw("Invalid or Expired Key")

	data = {
		"languages": frappe.db.get_all("Language", ["language_name", "language_code"]),
		"currencies": frappe.db.get_all("Currency", pluck="name"),
		"country": account_request.country,
	}
	data.update(get_country_timezone_info())

	return data


@frappe.whitelist(allow_guest=True)
def get_trial_end_date(site):
	if not site or not isinstance(site, str):
		frappe.throw("Invalid Site")

	return frappe.db.get_value("Site", site, "trial_end_date")


@frappe.whitelist(allow_guest=True)
def send_login_link(site):
	if not site or not isinstance(site, str) or not frappe.db.exists("Site", site):
		frappe.throw("Invalid site")

	from press.api.account import send_login_link as send_link

	# send link to site owner
	email = frappe.db.get_value("Site", site, "team")
	send_link(email)