def update_user_info(self):
"""
更新用户信息
:return: user模型实例
"""
user = self.user
user.email = self.email.data
user.role = User.ROLE_SEEKER
if self.is_new_user:
user.password = self.password.data
user.username = self.username.data
else:
if self.password.data:
user.password = self.password.data
db.session.add(user)
db.session.commit()
seeker = Seeker(user=user)
seeker.gender = self.gender.data
seeker.phone = self.phone.data
seeker.name = self.name.data
seeker.college = self.college.data
seeker.education = self.education.data
seeker.major = self.major.data
seeker.service_year = self.service_year.data
db.session.add(seeker)
db.session.commit()
return user
def update_user(self):
seeker = Seeker()
# username 不能更改
# user.username = self.username.data
# 连接 user 表
seeker.user = current_user
seeker.gender = self.gender.data
seeker.phone = self.phone.data
seeker.name = self.name.data
seeker.college = self.college.data
seeker.education = self.education.data
seeker.major = self.major.data
seeker.service_year = self.service_year.data
db.session.add(seeker)
db.session.commit()
return seeker
def register(self):
user = User(email=self.email.data,
password=self.password.data,
role=User.ROLE_SEEKER)
seeker = Seeker(user=user, name=self.name.data)
db.session.add(user)
db.session.add(seeker)
db.session.commit()
def create_user(self):
user = User(name=self.name.data,
email=self.email.data,
password=self.password.data)
db.session.add(user)
db.session.commit()
seeker = Seeker(user_id=user.id)
db.session.add(seeker)
db.session.commit()
return user
def iter_seeker():
users = User.query.filter_by(role=20).all()
for user in users:
yield Seeker(user_id=user.id,
gender=choice([10, 20]),
phone=fake.phone_number(),
name=fake.name(),
college='清华大学',
education=choice([1, 2, 3, 4, 5]),
major=fake.job(),
service_year=randint(0, 11))
def save(self):
user = User(email=self.email.data,
password=self.password.data,
role=self.role.data)
db.session.add(user)
#
if user.is_company:
company = Company(user=user, name=self.name.data)
db.session.add(company)
elif user.is_seeker:
seeker = Seeker(user=user, name=self.name.data)
db.session.add(seeker)
db.session.commit()
def create_seeker(self):
seeker = Seeker()
seeker.user.id = current_user.id
seeker.name = self.name.data
seeker.gender = self.gender.data
seeker.age = self.age.data
seeker.education = self.education.data
seeker.college = self.college.data
seeker.major = self.major.data
seeker.working_years = self.working_years.data
seeker.current_position = self.current_position.data
seeker.except_position = self.except_position.data
db.session.add(seeker)
db.session.commit()
return seeker