mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-12-01 21:09:10 -06:00
116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2022, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
#########################################################################
|
|
|
|
"""This file contains functions fetching different utility paths."""
|
|
|
|
import os
|
|
|
|
from flask import current_app, url_for
|
|
from flask_security import current_user
|
|
from werkzeug.exceptions import InternalServerError
|
|
|
|
|
|
def get_storage_directory(user=current_user):
|
|
import config
|
|
if config.SERVER_MODE is not True:
|
|
return None
|
|
|
|
storage_dir = getattr(
|
|
config, 'STORAGE_DIR',
|
|
os.path.join(
|
|
os.path.realpath(
|
|
os.path.expanduser('~/.pgadmin/')
|
|
), 'storage'
|
|
)
|
|
)
|
|
|
|
if storage_dir is None:
|
|
return None
|
|
|
|
def _preprocess_username(un):
|
|
ret_un = un
|
|
if len(ret_un) == 0 or ret_un[0].isdigit():
|
|
ret_un = 'pga_user_' + un
|
|
|
|
ret_un = ret_un.replace('@', '_')\
|
|
.replace('/', 'slash')\
|
|
.replace('\\', 'slash')
|
|
|
|
return ret_un
|
|
|
|
username = _preprocess_username(user.username.split('@')[0])
|
|
|
|
# Figure out the old-style storage directory name
|
|
old_storage_dir = os.path.join(
|
|
storage_dir.decode('utf-8') if hasattr(storage_dir, 'decode')
|
|
else storage_dir,
|
|
username
|
|
)
|
|
|
|
username = _preprocess_username(user.username)
|
|
|
|
# Figure out the new style storage directory name
|
|
storage_dir = os.path.join(
|
|
storage_dir.decode('utf-8') if hasattr(storage_dir, 'decode')
|
|
else storage_dir,
|
|
username
|
|
)
|
|
|
|
# Rename an old-style storage directory, if the new style doesn't exist
|
|
if os.path.exists(old_storage_dir) and not os.path.exists(storage_dir):
|
|
current_app.logger.warning(
|
|
'Renaming storage directory %s to %s.',
|
|
old_storage_dir, storage_dir
|
|
)
|
|
os.rename(old_storage_dir, storage_dir)
|
|
|
|
if not os.path.exists(storage_dir):
|
|
os.makedirs(storage_dir, int('700', 8))
|
|
|
|
return storage_dir
|
|
|
|
|
|
def init_app(app):
|
|
import config
|
|
if config.SERVER_MODE is not True:
|
|
return None
|
|
|
|
storage_dir = getattr(
|
|
config, 'STORAGE_DIR',
|
|
os.path.join(
|
|
os.path.realpath(
|
|
os.path.expanduser('~/.pgadmin/')
|
|
), 'storage'
|
|
)
|
|
)
|
|
|
|
if storage_dir and not os.path.isdir(storage_dir):
|
|
if os.path.exists(storage_dir):
|
|
raise InternalServerError(
|
|
'The path specified for the storage directory is not a '
|
|
'directory.'
|
|
)
|
|
os.makedirs(storage_dir, int('700', 8))
|
|
|
|
if storage_dir and not os.access(storage_dir, os.W_OK | os.R_OK):
|
|
raise InternalServerError(
|
|
'The user does not have permission to read and write to the '
|
|
'specified storage directory.'
|
|
)
|
|
|
|
|
|
def get_cookie_path():
|
|
cookie_root_path = '/'
|
|
pgadmin_root_path = url_for('browser.index')
|
|
if pgadmin_root_path != '/browser/':
|
|
cookie_root_path = pgadmin_root_path.replace(
|
|
'/browser/', ''
|
|
)
|
|
return cookie_root_path
|