def subEvent(msg):
data = tools.client.user.get(msg.source)
try:
username = data["nickname"]
users = User.objects.filter(openId=msg.source)
if len(users) > 0:
user = users[0]
else:
user = User(name=data["nickname"],
sex=int(data["sex"]),
openId=msg.source,
comment=u"现在还是空的><")
user.save()
team = Team(name=basic_tools.teamName(None, username), type=0)
team.save()
team.members.add(user)
user.name = username
user.uid = int(random.random() * 100)
if "headimgurl" in data:
user.image = data["headimgurl"]
user.save()
date = basic_tools.getDate()
#如果DayData不存在，则创建
if DayData.objects.filter(user=user, date=date).count() == 0:
dayData = DayData(user=user, date=date)
basic_tools.updateDayData(dayData, user)
dayData.save()
except:
username = ""
return HttpResponse(
create_reply(u"%s!欢迎使用新中韩无敌了的手环公众号!\n%s" % (username, tools.help_text),
message=msg))
def callback(request):
# Get the state saved in session
expected_state = request.session.pop('auth_state', '')
# Make the token request
token = get_token_from_code(request.get_full_path(), expected_state)
# Get the user's profile
user = get_user(token)
print(profstat)
print(studstat)
print(user)
# Save token and user
store_token(request, token)
store_user(request, user)
if User.objects.filter(username=user['surname']).exists():
print("Exists")
else:
new_user = User()
new_user.is_student = studstat
new_user.is_professor = profstat
new_user.username = user['surname']
new_user.first_name = user['displayName'].split(" ")[0]
new_user.last_name = user['displayName'].split(" ")[1]
new_user.email = user['mail'] if (
user['mail'] != None) else user['userPrincipalName']
print(new_user)
new_user.save()
store(new_user, studstat, profstat)
if studstat == True:
return HttpResponseRedirect(reverse('home'))
else:
return HttpResponseRedirect(reverse('profhome'))
def setUp(self):
self.user = User(openId="ose6Ut8Ir-41wB7gQx89BifYa49Q",name="test")
self.user.save()
self.data = {
"plan_name": "test_name",
"begintime": u"2016-01-01 10:10",
"endtime": u"2016-01-04 10:10",
"goal": 10000,
"tag0": "test_tag_0",
"tag1": "test_tag_1",
"tags": "test_tag_s",
"friend0": self.user.openId,
"comment": "test_comment"
}
def register():
if current_user.is_authenticated:
return redirect(url_for('main.hello'))
form = RegisterForm()
if form.validate_on_submit():
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)
db.session.add(user)
db.session.commit()
flash('Your account have been created! you may now login !', 'success')
return redirect(url_for('main.hello'))
return render_template('register.html', title='Register', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
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) #creating account
db.session.add(user) #adding user to the database
db.session.commit()
flash('Your account has been created! You are now able to log in',
category='success')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
