def create_user(self, is_company=False):
if not is_company:
user = User(username=self.username.data,
email=self.email.data,
password=self.password.data)
else:
user = User(username=self.username.data,
email=self.email.data,
password=self.password.data,
role=User.ROLE_COMPANY)
db.session.add(user)
db.session.commit()
return user
def create_user(self):
user = User(username=self.username.data,
email=self.email.data,
password=self.password.data) #不用重新设置，默认就是我们需要的值
db.session.add(user)
db.session.commit()
return user
def create_user_and_company():
with open('scripts/data_company.json') as f1, \
open('scripts/data_job.json') as f2:
job_name_list = [i['name'] for i in json.load(f2)]
experience_list = ['不限', '1-3年', '3-5年', '5-10年']
degree_list = ['不限', '本科', '博士后']
for i in json.load(f1):
user = User(username=fake.name(), email=fake.email())
user.password = '******'
yield user
company = Company(
name = i['name'],
logo = i['logo'],
location = i['location'],
description = i['description'],
about = i['about'] if len(i['about']) > 9 else \
fake.sentence(nb_words=11),
tags = i['tags'],
user = user,
)
yield company
for i in range(randint(1, 5)):
job = Job(
name = job_name_list[randint(1, len(job_name_list))-1],
salary_low = randint(3, 8) * 1000,
salary_high = randint(9, 15) * 1000,
location = fake.address(),
is_fulltime = 1,
experience_requirement = experience_list[
randint(1, len(experience_list)) - 1],
degree_requirement = degree_list[
randint(1, len(degree_list)) - 1],
company = company
)
yield job
def create_user(self):
user = User()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
return user
class CompanyCreateForm(CompanyBaseForm):
is_new_user = True
user = User()
username = StringField('用户名', validators=[DataRequired(message='用户名不可为空'), Length(3, 24)])
password = PasswordField('密码', validators=[DataRequired(message='密码不可为空'), Length(6, 24, message='密码长度应为6~24个字符')])
repeat_password = PasswordField('重复密码', validators=[DataRequired(message='密码不可为空'), EqualTo('password', message='两次输入的密码不一致')])
submit = SubmitField('创建用户')
def __init__(self):
"""
初始化，主要为了把重写的 Fields 移到前面
"""
super(CompanyCreateForm, self).__init__()
self._fields.move_to_end(self.repeat_password.name, last=False)
self._fields.move_to_end(self.password.name, last=False)
self._fields.move_to_end(self.email.name, last=False)
self._fields.move_to_end(self.username.name, last=False)
self._fields.move_to_end(self.submit.name, last=True)
def validate_email(self, field):
if User.query.filter_by(email=field.data).first():
raise ValidationError('邮箱已经存在')
def validate_username(self, field):
user = User.query.filter_by(username=field.data).first()
if user:
raise ValidationError('用户名已存在')
def validate_name(self, field):
company = Company.query.filter_by(name=field.data).first()
if company:
raise ValidationError('该公司已注册')
def create_boss(company):
user = User(username="******",
email='*****@*****.**',
password='******',
role=User.ROLE_BOSS)
user.company = company
return user
def add_user(self):
user = User()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
flash("添加成功", 'success')
return user
def generate_data():
db.create_all()
with open(os.path.join(os.path.dirname(__file__), '.', 'datas',
'data.json'),
'r',
encoding='UTF-8') as file:
datas = json.load(file)
for data in datas:
if not data['company_name']:
data['company_name'] = fake.name()
user = User(username=data['company_name'],
email=fake.word() + '@163.com',
password='******',
role=User.ROLE_COMPANY)
company = ComInfo(com_phone=data['phone'],
com_web=data['website'],
com_location=data['company_locate'],
com_logo=data['company_logo'],
com_desc_less=data['desc_simple'],
com_desc_more=data['desc_more'],
user=user)
db.session.add(user)
db.session.add(company)
for key, value in data['jobs'].items():
if not key:
key = fake.word()
job = JobInfo(title=key,
work_place='not on earth',
tags='one,two,three',
company=company)
db.session.add(job)
db.session.commit()
def insert_company(company):
logo_base_url = 'https://www.lgstatic.com/'
# 获取到企业的logo地址，下载后放到项目文件中
logo_url = logo_base_url + company['companyLogo']
file_name = download_img(logo_url)
with app.app_context():
company_user = User()
company_user.name = company['companyFullName']
company_user.email = random_email()
company_user.password = '******'
company_user.role = User.ROLE_COMPANY
db.session.add(company_user)
db.session.commit()
print(company_user.id)
new_company = Company(lagou_company_id=company['companyId'],
name=company['companyFullName'],
logo=file_name if file_name else 'avatar.png',
short_description=company['companyFeatures'],
short_name=company['companyShortName'],
welfare=company['otherLabel'],
trade=company['industryField'],
funding=company['financeStage'],
size=company['companySize'],
city=company['city'],
user_id=company_user.id)
db.session.add(new_company)
db.session.commit()
def fake_company(self):
for company in self.lagou.company:
c = User(
name=company['companyShortName'],
email=fake.email(),
role=User.ROLE_COMPANY
)
c.password = '******'
db.session.add(c)
try:
db.session.commit()
except:
db.session.rollback()
continue
d = CompanyDetail(
logo='https://static.lagou.com/thumbnail_160x160/' + company['companyLogo'],
site='https://shiyanlou.com',
location=company['city'],
field=company['industryField'],
finance_stage=company['financeStage']
)
d.user_id = c.id
db.session.add(d)
db.session.commit()
def iter_admin():
yield User(
username=username,
email=email,
password=password,
role=30,
)
def iter_users():
for user in user_lst:
yield User(name=user[0],
username=user[1],
email=user[2],
password=user[3],
role=user[4])
def create_user(self): #在数据中创建User
user = User(username=self.name.data,
email=self.email.data,
password=self.password.data) #将form 相应数据传入
db.session.add(user)
db.session.commit()
return user
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()
return user
def register():
form = RegisterForm()
if form.validate_on_submit():
form.create_user(User())
flash('用户注册成功，请登录！', 'success')
return redirect(url_for('.login'))
return render_template('register.html', form=form)
def iter_users():
for company in company_detail:
name = company['name']
email = fake.email()
while email in email_set:
email = fake.email()
email_set.add(email)
yield User(username=name, email=email, password='******', role=30)
def create_company(self):
company = User(username=self.username.data,
email=self.email.data,
password=self.password.data,
role=20)
db.session.add(company)
db.session.commit()
return company
def iter_users():
company = Company.query.filter_by(id=1).first()
yield User(
name='company1',
username='******',
email='*****@*****.**',
password='******',
role=20,
company=company
)
yield User(
name='user1',
username='******',
email='*****@*****.**',
password='******',
role=10
)
def company_register():
form = RegisterForm()
form.username.label = u'企业名称'
if form.validate_on_submit():
form.create_user(User(role=User.ROLE_COMPANY))
flash('企业注册成功，请登录！', 'success')
return redirect(url_for('.login'))
return render_template('company_register.html', form=form)
def iter_user_data():
with open(os.path.join(os.path.dirname(__file__), 'company.json')) as f:
job_datas = json.load(f)
for job in job_datas:
yield User(username=fake.name(),
email=fake.email(),
password='******',
role=User.ROLE_VISTER)
def iter_user():
for i in range(60):
yield User(
name=str(f.random_int())+f.user_name(),
email=f.ascii_email(),
password='******',
role=f.random_element(elements=(10,20,30)),
)
def create_user(self, register_role):
user = User(username=self.username.data,
email=self.email.data,
password=self.password.data,
role=register_role)
db.session.add(user)
db.session.commit()
return user
def create_user(self):
user = User()
user.username = self.username.data
user.email = self.email.data
user.password = self.password.data
db.session.add(user)
db.session.commit()
return user
def fake_admin():
admin = User(username='******',
email='*****@*****.**',
password='******',
role=30)
db.session.add(admin)
db.session.commit()
def register(self):
user = User(email=self.email.data,
password=self.password.data,
role=User.ROLE_COMPANY)
company = Company(user=user, name=self.name.data)
db.session.add(user)
db.session.add(company)
db.session.commit()
def create_user(self):
user = User()
self.populate_obj(user)
db.session.add(user)
try:
db.session.commit()
except:
db.session.rollback()
return user
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_company(self):
user = User()
user.username = self.name.data
user.email = self.email.data
user.password = self.password.data
user.role = User.ROLE_COMPANY
db.session.add(user)
db.session.commit()
return user
def create_company(self, role):
user = User()
user.username = self.username.data
user.email = self.email.data
user.password = self.password.data
user.role = role
db.session.add(user)
db.session.commit()
return user
def create_user(self):
user = User() #添加企业用户的普通信息
user.name = self.name.data
user.email = self.email.data
user.password = self.image.data
user.role = 30 #管理员的权限是20
db.session.add(user)
db.session.commit()
return user
