mirror of
https://github.com/apache/superset.git
synced 2026-04-13 05:07:53 +00:00
ebaad10d6c
Co-authored-by: Antonio Rivero Martinez <38889534+Antonio-RiveroMartnez@users.noreply.github.com> Co-authored-by: Elizabeth Thompson <eschutho@gmail.com>
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from typing import Any, Dict
|
|
|
|
import sqlalchemy as sa
|
|
from flask import current_app
|
|
from flask_appbuilder import Model
|
|
from sqlalchemy.orm import backref, relationship
|
|
from sqlalchemy_utils import EncryptedType
|
|
|
|
from superset.models.core import Database
|
|
from superset.models.helpers import (
|
|
AuditMixinNullable,
|
|
ExtraJSONMixin,
|
|
ImportExportMixin,
|
|
)
|
|
|
|
app_config = current_app.config
|
|
|
|
|
|
class SSHTunnel(Model, AuditMixinNullable, ExtraJSONMixin, ImportExportMixin):
|
|
"""
|
|
A ssh tunnel configuration in a database.
|
|
"""
|
|
|
|
__tablename__ = "ssh_tunnels"
|
|
|
|
id = sa.Column(sa.Integer, primary_key=True)
|
|
database_id = sa.Column(
|
|
sa.Integer, sa.ForeignKey("dbs.id"), nullable=False, unique=True
|
|
)
|
|
database: Database = relationship(
|
|
"Database",
|
|
backref=backref("ssh_tunnels", uselist=False, cascade="all, delete-orphan"),
|
|
foreign_keys=[database_id],
|
|
)
|
|
|
|
server_address = sa.Column(sa.Text)
|
|
server_port = sa.Column(sa.Integer)
|
|
username = sa.Column(EncryptedType(sa.String, app_config["SECRET_KEY"]))
|
|
|
|
# basic authentication
|
|
password = sa.Column(
|
|
EncryptedType(sa.String, app_config["SECRET_KEY"]), nullable=True
|
|
)
|
|
|
|
# password protected pkey authentication
|
|
private_key = sa.Column(
|
|
EncryptedType(sa.String, app_config["SECRET_KEY"]), nullable=True
|
|
)
|
|
private_key_password = sa.Column(
|
|
EncryptedType(sa.String, app_config["SECRET_KEY"]), nullable=True
|
|
)
|
|
|
|
@property
|
|
def data(self) -> Dict[str, Any]:
|
|
return {
|
|
"server_address": self.server_address,
|
|
"server_port": self.server_port,
|
|
"username": self.username,
|
|
}
|