mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-11-24 17:50:20 -06:00
7c60fb3377
This uses a single default user in the config database, which is auto-logged into the app when SERVER_MODE = False. In this mode we also hide/remove user-related functions in the UI.
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2014, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
|
|
"""Perform the initial setup of the application, by creating the auth
|
|
and settings database."""
|
|
|
|
from flask import Flask
|
|
from flask.ext.sqlalchemy import SQLAlchemy
|
|
from flask.ext.security import Security, SQLAlchemyUserDatastore
|
|
from flask.ext.security.utils import encrypt_password
|
|
from settings_model import db, Role, User
|
|
|
|
import getpass, os, random, sys, string
|
|
|
|
# Configuration settings
|
|
import config
|
|
|
|
app = Flask(__name__)
|
|
app.config.from_object(config)
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + config.SQLITE_PATH.replace('\\', '/')
|
|
db.init_app(app)
|
|
|
|
print "pgAdmin 4 - Application Initialisation"
|
|
print "======================================\n"
|
|
|
|
local_config = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'config_local.py')
|
|
if not os.path.isfile(local_config):
|
|
print "The configuration file %s does not exist.\n" % local_config
|
|
print "Before running this application, ensure that config_local.py has been created"
|
|
print "and sets values for SECRET_KEY, SECURITY_PASSWORD_SALT and CSRF_SESSION_KEY"
|
|
print "at bare minimum. See config.py for more information and a complete list of"
|
|
print "settings. Exiting..."
|
|
sys.exit(1)
|
|
|
|
# Check if the database exists. If it does, tell the user and exit.
|
|
if os.path.isfile(config.SQLITE_PATH):
|
|
print "The configuration database %s already exists and will not be overwritten.\nExiting..." % config.SQLITE_PATH
|
|
sys.exit(1)
|
|
|
|
if config.SERVER_MODE == 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: '))
|
|
|
|
p1, p2 = pprompt()
|
|
while p1 != p2:
|
|
print('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)
|
|
user_datastore.add_role_to_user(email, 'Administrators')
|
|
db.session.commit()
|
|
|
|
# Done!
|
|
print ""
|
|
print "The configuration database has been created at %s" % config.SQLITE_PATH
|