mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-11-27 19:20:21 -06:00
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2017, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
|
|
from flask import current_app
|
|
|
|
from .registry import DriverRegistry
|
|
|
|
|
|
def get_driver(type, app=None):
|
|
if app is not None:
|
|
DriverRegistry.load_drivers()
|
|
|
|
drivers = getattr(app or current_app, '_pgadmin_server_drivers', None)
|
|
|
|
if drivers is None or not isinstance(drivers, dict):
|
|
drivers = dict()
|
|
|
|
if type in drivers:
|
|
return drivers[type]
|
|
|
|
driver = DriverRegistry.create(type)
|
|
|
|
if driver is not None:
|
|
drivers[type] = driver
|
|
setattr(app or current_app, '_pgadmin_server_drivers', drivers)
|
|
|
|
return driver
|
|
|
|
|
|
def init_app(app):
|
|
drivers = dict()
|
|
|
|
setattr(app, '_pgadmin_server_drivers', drivers)
|
|
DriverRegistry.load_drivers()
|
|
|
|
return drivers
|
|
|
|
|
|
def ping():
|
|
drivers = getattr(current_app, '_pgadmin_server_drivers', None)
|
|
|
|
for type in drivers:
|
|
drivers[type].gc()
|