def init_db(self):
engine = self.session.get_bind(mapper=None, clause=None)
inspector = Inspector.from_engine(engine)
if 'ab_user' not in inspector.get_table_names():
print "Security DB not found Creating..."
Base.metadata.create_all(engine)
print "Security DB Created"
self.migrate_db()
if self.session.query(Role).filter_by(name = self.auth_role_admin).first() is None:
role = Role()
role.name = self.auth_role_admin
self.session.add(role)
self.session.commit()
print "Inserted Role for public access", self.auth_role_admin
if not self.session.query(Role).filter_by(name = self.auth_role_public).first():
role = Role()
role.name = self.auth_role_public
self.session.add(role)
self.session.commit()
print "Inserted Role for public access", self.auth_role_public
if not self.session.query(User).all():
user = User()
user.first_name = 'Admin'
user.last_name = 'User'
user.username = '******'
user.password = '******'
user.active = True
user.role = self.session.query(Role).filter_by(name = self.auth_role_admin).first()
self.session.add(user)
self.session.commit()
print "Inserted initial Admin user"
print "Login using Admin/general"
def run(self):
admin_role = Role()
admin_role.name = 'admin'
admin_role.description = 'Umsjónarhlutverk með aðgang að notendastillingum'
user_role = Role()
user_role.name = 'Notandi'
user_role.description = 'Venjulegur notandi með grunn aðgang'
db.session.add(admin_role)
db.session.add(user_role)
db.session.commit()
def seed_roles(db):
print("++ Seeding roles")
role_usr = Role()
role_usr.name = "user"
role_usr.description = "Simple user"
role_adm = Role()
role_adm.name = "admin"
role_adm.description = "Admin user"
db.session.add(role_usr)
db.session.add(role_adm)
db.session.commit()
def load_roles():
role = Role()
role.name = "Client"
role.save()
print("Roles loaded")
def seed_users(db):
print("++ Seeding users")
role_usr = Role()
role_usr.name = "user"
role_usr.description = "Simple user"
role_adm = Role()
role_adm.name = "admin"
role_adm.description = "Admin user"
db.session.add(role_usr)
db.session.add(role_adm)
user_datastore.create_user(
email="*****@*****.**", password="******", name="toto", timezone="UTC", roles=[role_adm]
)
db.session.commit()
return
def create_role():
name = request.json.get("name", None)
role = Role()
role.name = request.json.get("name", "")
role.save()
return jsonify({"success": "Register Successfully"}), 200
async def get_role(self, ctx, *, role_name):
"""
Gives the user the requested custom role if not locked..
"""
found_role = Role.where('name', role_name)\
.where('guild_id', ctx.guild.id).first()
if found_role:
if found_role.is_locked:
await ctx.send("That custom role is currently locked and "
"cannot assigned, if you believe this is an "
"error then please contact an Admin")
return
discord_guild = self.bot.get_guild(found_role.guild_id)
discord_role = discord_guild.get_role(found_role.role_id)
if discord_role is None:
await ctx.send("That custom role appears to be broken, "
"please contact an admin")
return
await ctx.author.add_roles(discord_role)
await ctx.send("You have been given the custom role "
f"`{role_name}`")
return
for role in ctx.guild.roles:
if lev.distance(role_name.lower(), role.name.lower()) <= 1:
await ctx.send("That custom role does not exist but cannot be "
"created as it's name is too similar to the "
f"pre-existing role `{role.name}`")
return
result, _ = await ut.get_confirmation(
ctx.channel, ctx.author, self.bot,
"It looks like that custom role doesn't exist, "
"would you like to create it?")
if result:
role = Role()
role.name = role_name
role.guild_id = ctx.guild.id
role.created_by = ctx.author.id
discord_role = await ctx.guild.create_role(
name=role_name,
mentionable=True,
colour=Colour(random.choice(ROLE_COLOURS)),
reason="Auto-Generated")
role.role_id = discord_role.id
role.save()
await ctx.author.add_roles(discord_role)
await ctx.send(f"You have been given the role {role_name}")
def postRoles():
data = request.json
rol = Role() #PASOXXX
rol.name = data["name"]
rol.code = data["code"]
db.session.add(rol) #PASOXXX
db.session.commit()
return jsonify(rol.serialize()), 200
def create_role_body(errors, form):
role = Role()
role.name = form.name.data
role.description = form.description.data
try:
db.session.add(role)
db.session.commit()
except:
errors.append("Can't save in db role id{}".format(role.id))
return errors
def init_db(self):
try:
engine = self.session.get_bind(mapper=None, clause=None)
inspector = Inspector.from_engine(engine)
if 'ab_user' not in inspector.get_table_names():
log.info("Security DB not found Creating")
Base.metadata.create_all(engine)
log.info("Security DB Created")
self.migrate_db()
if self.session.query(Role).filter_by(name=self.auth_role_admin).first() is None:
role = Role()
role.name = self.auth_role_admin
self.session.add(role)
self.session.commit()
log.info("Inserted Role for public access %s" % (self.auth_role_admin))
if not self.session.query(Role).filter_by(name=self.auth_role_public).first():
role = Role()
role.name = self.auth_role_public
self.session.add(role)
self.session.commit()
log.info("Inserted Role for public access %s" % (self.auth_role_public))
if not self.session.query(User).all():
user = User()
user.first_name = 'Admin'
user.last_name = 'User'
user.username = '******'
user.password = generate_password_hash('general')
user.email = '*****@*****.**'
user.active = True
user.role = self.session.query(Role).filter_by(name=self.auth_role_admin).first()
self.session.add(user)
self.session.commit()
log.info("Inserted initial Admin user")
log.info("Login using Admin/general")
except Exception as e:
log.error("DB Creation and initialization failed, if just upgraded to 0.7.X you must migrate the DB. {0}".format(str(e)))
def reset_database():
DB.drop_all()
DB.create_all()
role = Role()
role.name = 'Administrator'
user = User()
user.public_id = 'd38924fb-9417-4a50-b715-01f805c28063'
# password
user.password = '******'
user.name = 'admin'
user.email = '*****@*****.**'
user.role = role
coffee_machine = CoffeeMachine()
coffee_machine.name = 'Winston'
coffee_machine.repository = 'hidden-firefly'
coffee_type = CoffeeType()
coffee_type.name = 'Arabica'
coffee_brand = CoffeeBrand()
coffee_brand.name = 'Dallmayr'
coffee_product = CoffeeProduct()
coffee_product.name = 'Dallmayr Prodomo'
coffee_product.coffee_brand = coffee_brand
coffee_product.coffee_type = coffee_type
profile = Profile()
profile.name = 'Morning'
profile.water_in_percent = 32 * 1000
profile.coffee_strength_in_percent = 42
profile.user = user
job = Job()
current_time = time.time()
job.create_date = current_time
job.coffee_strength_in_percent = 66
job.water_in_percent = 44
job.price = 10
job.doses = 1
job.user = user
job.coffee_machine = coffee_machine
job.coffee_product = coffee_product
DB.session.add(job)
DB.session.commit()
def add_role():
for index, breadcrumb in enumerate(session["breadcrumbs"]):
if "添加角色" == breadcrumb["text"]:
session["breadcrumbs"] = session["breadcrumbs"][: index + 1]
break
else:
session["breadcrumbs"].append({"text": "添加角色", "url": url_for("auth.add_role")})
if request.method == "POST":
other_role = Role.query.filter_by(name=request.form["name"]).first()
if other_role:
flash(u"同名角色已存在", "danger")
else:
all_id = [str(menu_id) for menu_id in request.form.getlist("role_menu") if menu_id]
query_sql = "SELECT DISTINCT parent_id FROM tms_menu WHERE id IN(%s)" % ",".join(all_id)
for row in db.engine.execute(query_sql):
if row[0] != "None" and row[0] != None:
current_app.logger.info(row[0])
all_id.append(str(row[0]))
role = Role()
role.menu = ",".join(list(set(all_id)))
role.name = request.form["name"]
role.index_page = request.form["index_page"]
db.session.add(role)
db.session.commit()
users = User.query.filter(User.id.in_(request.form.getlist("role_remember")))
for user in users:
user.role = role
db.session.add(user)
db.session.commit()
return redirect(url_for("auth.role_page"))
menus = SystemMenu.query.filter_by(activate=True).order_by(SystemMenu.order).all()
users = User.query.all()
return render_template(
"admin/account/role_edit.html", role=None, menus=menus, users=users, current_page="auth.role_page"
)
import os
import sys
from flask_security import utils
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from app import create_app
from models import Page, db, Role, User
app = create_app()
with app.app_context():
admin_role = Role()
admin_role.name = 'admin'
db.session.add(admin_role)
db.session.commit()
root = User()
root.email = '*****@*****.**'
root.password = utils.hash_password("123456")
root.active = True
root.roles.append(admin_role)
db.session.add(root)
db.session.commit()
page = Page()
page.title = "Home Page"
page.content = "<h1><b>Hello from flask - docker!<b></h1>"
page.is_homepage = True
from models import Role, database r = Role() r.name = 'conversationee' r.description = 'day-to-day system user' with database.transaction(): r.save()
def _save_role(cd):
role = Role()
role.id = cd['id'] or None
role.name = cd['name']
role.save()
return role
