mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-08-27 18:36:08 +00:00
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
|
# Copyright 2021 Collate
|
||
|
|
# Licensed 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.
|
||
|
|
|
||
|
|
"""
|
||
|
|
Restore utility for the metadata CLI
|
||
|
|
"""
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import List, Optional
|
||
|
|
|
||
|
|
import click
|
||
|
|
from sqlalchemy.engine import Engine
|
||
|
|
|
||
|
|
from metadata.cli.utils import get_engine
|
||
|
|
from metadata.utils.logger import cli_logger
|
||
|
|
|
||
|
|
logger = cli_logger()
|
||
|
|
|
||
|
|
|
||
|
|
def execute_sql_file(engine: Engine, input: Path, schema: str = None) -> None:
|
||
|
|
|
||
|
|
with open(input, encoding="utf-8") as f:
|
||
|
|
for query in f.readlines():
|
||
|
|
# `%` is a reserved syntax in SQLAlchemy to bind parameters. Escaping it with `%%`
|
||
|
|
clean_query = query.replace("%", "%%")
|
||
|
|
|
||
|
|
with engine.connect() as conn:
|
||
|
|
conn.execute(clean_query)
|
||
|
|
|
||
|
|
|
||
|
|
def run_restore(
|
||
|
|
host: str,
|
||
|
|
user: str,
|
||
|
|
password: str,
|
||
|
|
database: str,
|
||
|
|
port: str,
|
||
|
|
input: str,
|
||
|
|
options: List[str],
|
||
|
|
arguments: List[str],
|
||
|
|
schema: Optional[str] = None,
|
||
|
|
) -> None:
|
||
|
|
"""
|
||
|
|
Run and restore the
|
||
|
|
buckup. Optionally, download it from S3.
|
||
|
|
|
||
|
|
:param host: service host
|
||
|
|
:param user: service user
|
||
|
|
:param password: service pwd
|
||
|
|
:param database: database to back up
|
||
|
|
:param port: database service port
|
||
|
|
:param intput: local path of file to restore the backup
|
||
|
|
:param options: list of other connection options
|
||
|
|
:param arguments: list of connection arguments
|
||
|
|
:param schema: Run the process against Postgres with the given schema
|
||
|
|
"""
|
||
|
|
click.secho(
|
||
|
|
f"Restoring OpenMetadata backup for {host}:{port}/{database}...",
|
||
|
|
fg="bright_green",
|
||
|
|
)
|
||
|
|
|
||
|
|
engine = get_engine(
|
||
|
|
host, port, user, password, options, arguments, schema, database
|
||
|
|
)
|
||
|
|
|
||
|
|
execute_sql_file(engine=engine, input=input, schema=schema)
|
||
|
|
|
||
|
|
click.secho(
|
||
|
|
f"Backup restored from {input}",
|
||
|
|
fg="bright_green",
|
||
|
|
)
|