bob
/
newmediamonitoring
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add platform to userprofile when login

This commit is contained in:
baoliang 2021-03-28 11:13:34 +08:00
parent 56226f0d1b
commit 4aba9e8331
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
polls/views/user.py
View File

@ -12,7 +12,6 @@ import os
from pathlib import Path from pathlib import Path
from polls.models import Notice, VerifyCode from polls.models import Notice, VerifyCode
from polls.utils import generate_code, sent_sms_code from polls.utils import generate_code, sent_sms_code
from dashboard.models import Organization, Userprofile from dashboard.models import Organization, Userprofile
@ -38,6 +37,7 @@ def polls_login(request):
if request.method == 'POST': if request.method == 'POST':
phone = request.POST.get('phone') phone = request.POST.get('phone')
password = request.POST.get('password') password = request.POST.get('password')
platform = request.POST.get('platform', 'android')
UserModel = get_user_model() UserModel = get_user_model()
user = UserModel.objects.filter(username=phone).first() user = UserModel.objects.filter(username=phone).first()
if not user: if not user:
@ -63,6 +63,8 @@ def polls_login(request):
result['thumbnail'] = request.build_absolute_uri( result['thumbnail'] = request.build_absolute_uri(
profile.image.url) if profile.image else None profile.image.url) if profile.image else None
result['organization'] = profile.organization.name result['organization'] = profile.organization.name
profile.platform = platform
profile.save()
return JsonResponse({'status': 'success', 'message': result}) return JsonResponse({'status': 'success', 'message': result})
else: else:
return JsonResponse({'status': 'error', 'message': '用户名或密码错误'}) return JsonResponse({'status': 'error', 'message': '用户名或密码错误'})
@ -234,7 +236,7 @@ def upload_profile(request):
ext = source.name.split('.')[-1] ext = source.name.split('.')[-1]
if ext not in ['jpg', 'png', 'jpeg', 'gif']: if ext not in ['jpg', 'png', 'jpeg', 'gif']:
return JsonResponse({'status': 'error', 'message': '上传的文件不是图片'}) return JsonResponse({'status': 'error', 'message': '上传的文件不是图片'})
filename = "%s.%s" % (uuid.uuid4(), ext) filename = "%s.%s" % (uuid.uuid4(), ext)
foldpath = os.path.join(settings.MEDIA_ROOT, 'profile') foldpath = os.path.join(settings.MEDIA_ROOT, 'profile')
Path(foldpath).mkdir(parents=True, exist_ok=True) Path(foldpath).mkdir(parents=True, exist_ok=True)