def direct_message():
""" Function to save direct message content for user in databse """
try:
get_uid = request.args.get('userid', default=None, type=int)
uid = current_user.id
if get_uid is not None:
uid = get_uid
get_message = DirectMessage.byuser_id(uid)
dform = direct_message_form.direct_message(obj=get_message)
if not dform.message.data:
dform.message.data = "Hi {@username}, Thanks for your follow"
if dform.validate_on_submit():
if get_message:
get_message.message = dform.message.data
get_message.updated_at = func.now()
else:
message_obj = DirectMessage(user_id=uid,
message=dform.message.data,
created_at=func.now(),
updated_at=func.now())
db.session.add(message_obj)
db.session.commit()
flash(SAVE_MESSAGE, 'success')
return render_template('settings/direct_message.html', dform=dform)
except Exception as err:
errorlog.error('save direct message.', details=str(err))
return str(err)
def profile_target():
""" Function to show profile targets """
try:
qry = request.args.get('q', default=None, type=str)
get_uid = request.args.get('userid', default=None, type=int)
uid = current_user.id
if get_uid is not None and current_user.user_type == 'admin':
uid = get_uid
pf = TargetAccount.get_target(uid, 0, qry)
if qry is not None:
res = dict()
res['status'] = 'success'
res['account'] = render_template('targets/query_common.html',
pf=pf,
getpage='profile')
return json.dumps(res)
get_message = DirectMessage.byuser_id(uid)
if get_message:
message = get_message.message
else:
message = DIRECT_MESSAGE_DEFAULT
qry_template = render_template('targets/query_common.html',
pf=pf,
getpage='profile')
return render_template('targets/common.html',
pf=pf,
account_class="active",
get_page='profile_target',
message=message,
qry_template=qry_template,
btn_content='Add Target',
input_id='search-target-users',
action="targets")
except Exception as err:
errorlog.error('Profile Target Error', details=str(err))
return render_template('error.html', message="Error!")