mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-11-26 02:30:21 -06:00
bc63652811
Issue: Changes done by Ashesh in pgAdmin4.py file for setting up PYTHONHOME variable to sys.prefix was applicable only for windows only. Additionally I have also added exception handling for file provided by user for Backup/Restore/Import/Export.
116 lines
4.1 KiB
Python
116 lines
4.1 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2017, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
|
|
"""This is the main application entry point for pgAdmin 4. If running on
|
|
a webserver, this will provide the WSGI interface, otherwise, we're going
|
|
to start a web server."""
|
|
|
|
import os
|
|
import sys
|
|
|
|
# We need to include the root directory in sys.path to ensure that we can
|
|
# find everything we need when running in the standalone runtime.
|
|
root = os.path.dirname(os.path.realpath(__file__))
|
|
if sys.path[0] != root:
|
|
sys.path.insert(0, root)
|
|
|
|
import config
|
|
from pgadmin import create_app
|
|
|
|
# Get the config database schema version. We store this in pgadmin.model
|
|
# as it turns out that putting it in the config files isn't a great idea
|
|
from pgadmin.model import SCHEMA_VERSION
|
|
config.SETTINGS_SCHEMA_VERSION = SCHEMA_VERSION
|
|
|
|
##########################################################################
|
|
# Sanity checks
|
|
##########################################################################
|
|
|
|
# Check if the database exists. If it does not, create it.
|
|
if not os.path.isfile(config.SQLITE_PATH):
|
|
from pgadmin.utils import u, fs_encoding, file_quote
|
|
setupfile = os.path.join(
|
|
os.path.dirname(os.path.realpath(u(__file__, fs_encoding))), u'setup.py'
|
|
)
|
|
exec(open(file_quote(setupfile), 'r').read())
|
|
|
|
##########################################################################
|
|
# Server starup
|
|
##########################################################################
|
|
|
|
# Create the app!
|
|
app = create_app()
|
|
|
|
if config.DEBUG:
|
|
app.debug = True
|
|
else:
|
|
app.debug = False
|
|
|
|
# Start the web server. The port number should have already been set by the
|
|
# runtime if we're running in desktop mode, otherwise we'll just use the
|
|
# Flask default.
|
|
PGADMIN_RUNTIME = False
|
|
if 'PGADMIN_PORT' in globals():
|
|
app.logger.debug('Running under the desktop runtime, port: %s',
|
|
globals()['PGADMIN_PORT'])
|
|
server_port = int(globals()['PGADMIN_PORT'])
|
|
PGADMIN_RUNTIME = True
|
|
elif 'PGADMIN_PORT' in os.environ:
|
|
port = os.environ['PGADMIN_PORT']
|
|
app.logger.debug(
|
|
'Not running under the desktop runtime, port: %s',
|
|
port)
|
|
server_port = int(port)
|
|
else:
|
|
app.logger.debug(
|
|
'Not running under the desktop runtime, port: %s',
|
|
config.DEFAULT_SERVER_PORT)
|
|
server_port = config.DEFAULT_SERVER_PORT
|
|
|
|
# Let the application save the status about the runtime for using it later.
|
|
app.PGADMIN_RUNTIME = PGADMIN_RUNTIME
|
|
|
|
# Set the key if appropriate
|
|
if 'PGADMIN_KEY' in globals():
|
|
app.PGADMIN_KEY = globals()['PGADMIN_KEY']
|
|
app.logger.debug("Desktop security key: %s" % app.PGADMIN_KEY)
|
|
else:
|
|
app.PGADMIN_KEY = ''
|
|
|
|
# Output a startup message if we're not under the runtime and startup.
|
|
# If we're under WSGI, we don't need to worry about this
|
|
if __name__ == '__main__':
|
|
if not PGADMIN_RUNTIME:
|
|
print("Starting %s. Please navigate to http://%s:%d in your browser." %
|
|
(config.APP_NAME, config.DEFAULT_SERVER, server_port))
|
|
sys.stdout.flush()
|
|
else:
|
|
# For unknown reason the Qt runtime does not pass the environment
|
|
# variables (i.e. PYTHONHOME, and PYTHONPATH), to the Python
|
|
# sub-processes, leading to failures executing background processes.
|
|
#
|
|
# This has been observed only on windows. On *nix systems, it is likely
|
|
# picking the system python environment, which is good enough to run
|
|
# the process-executor.
|
|
#
|
|
# Setting PYTHONHOME launch them properly.
|
|
from pgadmin.utils import IS_WIN
|
|
if IS_WIN:
|
|
os.environ['PYTHONHOME'] = sys.prefix
|
|
|
|
try:
|
|
app.run(
|
|
host=config.DEFAULT_SERVER,
|
|
port=server_port,
|
|
use_reloader=((not PGADMIN_RUNTIME) and app.debug),
|
|
threaded=config.THREADED_MODE
|
|
)
|
|
except IOError:
|
|
app.logger.error("Error starting the app server: %s", sys.exc_info())
|