def admin_analytics(request, selected_id):
current_user = utils.current_user(request)
data = request.GET.dict().copy()
col = []
for key in data:
if key != 'ip':
col.append(key)
col = ', '.join(col)
if request.GET.get('ip', '') != '':
if len(col) == 0:
col = "*"
analytics = Analytics.hits_by_ip(request.GET['ip'], col=col)
else:
analytics = Analytics.objects_in_list()
cols = [key for key in analytics]
values = analytics.values()
num_data = range(len(values[0]))
return render(
request, 'admin/analytics.html', {
'current_user': current_user,
'cols': cols,
'values': values,
'num_data': num_data
})
def user_messages(request, user_id):
current_user = utils.current_user(request)
if request.method == "GET":
return render(
request, "users/messages/index.html", {
'current_user': current_user,
'available_recipients': User.objects.all()
})
else:
try:
cid = int(request.POST['creator_id'])
creator = User.objects.get(user_id=cid)
rid = int(request.POST['receiver_id'])
receiver = User.objects.get(user_id=rid)
msg = request.POST['message']
red = int(request.POST['read'])
now = timezone.now()
Message.objects.create(creator=creator,
receiver=receiver,
message=msg,
read=red,
created_at=now,
updated_at=now)
return redirect("/users/" + str(current_user.id) + "/messages")
except Exception as e:
messages.add_message(request, messages.INFO, str(e))
return render(
request, "users/messages/index.html", {
'current_user': current_user,
'available_receipients': User.objects.all()
})
def home(request):
user = utils.current_user(request)
context = user.__dict__
try:
income = user.work_info.first().income_to_int()
pto_taken = int(user.pto.first().pto_taken)
pto_remain = int(user.pto.first().pto_days_remaining())
sick_taken = int(user.pto.first().sick_days_taken)
sick_remain = int(user.pto.first().sick_days_remaining())
performance = int(user.u_id.first().score)
retirement = int(user.retirements.first().total)
context.update({
'income': income,
'pto_taken': pto_taken,
'pto_remain': pto_remain,
'sick_taken': sick_taken,
'sick_remain': sick_remain,
'performance': performance,
'401k': retirement
})
except:
pass
finally:
context.update({'current_user': user})
return render(request, 'dashboard/home.html', context=context)
def account_settings(request, user_id):
user = utils.current_user(request)
if not user:
return HttpResponse("User " + str(user_id) + " NOT FOUND")
else:
context = user.__dict__
context.update({'current_user': user})
return render(request, "users/account_settings.html", context=context)
def on_model_change(self, form, model, is_created):
if is_created:
model.client_id = gen_salt(24)
if model.token_endpoint_auth_method == 'none':
model.client_secret = ''
else:
model.client_secret = gen_salt(48)
user = current_user()
if user:
model.user_id = user.id
def user_pay_index(request, user_id): # pylint: disable=unused-argument
user = utils.current_user(request)
if user is not None:
direct_deposits = Pay.objects.filter(user=user)
return render(request, 'users/pay/index.html', {
'current_user': user,
'direct_deposits': direct_deposits
})
else:
return HttpResponseRedirect('/signup')
def upload(request):
id = utils.current_user(request).user_id
if 'myfile' in request.FILES:
Benefits.save_data(request.FILES['myfile'], request.POST['backup'])
messages.success(request, 'File was successfully uploaded!')
else:
messages.error(
request, 'Something went wrong! Are you sure you selected a file?')
return HttpResponseRedirect(
reverse('app:user_benefit_forms', kwargs={'user_id': id}))
def user_pay_index(request, user_id):
template = get_template('users/pay/index.html')
user = utils.current_user(request)
if user is not None:
direct_deposits = Pay.objects.filter(user=user)
return render(request, 'users/pay/index.html', {
'current_user': user,
'direct_deposits': direct_deposits
})
else:
return HttpResponseRedirect('/signup')
def index(request, user_id):
user = utils.current_user(request)
if not user:
return HttpResponse("User " + str(user_id) + " NOT FOUND")
pto = PaidTimeOff.objects.filter(user=user).first()
if not pto:
return HttpResponse("PTO " + str(user_id) + " NOT FOUND")
if request.method == "GET":
return index_get(request, user_id, user, pto)
elif request.method == "POST":
return index_post(request, user_id, user, pto)
else:
return HttpResponse("Invalid HTTP method")
def user_message(request, user_id, message_id):
current_user = utils.current_user(request)
try:
message = Message.objects.get(pk=message_id)
if request.method == "GET":
return render(request, "users/messages/show.html", {
'current_user': current_user,
'message': message
})
else:
message.delete()
return HttpResponse("Success!")
except Exception:
return redirect("/users/" + str(current_user.id) + "/messages")
def update_dd_info(request, user_id):
curr_user = utils.current_user(request)
form = request.POST
if not form:
return HttpResponse("Pay index")
Pay.objects.create(user=curr_user,
bank_account_num=form['bankAccNumInput'],
bank_routing_num=form['bankRouteNumInput'],
percent_of_deposit=form['percentDepositInput'],
created_at=timezone.now(),
updated_at=timezone.now())
return HttpResponseRedirect('/users/' + str(user_id) + '/pay')
def decrypt_bank_acct_num(request, user_id):
form = request.POST
if not form:
return HttpResponse("No form")
account_num = request.POST['account_number']
curr_user = utils.current_user(request)
response = HttpResponse()
try:
pay = Pay.objects.get(user=curr_user, bank_account_num=account_num)
decrypted_account_num = pay.decrypt_bank_num()
response['success'] = True
response.content = decrypted_account_num
return response
except Pay.DoesNotExist as e:
response['success'] = False
return response
def index():
if request.method == 'POST':
username = request.form.get('username')
user = User.query.filter_by(username=username).first()
# 创建一个帐号
if not user:
user = User(username=username)
db.session.add(user)
db.session.commit()
session['id'] = user.id
return redirect('/')
user = current_user()
if user:
# 获取该帐号的客户端
clients = OAuth2Client.query.filter_by(user_id=user.id).all()
else:
clients = []
return render_template('home.html', user=user, clients=clients)
def authorize():
""""认证页面"""
user = current_user()
if request.method == 'GET':
try:
grant = authorization.validate_consent_request(end_user=user)
except OAuth2Error as error:
current_app.logger.debug(error)
return error.error
current_app.logger.debug(f'user:{user}')
current_app.logger.debug(f'grant:{grant}')
return render_template('authorize.html', user=user, grant=grant)
if not user and 'username' in request.form:
username = request.form.get('username')
user = User.query.filter_by(username=username).first()
if request.form['confirm']:
grant_user = user
else:
grant_user = None
return authorization.create_authorization_response(grant_user=grant_user)
def admin_dashboard(request, selected_id): # pylint: disable=unused-argument
current_user = utils.current_user(request)
return render(request, 'admin/dashboard.html',
{'current_user': current_user})
def admin_dashboard(request, selected_id):
current_user = utils.current_user(request)
return render(request, 'admin/dashboard.html',
{'current_user': current_user})
def user_benefit_forms(request, user_id): # pylint: disable=unused-argument
user = utils.current_user(request)
return render(request,
'users/benefit_forms.html',
context={'current_user': user})
def user_retirement_index(request, user_id):
current_user = utils.current_user(request)
retirement_info = current_user.retirements.first()
context = {'retirement': retirement_info, 'current_user': current_user}
return render(request, 'users/retirement/index.html', context=context)
def user_benefit_forms(request, user_id):
user = utils.current_user(request)
return render(request,
'users/benefit_forms.html',
context={'current_user': user})
def user_performance_index(request, user_id): # pylint: disable=W0613
current_user = utils.current_user(request)
performance = current_user.u_id
t = get_template('users/performance/index.html')
html = t.render({'performance': performance, 'current_user': current_user})
return HttpResponse(html)
