import random import string from functools import wraps import jwt from django.conf import settings # 导入user模型 from django.contrib.auth import get_user_model, login, logout # views.py from django.contrib.auth.decorators import login_required from django.contrib.auth.models import User from django.contrib.sessions.models import Session from django.core.mail import send_mail # ajcx请求以json作为前后端交互 from django.http.response import JsonResponse from django.middleware.csrf import get_token from django.shortcuts import render, HttpResponse, redirect, reverse # 注册界面 from django.views.decorators.http import require_http_methods from rest_framework import status from rest_framework.authtoken.models import Token # User = get_user_model() from rest_framework.decorators import api_view, permission_classes from rest_framework.permissions import IsAuthenticated from .forms import RegisterForm, LoginForm, FundForm from .models import CaptchaModel, Fund from .serializers import FundSerializer from .utils import generate_jwt_token # @require_http_methods(['GET']) # def get_token(request): # token = request.session.get('token') # print('开始获取token') # print(f'token{token}') # print(f"会话数据2: {request.session.items()}") # if token: # return JsonResponse({'token': token}) # else: # return JsonResponse({'error': 'No token found'}, status=400) def get_csrf_token_view(request): return JsonResponse({'csrfToken': get_token(request)}) @api_view(['POST']) @permission_classes([IsAuthenticated]) def add_fund(request): print('开始调用') serializer = FundSerializer(data=request.data) if serializer.is_valid(): fund = serializer.save(user=request.user) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) @login_required def delete_fund(request, fund_id): fund = Fund.objects.get(id=fund_id, user=request.user) # 确保只能删除属于用户的基金 if fund: fund.delete() return redirect('funds_list') return render(request, 'error.html', {'message': '无权进行此操作'}) def validate_session(request): sessionid = request.GET.get('sessionid') if not sessionid: return JsonResponse({'error': 'No sessionid provided'}, status=400) try: session = Session.objects.get(session_key=sessionid) user_id = session.get_decoded().get('_auth_user_id') user = User.objects.get(pk=user_id) return JsonResponse({'username': user.username}) except Session.DoesNotExist: return JsonResponse({'error': 'Invalid sessionid'}, status=400) except User.DoesNotExist: return JsonResponse({'error': 'User not found'}, status=400) # @login_required # def get_current_user(request): # # 从请求头中获取令牌 # token = request.headers.get('Authorization') # # if not token: # return JsonResponse({'error': 'Authorization header is missing'}, status=401) # # try: # # 验证令牌 # decoded_token = jwt.decode(token, settings.SECRET_KEY, algorithms=['HS256']) # user_id = decoded_token.get('user_id') # if user_id: # # 如果令牌有效，返回用户数据 # user = User.objects.get(pk=user_id) # return JsonResponse({ # 'user_id': user.id, # 'username': user.username, # 'email': user.email # }) # else: # return JsonResponse({'error': 'Invalid token'}, status=401) # except jwt.ExpiredSignatureError: # return JsonResponse({'error': 'Token has expired'}, status=401) # except jwt.InvalidTokenError: # return JsonResponse({'error': 'Invalid token'}, status=401) # Create your views here. @login_required def get_current_user(request): sessionid = request.COOKIES.get('sessionid') print(f'sessionid2:{sessionid}') if not sessionid: return JsonResponse({'error': 'No sessionid provided'}, status=400) try: session = Session.objects.get(session_key=sessionid) user_id = session.get_decoded().get('_auth_user_id') user = User.objects.get(pk=user_id) return JsonResponse({'user_id': user.id, 'username': user.username}) except Session.DoesNotExist: return JsonResponse({'error': 'Invalid sessionid'}, status=400) except User.DoesNotExist: return JsonResponse({'error': 'User not found'}, status=400) def index(request): return render(request, '初始页面.html') @login_required def streamlit_test(request): sessionid = request.GET.get('sessionid') print(sessionid) return render(request, 'stream_test.html', {'sessionid': sessionid}) # return render(request, 'stream_test.html') # def login(request): # print('开始登录') # return render(request, '登录界面.html') @require_http_methods(['GET', 'POST']) def lg_login(request): print('开始登录') if request.method == 'GET': return render(request, '登录界面.html') else: form = LoginForm(request.POST) if form.is_valid(): email = form.cleaned_data.get('email') password = form.cleaned_data.get('password') remember = form.cleaned_data.get('remember') user = User.objects.filter(email=email).first() if user and user.check_password(password): # 登录 login(request, user) # 判断是否需要记住我 if not remember: print('不记住') # 如果没有点击记住我，那么就要设置过期时间为0，即浏览器关闭后就会过期 request.session.set_expiry(0) else: print('记住两周') request.session.set_expiry(1209600) # 设置会话两周（1209600秒）过期 # 如果点击了，那么就什么都不做，使用默认的2周的过期时间 # return render(request, 'stream_test.html', {'form': form}) # return redirect('/streamlit') print(f'session:{request.session.session_key}') return redirect(f'/streamlit/?sessionid={request.session.session_key}') else: print('邮箱或密码错误！') # form.add_error('email', '邮箱或者密码错误！') # return render(request, 'login.html', context={"form": form}) return render(request, '登录界面.html', {'form': form}) else: # 表单无效，返回表单错误 return render(request, '登录界面.html', {'form': form}) # # @require_http_methods(['GET', 'POST']) # def lg_login(request): # if request.method == 'GET': # return render(request, '登录界面.html', {'form': LoginForm()}) # else: # form = LoginForm(request.POST) # if form.is_valid(): # email = form.cleaned_data.get('email') # password = form.cleaned_data.get('password') # remember = form.cleaned_data.get('remember') # user = User.objects.filter(email=email).first() # if user and user.check_password(password): # # 登录 # login(request, user) # # 生成Token # token, created = Token.objects.get_or_create(user=user) # # 生成Token # # 存储token到会话中 # print(f'token:{token.key}') # request.session['token'] = token.key # print(f"存储的token: {request.session['token']}") # print(f"会话数据: {request.session.items()}") # # 判断是否需要记住我 # if not remember: # request.session.set_expiry(0) # 浏览器关闭后会话过期 # else: # request.session.set_expiry(1209600) # 两周（1209600秒）过期 # # 返回Token到前端 # # return JsonResponse({'token': token.key}) # return redirect(f'/streamlit/?token={token.key}') # else: # # return render(request, '登录界面.html', {'form': form, 'error': '邮箱或者密码错误！'}) # else: # return render(request, '登录界面.html', {'form': form, 'error': '表单无效！'}) # @require_http_methods(['GET', 'POST']) # def lg_login(request): # print('开始登录') # if request.method == 'GET': # csrf_token = get_token(request) # response = JsonResponse({'csrfToken': csrf_token}) # response.set_cookie('csrftoken', csrf_token) # return response # else: # form = LoginForm(request.POST) # if form.is_valid(): # print('表单有效') # email = form.cleaned_data.get('email') # password = form.cleaned_data.get('password') # remember = form.cleaned_data.get('remember') # user = User.objects.filter(email=email).first() # if user and user.check_password(password): # login(request, user) # if not remember: # request.session.set_expiry(0) # 浏览器关闭后会话过期 # else: # request.session.set_expiry(1209600) # 两周（1209600秒）过期 # return JsonResponse({'sessionid': request.session.session_key, 'csrfToken': csrf_token}) # return render(request, '登录界面.html', {'form': form, 'error': '邮箱或者密码错误！'}) # else: # return render(request, '登录界面.html', {'form': form, 'error': '表单无效！'}) def lg_logout(request): logout(request) return render(request, '登录界面.html') # 装饰器 @require_http_methods(['GET', 'POST']) def register(request): print('开始注册') if request.method == 'GET': return render(request, '注册界面.html') else: form = RegisterForm(request.POST) if form.is_valid(): print('注册成功') email = form.cleaned_data.get('email') username = form.cleaned_data.get('username') password = form.cleaned_data.get('password') User.objects.create_user(email=email, username=username, password=password) return render(request, '登录界面.html', {'form': form}) else: print('注册失败') print(form.errors) return render(request, '注册界面.html', {'form': form}) def sender_email_captcha(request): email = request.GET.get('email') if not email: return JsonResponse({"code": 400, "message": "必须传递邮箱"}) captcha = "".join(random.sample(string.digits, k=4)) CaptchaModel.objects.update_or_create(email=email, defaults={'captcha': captcha}) send_mail("流感基金关联分析系统", message=f"您的注册验证码是{captcha}", recipient_list=[email], from_email=None) return JsonResponse({"code": 200, "message": "邮箱验证码发送成功"}) # 生成验证码 # ee2 #更新数据 def update(request): import os print("Current working directory:", os.getcwd()) import sys # print("\n".join(sys.path)) sys.path.append('D:\\python\\pythonProject\\流感预测集成系统（django）\\liugan_yuce') # for item in os.listdir('.'): # print(item) from .liugan_zhoubao_spider import get_liuganzhoubao from .beijing_zhoubao_spider import get_beijing_zhoubao from .get_baidu_index import get_baidu_index from .tiantian_jijin_spider import get_tiantian_jijin_data,get_fund_data_by_code from .stock_data_spider import get_stock_data get_liuganzhoubao() get_beijing_zhoubao() get_baidu_index() get_tiantian_jijin_data() get_stock_data() # get_fund_data_by_code('000001') return HttpResponse('数据更新成功')