beneficiary/changemakers/api/__init__.py

import base64
from mimetypes import guess_type

import frappe

from changemakers import __mobile_version__ as mobile_version
from changemakers.utils.form import get_doctype_title_field

MOBILE_SUPPORTED_FIELD_TYPES = [
	"Select",
	"Link",
	"Data",
	"Int",
	"Datetime",
	"Check",
	"Geolocation",
	"Attach",
	"Small Text",
	"Section Break",
	"Table",
	"Text",
]


@frappe.whitelist()
def get_app_version():
	return mobile_version


@frappe.whitelist()
def get_current_user_info():
	current_user = frappe.session.user
	return frappe.db.get_value(
		"User", current_user, ["first_name", "full_name", "user_image"], as_dict=True
	)


@frappe.whitelist()
def get_form_fields(doctype):
	return frappe.db.get_all(
		"DocField",
		filters={
			"parenttype": "DocType",
			"parent": doctype,
			"fieldtype": ("in", MOBILE_SUPPORTED_FIELD_TYPES),
			"fieldname": ("!=", "amended_from"),
			"fetch_from": ("is", "not set"),  # TODO: handle later
		},
		fields=[
			"label",
			"fieldname",
			"idx",
			"fieldtype",
			"reqd",
			"default",
			"options",
			"read_only",
		],
		order_by="idx",
		ignore_permissions=True,
	)


@frappe.whitelist()
def get_doctype_options(doctype):
	title_field = get_doctype_title_field(doctype)

	fields = [title_field]
	if title_field != "name":
		fields.append("name")

	# maybe paginate later
	return {"docs": frappe.get_all(doctype, fields=fields), "title_field": title_field}


@frappe.whitelist()
def upload_base64_file(content, filename, dt=None, dn=None, fieldname=None):

	from frappe.handler import ALLOWED_MIMETYPES

	decoded_content = base64.b64decode(content)
	content_type = guess_type(filename)[0]
	if content_type not in ALLOWED_MIMETYPES:
		frappe.throw("You can only upload JPG, PNG, PDF, TXT or Microsoft documents.")

	return frappe.get_doc(
		{
			"doctype": "File",
			"attached_to_doctype": dt,
			"attached_to_name": dn,
			"attached_to_field": fieldname,
			"folder": "Home",
			"file_name": filename,
			"content": decoded_content,
			"is_private": 1,
		}
	).save(ignore_permissions=True)


@frappe.whitelist()
def get_attached_images(doctype: str, name: str) -> list:
	"""get list of image urls attached in form
	returns [{'data': <image base64 content>, 'filename': <filename>}]"""

	img_urls = frappe.db.get_list(
		"File",
		filters={
			"attached_to_doctype": doctype,
			"attached_to_name": name,
			"is_folder": 0,
		},
		fields=["file_url", "attached_to_name as docname", "name", "file_name"],
	)

	out = []
	for image in img_urls:
		content_type = guess_type(image.file_name)[0]
		if content_type not in ("image/jpeg", "image/png"):
			continue
		out.append({"url": image.file_url, "filename": image.name})

	return out


@frappe.whitelist()
def download_base64_file(file_url):
	filedoc = frappe.get_doc("File", {"file_url": file_url})

	base64content = base64.b64encode(filedoc.get_content())
	content_type = guess_type(filedoc.file_name)[0]
	name = filedoc.name

	data = f"data:{content_type};base64, " + base64content.decode("utf-8")
	return {"data": data, "name": name}