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()