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!")