def register_btn(self):
req = self.request.POST
uname = req.get("uname")
uphone = req.get("uphone")
uemail = req.get("uemail")
upwd = req.get("upwd")
cpwd = req.get("cpwd")
user = UserInfo.objects.filter(name=uname)
phone = UserInfo.objects.filter(phone=uphone)
email = UserInfo.objects.filter(email=uemail)
if user or phone or email or (upwd != cpwd):
# 不能注册
self.flag = 0
else:
# 可以注册
upwd = make_password(upwd, auth_check, "pbkdf2_sha1")
new_user = UserInfo(name=uname,
phone=uphone,
email=uemail,
password=upwd)
new_user.save()
print("注册成功")
self.request.session['uname'] = uname
print("session 存入成功")
self.request.session.set_expiry(0)
self.ret_dic["result"] = self.flag
return self.ret_dic
def test_bulk_create(request):
user_list = []
for i in range(100):
user = '******'.format(i)
user_list.append(UserInfo(user=user, pwd='123456'))
UserInfo.objects.bulk_create(user_list)
return HttpResponse('ok')
def username(request):
users = None
if 'username' in request.POST and request.POST['username']:
username = request.POST['username']
users = UserInfo.objects.filter(username=username)
if not users.exists():
user = UserInfo.create(username)
user.save()
users = UserInfo.objects.filter(username=username)
return render(request, 'second.html', {'users': users})
def register(request):
username = request.POST.get('username')
password = request.POST.get('password')
if not username or not password:
return JsonFailResponse('username or password is null')
try:
user = User.objects.create_user(request.POST['username'], '', request.POST['password'])
user.save()
except Exception as e:
print(e)
return JsonFailResponse('create user failed')
else:
user_info = UserInfo()
user_info.user = user
user_info.save()
login(request, user)
return JsonSuccessResponse('register success')
def do_edit(request):
nickname = request.POST.get('nickname')
email = request.POST.get('email')
signature = request.POST.get('signature')
gender = request.POST.get('gender')
job = request.POST.get('job')
education = request.POST.get('education')
f_avatar = request.FILES.get('avatar')
user_info = request.user.userinfo or UserInfo()
if not nickname or len(nickname) > 50:
return JsonFailResponse('username is null or too long')
user_info.nickname = nickname
if not email or not re.match(EMAIL_PATTERN, email) or email_exists(request):
return JsonFailResponse('invalid email')
request.user.email = email
if signature:
user_info.signature = signature
if gender:
if int(gender) in [i[0] for i in UserInfo.GENDER_CHOICES]:
user_info.gender = gender
else:
return JsonFailResponse('invalid gender')
if job:
user_info.job = Job.objects.get(id=job)
if education:
user_info.education = education
if f_avatar:
ext = os.path.splitext(f_avatar.name)[1]
if ext not in UserInfo.AVATAR_EXT_LIMITS:
return JsonFailResponse(f'unsupported ext {ext}')
if f_avatar.size > UserInfo.AVATAR_SIZE_LIMIT:
return JsonFailResponse(f'image size is too big')
f_avatar_con = ContentFile(f_avatar.read())
f_avatar_name = request.user.username + ext
user_info.avatar.save(f_avatar_name, f_avatar_con, save=False)
img = Image.open(os.path.join(settings.MEDIA_ROOT, 'upload', 'images', 'avatar', f_avatar_name))
out = img.resize(UserInfo.AVATAR_SIZE_DEFAULT, resample=Image.ANTIALIAS)
out.save(f_avatar_name)
user_info.save()
return JsonSuccessResponse('edit success')
def setUp(self):
self.base_dir = os.path.join(os.path.dirname(os.path.dirname(views.__file__)), '.git')
self.user_info = UserInfo.create(username='******')
self.user_info.save()
self.workspace = Workspace.create(user_info=self.user_info, workspace='main')
self.workspace.save()
self.directory = Directory.create(git_directory=self.base_dir, git_shortname='name', workspace=self.workspace)
self.directory.save()
self.branch = Branch.create(git_branch='branch_name', directory=self.directory)
self.branch.save()
def setUp(self):
self.user_info = UserInfo.create(username='******')
self.user_info.save()
self.workspace = Workspace.create(user_info=self.user_info,
workspace='main')
self.workspace.save()
self.directory = Directory.create(git_directory='pathname',
git_shortname='name',
workspace=self.workspace)
self.directory.save()
self.branch = Branch.create(git_branch='master',
directory=self.directory)
self.branch.save()
def setUp(self):
self.base_dir = os.path.join(
os.path.dirname(os.path.dirname(views.__file__)), '.git')
user_info = UserInfo.create(username='******')
user_info.save()
self.user_info = user_info
workspace = Workspace.create(user_info=user_info,
workspace='workspace_name')
workspace.save()
directory = Directory.create(git_directory='pathname',
git_shortname='git_branch_name',
workspace=workspace)
directory.save()
branch = Branch.create(git_branch='branch_name', directory=directory)
branch.save()
def test_fields(self):
user = User.objects.create(username='******')
user_info = UserInfo()
user_info.user = user
user_info.first_name = 'first_name'
user_info.last_name = 'last_name'
user_info.middle_name = 'middle_name'
user_info.gender = 'male'
user_info.occupation = 'developer'
user_info.save()
record = UserInfo.objects.get(pk=1)
self.assertEqual(record, user_info)
user_info.delete()
user.delete()