def run(self, user): from app import db user_datastore = SQLAlchemyUserDatastore(db, User, Role) u = User.query.filter_by(email=user).first() role = user_datastore.find_or_create_role('admin') user_datastore.add_role_to_user(u, role) db.session.commit()
def run(self, user): from app import db user_datastore = SQLAlchemyUserDatastore(db, User, Role) u = User.query.filter_by(email=user).first() role = user_datastore.find_or_create_role('admin') user_datastore.add_role_to_user(u, role) db.session.commit()
def do_setup(app): """Create a new settings database from scratch""" if config.SERVER_MODE is False: print("NOTE: Configuring authentication for DESKTOP mode.") email = config.DESKTOP_USER p1 = ''.join([ random.choice(string.ascii_letters + string.digits) for n in range(32) ]) else: print("NOTE: Configuring authentication for SERVER mode.\n") # Prompt the user for their default username and password. print(""" Enter the email address and password to use for the initial pgAdmin user \ account:\n""") email = '' while email == '': email = input("Email address: ") def pprompt(): return getpass.getpass(), getpass.getpass('Retype password:'******'Passwords do not match. Try again') p1, p2 = pprompt() # Setup Flask-Security user_datastore = SQLAlchemyUserDatastore(db, User, Role) security = Security(app, user_datastore) with app.app_context(): password = encrypt_password(p1) db.create_all() user_datastore.create_role(name='Administrators', description='pgAdmin Administrators Role') user_datastore.create_user(email=email, password=password) db.session.flush() user_datastore.add_role_to_user(email, 'Administrators') # Get the user's ID and create the default server group user = User.query.filter_by(email=email).first() server_group = ServerGroup(user_id=user.id, name="Servers") db.session.merge(server_group) # Set the schema version version = Version(name='ConfigDB', value=config.SETTINGS_SCHEMA_VERSION) db.session.merge(version) db.session.commit() # Done! print("") print("The configuration database has been created at {0}".format( config.SQLITE_PATH))
def do_setup(app): """Create a new settings database from scratch""" if config.SERVER_MODE is False: print("NOTE: Configuring authentication for DESKTOP mode.") email = config.DESKTOP_USER p1 = "".join([random.choice(string.ascii_letters + string.digits) for n in xrange(32)]) else: print("NOTE: Configuring authentication for SERVER mode.\n") # Prompt the user for their default username and password. print("Enter the email address and password to use for the initial pgAdmin user account:\n") email = "" while email == "": email = raw_input("Email address: ") pprompt = lambda: (getpass.getpass(), getpass.getpass("Retype password: "******"Passwords do not match. Try again") p1, p2 = pprompt() # Setup Flask-Security user_datastore = SQLAlchemyUserDatastore(db, User, Role) security = Security(app, user_datastore) with app.app_context(): password = encrypt_password(p1) db.create_all() user_datastore.create_role(name="Administrators", description="pgAdmin Administrators Role") user_datastore.create_user(email=email, password=password) db.session.flush() user_datastore.add_role_to_user(email, "Administrators") # Get the user's ID and create the default server group user = User.query.filter_by(email=email).first() server_group = ServerGroup(user_id=user.id, name="Servers") db.session.merge(server_group) # Set the schema version version = Version(name="ConfigDB", value=config.SETTINGS_SCHEMA_VERSION) db.session.merge(version) db.session.commit() # Done! print("") print("The configuration database has been created at %s" % config.SQLITE_PATH)
def do_setup(app): """Create a new settings database from scratch""" if config.SERVER_MODE is False: print("NOTE: Configuring authentication for DESKTOP mode.") email = config.DESKTOP_USER p1 = ''.join([ random.choice(string.ascii_letters + string.digits) for n in range(32) ]) else: print("NOTE: Configuring authentication for SERVER mode.\n") # Prompt the user for their default username and password. print(""" Enter the email address and password to use for the initial pgAdmin user \ account:\n""") email_filter = re.compile( "^[a-zA-Z0-9.!#$%&'*+\/=?^_`{|}~-]+@[a-zA-Z0-9]" "(?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9]" "(?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*$") email = '' input("Email address: ") while email == '' or not email_filter.match(email): print('Invalid email address. Please try again.') email = input("Email address: ") def pprompt(): return getpass.getpass(), getpass.getpass('Retype password:'******'Passwords do not match. Please try again.') p1, p2 = pprompt() # Setup Flask-Security user_datastore = SQLAlchemyUserDatastore(db, User, Role) Security(app, user_datastore) with app.app_context(): password = encrypt_password(p1) db.create_all() user_datastore.create_role( name='Administrator', description='pgAdmin Administrator Role' ) user_datastore.create_role( name='User', description='pgAdmin User Role' ) user_datastore.create_user(email=email, password=password) db.session.flush() user_datastore.add_role_to_user(email, 'Administrator') # Get the user's ID and create the default server group user = User.query.filter_by(email=email).first() server_group = ServerGroup(user_id=user.id, name="Servers") db.session.merge(server_group) # Set the schema version version = Version( name='ConfigDB', value=config.SETTINGS_SCHEMA_VERSION ) db.session.merge(version) db.session.commit() # Done! print("") print( "The configuration database has been created at {0}".format( config.SQLITE_PATH ) )
def init_db(): fake_app = create_app() with fake_app.test_request_context(): db.create_all() datastore = SQLAlchemyUserDatastore(db, User, Role) print("\n** ADDING ROLES **") for role in ['admin', 'manager']: try: print("adding role '%s'" % role) datastore.create_role(name=role, description=role) datastore.commit() print(" ...OK") except IntegrityError as r: print(" '%s' role already exists" % role) db.session.rollback() print("\n** ADDING USERS **") users = [ fake_app.config['ADMIN_USER'], fake_app.config['MANAGER_USER'], fake_app.config['CLERK_USER'] ] for user in users: if user: name = user['name'] email = user['email'] role = user['role'] password = user['password'] try: print("adding user '%s'" % name) password = encrypt_password(password) datastore.create_user(email=email, password=password, name=name, active=True) datastore.commit() print(" ...OK") except IntegrityError as e: print(" user '%s' already exists" % name) db.session.rollback() if role: try: print("adding '%s' role to user '%s'" % (role, name)) this_user = db.session.query(User)\ .filter(User.name == name).first() this_role = db.session.query(Role).filter( Role.name == role).first() datastore.add_role_to_user(this_user, this_role) datastore.commit() print(" ...OK") except IntegrityError as e: print(" unable to add role '%s' to user '%s'" % (role, name)) from alembic.config import Config from alembic import command path = os.path.join(os.path.dirname(__file__), 'alembic.ini') alembic_cfg = Config(path) command.stamp(alembic_cfg, 'head')