seitime-frappe/frappe/chat/util/util.py

# imports - third-party imports
import requests

# imports - compatibility imports
import six

# imports - standard imports
from   collections import MutableSequence, Mapping, MutableMapping
if six.PY2:
	from urlparse import urlparse 	  # PY2
else:
	from urllib.parse import urlparse # PY3
import json

# imports - module imports
from   frappe.model.document import Document
from   frappe.exceptions import DuplicateEntryError
from   frappe import _dict
import frappe

session = frappe.session

def get_user_doc(user = None):
	if isinstance(user, Document):
		return user

	user = user or session.user
	user = frappe.get_doc('User', user)
	
	return user

def squashify(what):
	if isinstance(what, MutableSequence) and len(what) == 1:
		return what[0]

	return what

def safe_json_loads(*args):
	results = [ ]

	for arg in args:
		try:
			arg = json.loads(arg)
		except Exception as e:
			pass
			
		results.append(arg)
	
	return squashify(results)

def filter_dict(what, keys, ignore = False):
	copy = dict()

	if keys:
		for k in keys:
			if k not in what and not ignore:
				raise KeyError('{key} not in dict.'.format(key = k))
			else:
				copy.update({
					k: what[k]
				})
	else:
		copy = what.copy()

	return copy

def assign_if_none(a, b):
	if a is None:
		a = b
	return a
	
def listify(arg):
	if not isinstance(arg, list):
		arg = [arg]
	return arg

def dictify(arg):
	if isinstance(arg, MutableSequence):
		for i, a in enumerate(arg):
			arg[i] = dictify(a)
	elif isinstance(arg, MutableMapping):
		arg = _dict(arg)

	return arg

def check_url(what, raise_err = False):
	if not urlparse(what).scheme:
		if raise_err:
			raise ValueError('{what} not a valid URL.')
		else:
			return False
	
	return True

def create_test_user(module):
	try:
		test_user = frappe.new_doc('User')
		test_user.first_name = '{module}'.format(module = module)
		test_user.email      = 'testuser.{module}@example.com'.format(module = module)
		test_user.save()
	except DuplicateEntryError:
		frappe.log('Test User Chat Profile exists.')

def get_emojis():
	redis  = frappe.cache()
	emojis = redis.hget('frappe_emojis', 'emojis')

	if not emojis:
		resp  = requests.get('http://git.io/frappe-emoji')
		if resp.ok:
			emojis = resp.json()
			redis.hset('frappe_emojis', 'emojis', emojis)
	
	return dictify(emojis)