kopia lustrzana https://github.com/simonw/datasette
153 wiersze
4.6 KiB
Python
153 wiersze
4.6 KiB
Python
from datasette import hookimpl
|
|
import click
|
|
import json
|
|
import os
|
|
from subprocess import check_call, check_output
|
|
|
|
from .common import (
|
|
add_common_publish_arguments_and_options,
|
|
fail_if_publish_binary_not_installed,
|
|
)
|
|
from ..utils import temporary_docker_directory
|
|
|
|
|
|
@hookimpl
|
|
def publish_subcommand(publish):
|
|
@publish.command()
|
|
@add_common_publish_arguments_and_options
|
|
@click.option(
|
|
"-n",
|
|
"--name",
|
|
default="datasette",
|
|
help="Application name to use when building",
|
|
)
|
|
@click.option(
|
|
"--service", default="", help="Cloud Run service to deploy (or over-write)"
|
|
)
|
|
@click.option("--spatialite", is_flag=True, help="Enable SpatialLite extension")
|
|
@click.option(
|
|
"--show-files",
|
|
is_flag=True,
|
|
help="Output the generated Dockerfile and metadata.json",
|
|
)
|
|
def cloudrun(
|
|
files,
|
|
metadata,
|
|
extra_options,
|
|
branch,
|
|
template_dir,
|
|
plugins_dir,
|
|
static,
|
|
install,
|
|
plugin_secret,
|
|
version_note,
|
|
title,
|
|
license,
|
|
license_url,
|
|
source,
|
|
source_url,
|
|
about,
|
|
about_url,
|
|
name,
|
|
service,
|
|
spatialite,
|
|
show_files,
|
|
):
|
|
fail_if_publish_binary_not_installed(
|
|
"gcloud", "Google Cloud", "https://cloud.google.com/sdk/"
|
|
)
|
|
project = check_output(
|
|
"gcloud config get-value project", shell=True, universal_newlines=True
|
|
).strip()
|
|
|
|
if not service:
|
|
# Show the user their current services, then prompt for one
|
|
click.echo("Please provide a service name for this deployment\n")
|
|
click.echo("Using an existing service name will over-write it")
|
|
click.echo("")
|
|
existing_services = get_existing_services()
|
|
if existing_services:
|
|
click.echo("Your existing services:\n")
|
|
for existing_service in existing_services:
|
|
click.echo(
|
|
" {name} - created {created} - {url}".format(
|
|
**existing_service
|
|
)
|
|
)
|
|
click.echo("")
|
|
service = click.prompt("Service name", type=str)
|
|
|
|
extra_metadata = {
|
|
"title": title,
|
|
"license": license,
|
|
"license_url": license_url,
|
|
"source": source,
|
|
"source_url": source_url,
|
|
"about": about,
|
|
"about_url": about_url,
|
|
}
|
|
|
|
environment_variables = {}
|
|
if plugin_secret:
|
|
extra_metadata["plugins"] = {}
|
|
for plugin_name, plugin_setting, setting_value in plugin_secret:
|
|
environment_variable = (
|
|
"{}_{}".format(plugin_name, plugin_setting)
|
|
.upper()
|
|
.replace("-", "_")
|
|
)
|
|
environment_variables[environment_variable] = setting_value
|
|
extra_metadata["plugins"].setdefault(plugin_name, {})[
|
|
plugin_setting
|
|
] = {"$env": environment_variable}
|
|
|
|
with temporary_docker_directory(
|
|
files,
|
|
name,
|
|
metadata,
|
|
extra_options,
|
|
branch,
|
|
template_dir,
|
|
plugins_dir,
|
|
static,
|
|
install,
|
|
spatialite,
|
|
version_note,
|
|
extra_metadata,
|
|
environment_variables,
|
|
):
|
|
if show_files:
|
|
if os.path.exists("metadata.json"):
|
|
print("=== metadata.json ===\n")
|
|
print(open("metadata.json").read())
|
|
print("\n==== Dockerfile ====\n")
|
|
print(open("Dockerfile").read())
|
|
print("\n====================\n")
|
|
|
|
image_id = "gcr.io/{project}/{name}".format(project=project, name=name)
|
|
check_call("gcloud builds submit --tag {}".format(image_id), shell=True)
|
|
check_call(
|
|
"gcloud beta run deploy --allow-unauthenticated --platform=managed --image {} {}".format(
|
|
image_id, service,
|
|
),
|
|
shell=True,
|
|
)
|
|
|
|
|
|
def get_existing_services():
|
|
services = json.loads(
|
|
check_output(
|
|
"gcloud beta run services list --platform=managed --format json",
|
|
shell=True,
|
|
universal_newlines=True,
|
|
)
|
|
)
|
|
return [
|
|
{
|
|
"name": service["metadata"]["name"],
|
|
"created": service["metadata"]["creationTimestamp"],
|
|
"url": service["status"]["address"]["url"],
|
|
}
|
|
for service in services
|
|
]
|