python-storage/.librarian/generator-input/librarian.py at 0fd3e3e896f0d6ee45cf39d74c50ed413aa3641a · googleapis/python-storage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# Copyright 2020 Google LLC
#
# 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.

"""This script is used to synthesize generated parts of this library."""

import json
from pathlib import Path
import shutil

import synthtool as s
from synthtool import gcp
from synthtool.languages import python

# ----------------------------------------------------------------------------
# Copy the generated client from the owl-bot staging directory
# ----------------------------------------------------------------------------

# Load the default version defined in .repo-metadata.json.
default_version = json.load(open(".repo-metadata.json", "rt")).get("default_version")

for library in s.get_staging_dirs(default_version):
    s.move(
        [library],
        excludes=[
            ".coveragerc",
            "docs/**/*",
            "scripts/fixup*.py",
            "setup.py",
            "noxfile.py",
            "README.rst",
            # Exclude autogenerated default import `google.cloud.storage`
            "google/cloud/storage/*",
            # Exclude autogenerated constraints files for Python 3.7/3.9
            "testing/constraints-3.7.txt",
            "testing/constraints-3.9.txt",
            "tests/unit/__init__.py",
            # Temporarily exclude google/cloud/_storage_v2/services/storage/client.py
            # due to customization.
            # This exclusion will be removed in the google-cloud-python migration PR
            # When librarian.py is dropped.
            "google/cloud/storage_v2/services/storage/client.py",
        ],
    )

    source_path = Path("google/cloud/storage_v2")
    renamed_path = Path("google/cloud/_storage_v2")

    # Remove the old _storage_v2 directory if it exists
    if renamed_path.exists():
        shutil.rmtree(renamed_path)

    if source_path.exists():
        source_path.rename(renamed_path)

    if renamed_path.exists():
        s.replace(
        renamed_path / "**/*.py",
        "google.cloud.storage_v2",
        "google.cloud._storage_v2",
    )

    # Fix imports in the unit tests
    s.replace(
    "tests/unit/gapic/storage_v2/**/*.py",
    "google.cloud.storage_v2",
    "google.cloud._storage_v2",
    )


s.remove_staging_dirs()

common = gcp.CommonTemplates()

# ----------------------------------------------------------------------------
# Add templated files
# ----------------------------------------------------------------------------
templated_files = common.py_library(
    cov_level=99,
    split_system_tests=True,
    intersphinx_dependencies={
        # python-requests url temporary change related to
        # https://github.com/psf/requests/issues/6140#issuecomment-1135071992
        "requests": "https://requests.readthedocs.io/en/stable/"
    },
)

s.move(
    templated_files,
    excludes=[
        ".coveragerc",
        "docs/multiprocessing.rst",
        "noxfile.py",
        "CONTRIBUTING.rst",
        "README.rst",
        ".kokoro/**",
        ".github/**",
    ],
)

python.py_samples(skip_readmes=True)

s.shell.run(["nox", "-s", "blacken"], hide_output=False)