def listuser(request):
# return render(request, 'template3.html', {'data': User.objects.all()})
if request.method == 'GET':
userid = request.GET.get('userid','')
if userid != "":
# User.objects.all().get(userid = userid).delete()
User.objects.all().filter(userid = userid)[0].delete()
return redirect('/listuser')
q = request.GET.get('q', '')
data = User.objects.all()
if q != '' :
data = User.objects.all().filter(name__contains = q)
return render(request, 'template2.html', {'data': data})
else:
Userid = request.POST.get('Userid')
Name = request.POST.get('Name')
Age = int(request.POST.get('Age'))
Hobby = request.POST.get('Hobby')
user = User(userid = Userid, name = Name, age = Age, hobby = Hobby)
user.save()
return redirect('/listuser')
def add_user(request):
if request.method == 'POST':
try:
payload = json.loads(request.body)
username = payload['username']
email = payload['email']
password = payload['password']
avatar = payload['avatar']
user_id = uuid.uuid4().hex[:10]
hashed_password = make_password(password)
user = User(user_id, username, email, hashed_password, avatar)
user_query = User.objects.filter(username=username).exists()
email_query = User.objects.filter(email=email).exists()
if user_query:
response = json.dumps([{'error': 'username exists'}])
elif email_query:
response = json.dumps([{'error': 'email exists'}])
else:
user.save()
response = json.dumps([{
'success': 'User added successfully!'
}])
except:
response = json.dumps([{'error': 'User could not be added!'}])
return HttpResponse(response, content_type='text/json')
else:
response = json.dumps([{'error': 'User could not be added!'}])
return HttpResponse(response, status='404', content_type='text/json')
def register(request):
role = Role.objects.get(name="user")
if request.method == "POST":
print("hello")
# first_name = request.POST.get('first_name')
# last_name = request.POST.get('last_name')
email = request.POST.get('email')
password = request.POST.get('password')
confirm_passsword = request.POST.get('confirm_passsword')
role_id = request.POST.get('role_id')
roleid = Role.objects.get(pk=role_id)
if password != confirm_passsword:
print(password)
messages.error(request,
"new password or confirm password not match")
return redirect(register)
else:
password = make_password(password)
user = User(username=email,
email=email,
password=password,
role_id=roleid)
user.save()
return redirect(login)
print("hiiii")
return render(request, 'user/registration.html', {'role': role})
def registerok(request):
email = request.POST.get("text1")
psd = request.POST.get("password")
print(psd)
sender = '*****@*****.**'
key = request.POST.get("key")
request.session['key'] = key
print(key)
url = "http://127.0.0.1/registerlock?username=%s&key=%s" % (email, key)
# stmp.stmpemail(sender,email,url)
subject = '旺享笔记注册'
message = '感谢您注册旺享笔记,这是您的注册激活链接:%s' % url
print(message)
sender = '*****@*****.**'
receivers = [email]
try:
send_mail(subject, message, sender, receivers, fail_silently=False)
print("发送成功!")
except:
return render(request, 'registerfail.html')
user = User(username=email,
psd=keyrandom.psdmd5(psd),
islock=True,
time=timezone.now())
user.save()
return render(request, 'registerok.html')
def test_user_to_json(self):
# No database dependencies here :)
user = User(first_name='blah', username='******', id='blah')
json_dict = user.to_json()
self.assertEqual(user.first_name, json_dict['first_name'])
self.assertEqual(user.username, json_dict['username'])
self.assertEqual(user.id, json_dict['id'])
def register():
if current_user.is_authenticated:
return redirect(
url_for('main.home')
) #si le user est authentifié il est renvoyé à la page d'acceuil
form = RegistrationForm()
if form.validate_on_submit(
): #si il valide son enregistrement, ses donnnées sont envoyées à la DB
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user = User(username=form.username.data,
email=form.email.data,
password=hashed_password,
admin=True,
confirmed=True,
publisher=False)
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
send_mail(user.email,
'Confirm your Account',
'registration_confirmation',
user=user,
token=token)
flash(
'A confirmation mail has been sent to you ! Check your mail box and confirm your account please !',
'success')
return redirect(url_for('users.login'))
return render_template('register.html', title='Register', form=form)
def test_user_set_password():
user = User(username="******")
assert user.salt == None
assert user.password_hash == None
user.set_password("test_password")
assert user.salt
assert user.password_hash and user.password_hash != "test_password"
def register(request):
if request.method == "POST":
username = request.POST['username']
email = request.POST['email']
password = request.POST['password']
confirmpassword = request.POST['confirmpassword']
# password = request.POST['username']
if len(username) > 10:
messages.error(request, "Username must be under the 10 characters")
return redirect(register)
if password != confirmpassword:
messages.error(request,
"new password or confirm password not match")
return redirect(register)
else:
password = make_password(password)
user = User(username=username, email=email, password=password)
subject = "Greeting"
message = "successfully account created"
to = email
res = send_mail(subject, message, settings.EMAIL_HOST_USER, [to])
if res == 1:
print("mail send successfully")
else:
print("mail not sent")
user.save()
return redirect(login)
return render(request, "myapp/register.html")
def create_user():
if 'username' not in request.json or 'first_name' not in request.json:
return Response('Missing required arguments', status=400)
user = User(first_name=request.json['first_name'],
username=request.json['username'])
DB.session.add(user)
DB.session.commit()
return jsonify(user.to_json())
def register_api():
if request.method == "GET":
#此处可以重定向到登录页面
return "注册成功"
else:
#解析参数
params = request.form
u_phone = params.get("u_phone")
u_name = params.get("u_name")
u_password = params.get("u_password")
email = params.get("email")
u_icon = request.form.get("u_icon", None)
#校验参数
if u_name and u_password and u_phone and len(u_name) > 3 and len(
str(u_phone)) == 11:
#校验手机号
enc_phone = User.query.filter(User.u_phone == u_phone).all()
#若为True 则手机号注册过
if len(enc_phone) == 0:
# 给密码加密
enc_pwd_str = enc_pwd(u_password)
#实例化
u = User()
u.u_name = u_name
u.u_password = enc_pwd_str
#校验用户头像，若为None，则没上传
if u_icon == None:
return "没上传头像，请上传头像"
else:
#生成唯一头像名字
file_name = create_rand_str() + ".jpg"
#拼接文保存路径
file_path = os.path.join(BASE_DIR,
'static/icon' + file_name)
#保存文件
file = open(file_path, "wb")
for i in u_icon.chunks():
u.u_icon = file_path
#保存文件路径
#将用户对象保存到数据库
db.session.add(u)
db.session.commit()
res = send_mail(email, request.host)
# 设置缓存
cache.set(res, u.id, 60 * 60)
data = {"code": SUCCESS, "msg": '注册成功', "data": []}
return jsonify(data)
else:
#手机号已被注册
return jsonify({
"code": FAIL,
"msg": "手机号已被注册，请换个手机号重新注册",
"data": []
})
else:
#参数不合适
return jsonify({"code": NOT_LOGIN, "msg": "参数不对，请核对", "data": []})
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
return redirect(url_for('login'))
return render_template('register.html', title="Kaydol", form=form)
def signup():
if session.get("user_name"):
flash(
f"You are already a user {session.get('user_name')}. You need to logout if you want to create a new account!!",
category="success")
return redirect(url_for("mytodos"))
if request.method == "POST":
form_return_data = [
request.form.get("username"),
request.form.get("email"),
request.form.get("password"),
request.form.get("repeatpassword")
]
email_list = []
for i in User.query.all():
email_list.append(i.get("email"))
if request.form.get("email") not in email_list:
if request.form.get("password") == request.form.get(
"repeatpassword"):
user_list = []
for i in User.query.all():
user_list.append(i.get("name"))
if request.form.get("username") not in user_list:
try:
new_user = User(name=request.form.get("username"),
email=request.form.get("email"),
password="")
new_user.set_password(request.form.get("password"))
db.session.add(new_user)
db.session.commit()
session['redirect_to'] = "home"
return redirect(url_for("login"))
except Exception as e:
print(e)
flash(
"some error happened while trying to create your User Model!! Try Again",
category="model")
else:
flash("your username is already taken!!", category="user")
else:
flash("there was some error in your password!!",
category="error")
flash("your passwords don't match!!", category="password")
else:
flash("Sorry but there was some error with your email!!",
category="error")
return render_template("signup.html", form_data=form_return_data)
return render_template("signup.html")
def registerPage(request):
if request.method == "GET":
return render(request, "register.html")
else:
email1 = request.POST["email"]
password1 = make_password(request.POST["password"])
b = User(email=email1, password=password1)
b.save()
return redirect("/login")
def new_user(create_db):
user = User(username='******')
user.set_password('test_password')
db.session.add(user)
db.session.commit()
yield user
db.session.delete(user)
db.session.commit()
def weibo_back(request):
#获取code
code = request.GET.get('code')
#换取网址
access_token_url = "https://api.weibo.com/oauth2/access_token"
#发送请求进行换取
re_dict = requests.post(access_token_url,
data={
"client_id":
'2636039333',
"client_secret":
"4e2fbdb39432c31dc5c2f90be3afa5ce",
"grant_type":
"authorization_code",
"code":
code,
"redirect_uri":
"http://127.0.0.1:8000/md_admin/weibo"
})
print(re_dict.text)
#强转
re_dict = re_dict.text
re_dict = eval(re_dict)
#判断是否用新浪登陆过
user = User.objects.filter(username=str(re_dict['uid'])).first()
sina_id = ''
user_id = ''
if user:
#代表曾经登录过
sina_id = user.username
user_id = user.id
else:
#自动创建账号
user = User(username=str(re_dict['uid']), password='', type=0)
#保存
user.save()
#查询用户id
user = User.objects.filter(username=str(re_dict['uid'])).first()
sina_id = str(re_dict['uid'])
user_id = user.id
#跳转操作
return redirect('http://localhost:8080?sina_id=' + str(sina_id) + "&uid=" +
str(user_id))
def create_user(account):
""" Find existing user or create new user """
user = User(username=u'araikc',
password=u'alalo',
email=u'*****@*****.**',
role='admin',
confirmed=1)
user.account = account
db.session.add(user)
#db.session.commit()
return user
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def login():
user_name = request.form.get('username', '')
user_password = request.form.get('password', '')
if user_name != '' and user_password != '':
user = User.query.filter_by(account=user_name).first()
if not user:
user = User(name='', account=user_name)
user.save()
login_user(user)
return redirect('/#/index')
else:
return render_template('login.html')
def load_user():
with open('user.csv') as f:
mr = csv.reader(f)
data = list(mr)
for row in data:
if row[0] != '':
new_user = User()
new_user.username = row[0]
new_user.email = row[1]
db.session.add(new_user)
db.session.commit()
def add_user(request):
if request.method == 'GET':
return HttpResponse("Hi, here is SLL-'s web server.")
userId_ = request.POST.get('userId')
operate_ = request.POST.get('operate')
cntUser = User.objects.filter(userId=userId_).count()
cntEmployee = Employee.objects.filter(userId=userId_).count()
if operate_ == '2':
if cntUser == 1:
User.objects.get(userId=userId_).delete()
Employee.objects.get(userId=userId_).delete()
return HttpResponse('successful')
else:
return HttpResponse('failed')
yearMonth_ = request.POST.get('yearMonth')
name_ = request.POST.get('name')
phone_ = request.POST.get('phone')
address_ = request.POST.get('address')
identityCard_ = request.POST.get('identityCard')
if cntUser == 1:
if cntEmployee == 1:
res = Employee.objects.get(userId=userId_)
res.name = name_
res.phone = phone_
res.address = address_
res.identityCard = identityCard_
res.save()
else:
res = Employee(userId=userId_,
name=name_,
phone=phone_,
identityCard=identityCard_,
address=address_,
yearMonth=yearMonth_)
res.save()
else:
user_ = User(userId=userId_, password='******')
user_.save()
res = Employee(userId=userId_,
name=name_,
phone=phone_,
identityCard=identityCard_,
address=address_,
yearMonth=yearMonth_)
res.save()
return HttpResponse('successful')
def index(request):
if request.method == 'POST':
iname = request.POST.get('inputEmail3')
user = User(name=iname)
print(user.id)
user.save()
user = User.objects.get(name=iname)
print(user.id)
request.session['user_id'] = user.id
print(request.session.items())
context = {'name': iname}
return redirect('first')
return render(request, 'aut_in.html')
def load_or_creator_user(token):
'''根据token载入或创建用户
'''
import datetime
user = User.query.filter_by(uid = int(token.uid)).first()
if user is None:
print 'user is not created'
# create user
user_info = client.users.show.get(uid = token.uid)
user = User(token.uid, json.dumps(user_info), json.dumps(token))
user.active = True
user.create_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
user.last_login_time = user.create_time
db.session.add(user)
db.session.commit()
login_user(user)
else:
print 'user has been created'
# load user
user.token = json.dumps(token)
user_info = client.users.show.get(uid = token.uid)
user.info = json.dumps(user_info)
user.active = True
user.last_login_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
db.session.commit()
login_user(user)
def wb_back(request):
#接收参数
code = request.GET.get('code', None)
#定义token接口地址
url = "https://api.weibo.com/oauth2/access_token"
#定义参数
re = requests.post(url,
data={
"client_id": "2949825616",
"client_secret": "efcf35e94890965ae1d71eb7b971c693",
"grant_type": "authorization_code",
"code": code,
"redirect_uri":
"http://127.0.0.1:8000/md_admin/weibo"
})
print(re.json())
#换取新浪微博用户昵称
res = requests.get('https://api.weibo.com/2/users/show.json',
params={
'access_token': re.json()['access_token'],
'uid': re.json()['uid']
})
print(res.json())
sina_id = ''
user_id = ''
# 判断是否用新浪微博登录过
user = User.objects.filter(username=str(res.json()['name'])).first()
if user:
# 代表曾经用该账号登录过
sina_id = user.username
user_id = user.id
else:
# 首次登录，入库新浪微博账号
user = User(username=str(res.json()['name']), password='')
user.save()
user = User.objects.filter(username=str(res.json()['name'])).first()
sina_id = user.username
user_id = user.id
print(sina_id, user_id)
# 重定向
return redirect("http://localhost:8080?sina_id=" + str(sina_id) + "&uid=" +
str(user_id))
def addUser(args):
#if args[0] == addUser, adds user named args[1]
#returns "User <args[1]> added" or "User <args[1]> exists"
#else returns ""
if args[0] == "addUser":
user_count = User.objects.filter(name=args[1]).count()
if user_count > 0:
return "user " + args[1] + " exists"
user = User(name=args[1])
user.save()
return "user " + user.name + " added"
else:
return ""
def create_admin(create_db):
username = "******"
password = "******"
new_user = User(username=username)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
yield new_user
db.session.delete(new_user)
db.session.commit()
def register(request):
if request.method == "GET":
return render(request, 'register.html')
elif request.method == "POST":
account = request.POST.get('account')
password = request.POST.get('password')
username = request.POST.get('nickname')
passwd = request.POST.get('passwd')
verifycode = request.POST.get('verifycode')
verifycode = verifycode.upper()
print(print("verifycode:{}".format(verifycode)))
print(print("rand_str:{}".format(rand_str)))
print(verifycode == rand_str)
if verifycode != rand_str:
return HttpResponse('验证码错误')
if (passwd == password) and password and username and (verifycode
== rand_str):
try:
user = User()
user.name = username
user.password = generate_password(password)
user.token = uuid.uuid3(uuid.uuid4(), 'username')
user.account = account
user.save()
response = redirect('paixienet:mainweb')
response.set_cookie('token', user.token)
return response
except Exception as e:
return HttpResponse('注册失败')
else:
return HttpResponse('注册失败')
def test_create_user(self):
consumer = models.OauthConsumer.objects.create(name="smith")
user = User(consumer=consumer, name="smith")
user.save()
self.assertEqual(user.consumer.content_type_id, consumer.content_type_id)
self.assertEqual(user.consumer.id, consumer.id)
self.assertEqual(user.consumer.key, consumer.key)
self.assertEqual(user.consumer.name, consumer.name)
self.assertEqual(user.consumer.object_id, consumer.object_id)
self.assertEqual(user.consumer.secret, consumer.secret)
self.assertEqual(user.consumer.type, consumer.type)
self.assertEqual(user.name, "smith")
def UserUpdate(request, user_id):
userupdate = User.objects.get(user_id=user_id)
if request.method == "POST":
username = userupdate.username
email = userupdate.email
password = userupdate.password
first_name = request.POST['first_name']
last_name = request.POST['last_name']
user_group_id = request.POST['user_group_id']
user_group = UserGroup.objects.get(pk=user_group_id)
role_id = request.POST.get('role_id')
roleid = Role.objects.get(pk=role_id)
is_active = request.POST['is_active']
code = request.POST['code']
address = request.POST['address']
phone_number = request.POST['phone_number']
created_by = request.POST['created_by']
modified_by = request.POST['modified_by']
created_date = userupdate.created_date
ip = getmac.get_mac_address()
if request.FILES:
avatar = request.FILES['avatar']
else:
avatar = userupdate.avatar
user = User(user_id=user_id,
username=username,
email=email,
password=password,
first_name=first_name,
last_name=last_name,
user_group_id=user_group,
role_id=roleid,
avatar=avatar,
is_active=is_active,
code=code,
address=address,
phone_number=phone_number,
created_by=created_by,
modified_by=modified_by,
ip=ip,
created_date=created_date)
user.save()
return redirect(UserDetail)
usergroup = UserGroup.objects.all()
role = Role.objects.all()
return render(request, 'myapp/useradd.html', {
'usergroup': usergroup,
'role': role,
'userupdate': userupdate
})
def users_create():
t1 = time.clock()
for i in range(1, 1000):
ui = unicode(i)
u = User(email=ui+u'@qq.com', password=u'123', nickname=ui+u'q',
school=random.choice(schools), sex=random.choice(sex), status=random.choice(status), grade=u'国光13级硕士',
bday=u'1992-03-14', pubinfo=u'我的邮箱是123457568', contact=u'微信：6578678687', prinfo=u'我有很多很多秘密')
db.session.add(u)
db.session.add(u.follow(u))
db.session.commit() # 必须先提交。不然下面找不到id
"""自动在Visitors表里增加自己。 免得每次重复检测是否存在 add by lee @ 14.8.15"""
db.session.add(Visitors(user_id=u.id, visits=0))
db.session.commit()
t2 = time.clock()
return t2 -t1
def test_db(self):
assert User.query.all() == []
user = User(username="******", email="*****@*****.**")
db_session.add(user)
db_session.commit()
assert db_session.query(User).get(user.id) == user
user.username = "******"
db_session.commit()
assert db_session.query(User).get(user.id).username == "tester_fixed"
db_session.delete(user)
db_session.commit()
assert User.query.all() == []
def post(self):
args = self.reqparse.parse_args()
if User.query.filter_by(username=args['username']).first() is not None:
abort(400, message="Username already exists.")
if User.query.filter_by(email=args['email']).first() is not None:
abort(400, message="Email already exists.")
user = User(
email=args['email'],
username=args['username'] if args['username'] else args['email'])
if args['password']:
user.password = args['password']
db.session.add(user)
db.session.commit()
return {'user': marshal(user, user_fields)}, 201, \
{'Location': url_for('api.user', id=user.id)}
class UserTest(TestCase):
def setUp(self):
self.new_user = User(name='test', age=10)
self.new_user_if_exists_patcher = patch('myapp.models.User.if_exists')
self.new_user_if_exists_mock = self.new_user_if_exists_patcher.start()
def test_if_user_does_not_exist(self):
self.new_user_if_exists_mock.return_value = False
self.assertFalse(self.new_user.if_exists())
def test_if_user_exists(self):
new_user = User.objects.create(name='test',age=10)
self.assertTrue(new_user.if_exists())
def test_if_user_age_is_less_than_eighteen(self):
new_user = User(name='test', age=17)
self.assertFalse(new_user.if_adult())
def tearDown(self):
self.new_user_if_exists_patcher.stop()
def run(self, name, password, role):
if not name:
print "missing name"
return
if not password:
print "missing password"
return
user = User(name, password)
print "username: "******"role must be 0 or 1"
return
user.role = role
print "role: ", role
if password: print "password: ", password
db.session.add(user)
db.session.commit()
def register(request):
if request.method == "POST" :
if ('password' in request.POST.keys()):
print(request.POST)
username = request.POST['username']
password = request.POST['password']
email = request.POST['email']
testIdCode=request.POST['test']
if testIdCode==request.session['code'] :
user = User()
user.username = username
user.password = password
user.email = email
user.userimg='/static/img/jslogo.png'
user.save()
blog_list = BlogsPost.objects.all()
request.session['username'] = username
print(username)
print("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!")
for blog in blog_list:
blog.url = "/article/" + blog.title
comments=Comment.objects.filter(title=blog_list[0].title)
return HttpResponseRedirect('/')
#return render_to_response('home.html', {'posts': blog_list, 'post': blog_list[0], 'username': username,'comments':comments})
else:
return render_to_response('register.html')
else:
print("1")
req = json.loads((request.body).decode())
if not req["testEmail"]:
print("2")
response_data = {}
if req["username"]:
if User.objects.filter(username=req["username"]):
response_data['message'] = '该用户名已存在'
return HttpResponse(json.dumps(response_data), content_type="application/json")
else:
response_data['message'] = '您可以使用这个用户名'
return HttpResponse(json.dumps(response_data), content_type="application/json")
elif req["email"]:
if User.objects.filter(email=req["email"]):
response_data['message'] = '该邮箱已存在'
return HttpResponse(json.dumps(response_data), content_type="application/json")
elif req["requireIdcode"]:
response_data['message'] = '等待发送验证码'
sendEmail(req["email"],request)
return HttpResponse(json.dumps(response_data), content_type="application/json")
else:
response_data['message'] = '您可以使用这个邮箱'
return HttpResponse(json.dumps(response_data), content_type="application/json")
else:
return render_to_response('register.html')
def create_user(request):
if request.method == 'POST':
username, password = request.POST.get('username'), request.POST.get('password')
user = User.create(username, password)
return HttpResponseRedirect(reverse('user', args=(user.id, )))
def setUp(self):
self.new_user = User(name='test', age=10)
self.new_user_if_exists_patcher = patch('myapp.models.User.if_exists')
self.new_user_if_exists_mock = self.new_user_if_exists_patcher.start()
def test_if_user_age_is_less_than_eighteen(self):
new_user = User(name='test', age=17)
self.assertFalse(new_user.if_adult())
