def insert_admin():
from app.models import User, Role
user = User()
user.role = Role.ROLE_ADMIN
user.username = '******'
user.password = '******'
user.about_me = 'i am admin'
db.session.add(user)
try:
db.session.commit()
except Exception:
db.session.rollback()
def add_user():
form = AddUserFrom()
if form.validate_on_submit():
user = User()
user.name = form.name.data
user.confirmed = form.confirmed.data
user.location = form.location.data
user.role_id = form.role_id.data
user.about_me = form.about_me.data
user.password = form.password.data
db.session.add(user)
db.session.commit()
# user.API_KEY = user.generate_confirmed_token()
return redirect(url_for('manager.manager_all_users'))
return render_template('manager/add_user.html', form=form)
def generate_fake_users(count=100):
seed()
for i in range(count):
u = User(email=forgery_py.internet.email_address())
u.username = forgery_py.internet.user_name(True)
u.password = forgery_py.lorem_ipsum.word()
u.confirmed = True
u.name = forgery_py.name.full_name()
u.location = forgery_py.address.city()
u.about_me = forgery_py.lorem_ipsum.sentence()
u.member_since = forgery_py.date.date(True)
db.session.add(u)
try:
db.session.commit()
except IntegrityError:
db.session.rollback()
def test_edit_profile(self):
#original user
user_edit_profile = User(username="******", about_me="")
db.session.add(user_edit_profile)
db.session.commit()
#user goes to update profile
user_edit_profile.username = "******"
user_edit_profile.about_me = "test about me functionality"
db.session.add(user_edit_profile)
db.session.commit()
#test to see if the user's username has been updated
self.assertEqual(user_edit_profile.username, "testedit")
self.assertNotEqual(user_edit_profile.username, "originaluser")
self.assertNotEqual(user_edit_profile.about_me, "")
def manager_add_user():
form = UserForm()
if form.validate_on_submit():
user = User()
user.email = form.email.data
user.name = form.name.data
user.password = form.password.data
user.role_id = form.role_id.data
user.about_me = form.about_me.data
user.location = form.location.data
user.confirmed = form.confirmed.data
print(form.confirmed.data)
db.session.add(user)
db.session.commit()
return redirect(url_for('.manager_user'))
return render_template('manager/manager_add_user.html', form=form)
def _update_user_data(form, user=None):
if not user:
user = User()
else:
user.user_type = \
UserType.query.filter_by(name=form.user_type.data).one()
user.username = form.username.data
user.email = form.email.data
user.th_firstname = form.th_firstname.data
user.th_lastname = form.th_lastname.data
user.en_firstname = form.en_firstname.data
user.en_lastname = form.en_lastname.data
user.about_me = form.about_me.data
user.title = Title.query.get(form.title.data)
user.department = Department.query.get(form.department.data)
user.job = Job.query.get(form.job.data)
academic_position = \
AcademicPosition.query.get(form.academic_position.data)
if not user.faculty_info:
user.faculty_info = FacultyInfo(
car_license_plate=form.car_license_plate.data,
department_head=form.department_head.data,
academic_position=academic_position,
)
else:
user.faculty_info.car_license_plate = \
form.car_license_plate.data
user.faculty_info.department_head = form.department_head.data
user.faculty_info.academic_position = academic_position
office = RoomDirectory.query.get(form.office.data)
if not user.contact:
user.contact = Contact(
office=office,
mobile_phone=form.mobile_phone.data,
fax=form.fax.data,
)
else:
user.contact.office = office
user.contact.mobile_phone = form.mobile_phone.data
user.contact.fax = form.fax.data
def edit_profile():
user = User(current_user)
form = EditProfileForm()
if form.validate_on_submit():
about_me = form.about_me.data
name = form.name.data
user.about_me = about_me
party = user.party.id
party = Party(party)
party.name = name
user.save()
party.save()
flash(_('Your changes have been saved.'))
return redirect(url_for('main.edit_profile'))
elif request.method == 'GET':
form.email.data = user.email
form.name.data = user.party.name
form.about_me.data = user.about_me
return render_template('edit_profile.html', title=_('Edit Profile'),
form=form,user=user)
def add_user():
form = UserProfileForm()
if request.method == 'GET':
form = UserProfileForm(id=0,
username='',
nickname='',
email='',
about_me='')
return render_template('auth/useredit.html',
title='',
form=form,
is_new=True)
else:
if form.validate_on_submit():
if User.query.filter_by(username=form.username.data).first():
flash('用户名已经被使用，请用另外用户名!')
return render_template('auth/useredit.html',
title='',
form=form,
postbackurl=url_for('auth.add_user'),
is_new=True)
user = User()
user.username = form.username.data
user.nickname = form.nickname.data
user.email = form.email.data
user.set_password('123456')
user.about_me = form.about_me.data
db.session.add(user)
db.session.commit()
#flash('用户添加成功!')
data = {'code': 200, 'message': '用户添加成功!'}
#flash('用户更新成功!')
return success_page("用户信息更新成功!")
else:
return render_template('auth/useredit.html',
title='',
form=form,
postbackurl=url_for('auth.add_user'),
is_new=True)
def createUsers():
names = [
'Abigail',
'Alice',
'Amber',
'Amelia',
'Amy',
'Ava',
'Brooke',
'Caitlin',
'Charlotte',
'Chloe',
'Daisy',
'Elizabeth',
'Ella',
'Ellie',
'Emily',
'Emma',
'Erin',
'Eva',
'Evie',
'Freya',
'Grace',
'Hannah',
'Holly',
'Imogen',
'Isabel',
'Isabella',
'Isabelle',
'Isla',
'Jasmine',
'Jessica',
'Katie',
'Keira',
'Leah',
'Lilly',
'Lily',
'Lola',
'Lucy',
'Matilda',
'Megan',
'Mia',
'Millie',
'Molly',
'Olivia',
'Phoebe',
'Poppy',
'Ruby',
'Scarlett',
'Sophia',
'Sophie',
'Summer',
'Adam',
'Alex',
'Alexander',
'Alfie',
'Archie',
'Benjamin',
'Callum',
'Cameron',
'Charlie',
'Connor',
'Daniel',
'Dylan',
'Edward',
'Ethan',
'Finley',
'George',
'Harrison',
'Harry',
'Harvey',
'Henry',
'Isaac',
'Jack',
'Jacob',
'Jake',
'James',
'Jayden',
'Joseph',
'Joshua',
'Leo',
'Lewis',
'Liam',
'Logan',
'Lucas',
'Luke',
'Matthew',
'Max',
'Mohammed',
'Muhammad',
'Nathan',
'Noah',
'Oliver',
'Oscar',
'Owen',
'Rhys',
'Riley',
'Ryan',
'Samuel',
'Thomas',
'Tyler',
'William',
]
for name in names:
u = User()
if User.query.filter_by(username=name).first():
u.username = name + str(random.randint(0, 99))
else:
u.username = name
u.email = u.username.lower() + '@luther.edu'
u.avatar = random.choice(['helmet.png', 'axe.png', 'ship.png'])
u.set_password('password')
aboutlen = random.randint(0, 23)
aboutme = lipsum.generate_words(aboutlen)
u.about_me = aboutme
db.session.add(u)
db.session.commit()
