def create_fake_users(db, nb_users=50, fake=None):
from app.models import User
from app.models import UserRole
from app.models import Whitelist
if fake is None:
fake = Faker()
#logging.getLogger('faker.factory').setLevel(logging.ERROR)
wl1 = Whitelist(label=fake.word())
admin = UserRole(label=fake.word())
contributor = UserRole(label=fake.word())
db.session.add(wl1)
db.session.add(admin)
db.session.add(contributor)
db.session.commit()
roles = [admin, contributor]
whitelists = [wl1]
for i in range(0, nb_users):
u1 = User(username=fake.user_name(),
password=fake.user_name(),
email=fake.free_email(),
active=1,
first_name=fake.first_name(),
last_name=fake.last_name())
u1.role = roles[1]
u1.whitelists = whitelists
db.session.add(u1)
db.session.commit()
def add():
"""Add `email` to the whitelist."""
form = AddToWhitelistForm(request.form)
if form.user_type.data == 'fake_user':
if form.validate_on_submit():
fake_id = str(uuid.uuid4())
fake_email = fake_id[:10] + "@fake-users.com"
filename = form.fake_user_image.data
try:
fake_image = Image.objects().get(filename=filename)
fake_user = User(email=fake_email,
gplus_id=fake_id,
name=form.name.data,
user_type=form.user_type.data,
image=fake_image)
except Image.DoesNotExist:
fake_user = User(email=fake_email,
gplus_id=fake_id,
name=form.name.data,
user_type=form.user_type.data)
fake_user.save()
else:
app.logger.warning(form.errors)
else:
user_exists = User.objects(email=form.email.data).count() != 0
if form.validate_on_submit() and not user_exists:
wl = Whitelist(email=form.email.data,
user_type=form.user_type.data)
wl.save()
else:
app.logger.warning(form.errors)
return redirect(url_for('users.index'))
def create_fake_users(db, nb_users=50, fake=None):
from app.models import User
from app.models import UserRole
from app.models import Whitelist
if fake is None:
fake = Faker()
wl1 = Whitelist(label=fake.word())
admin = UserRole(label=fake.word())
contributor = UserRole(label=fake.word())
db.session.add(wl1)
db.session.add(admin)
db.session.add(contributor)
db.session.commit()
roles = [admin, contributor]
whitelists = [wl1]
try:
for i in range(0, nb_users):
u1 = User(username=fake.user_name(),
password=fake.user_name(),
email=fake.free_email(),
active=random.choice([0, 1]),
first_name=fake.first_name(),
last_name=fake.last_name())
u1.role = random.choice(roles)
u1.whitelists = [random.choice(whitelists)]
db.session.add(u1)
db.session.commit()
except IntegrityError as e:
db.session.rollback()
def api_post_whitelist(api_version):
"""
{
data: {
label : 'My new whitelist'
}
}
:return:
"""
data = request.get_json()
try:
new_whitelist_name = data.get("label")
whitelist = Whitelist(label=new_whitelist_name)
whitelist.users = [current_app.get_current_user()]
db.session.add(whitelist)
db.session.commit()
return make_200(data=whitelist.serialize())
except Exception as e:
db.session.rollback()
print(str(e))
return make_400(str(e))
