mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-07-29 05:04:05 +08:00
7cf3cb42cf
Now that SDK does not depend on opentelemetry-instrumentation anymore and opentelemetry-instrumentation has actual build time dependencies on the contrib repo, it makes maintanence a lot easier if we move opentelemetry-instrumentation to contrib repo. opentelemetry-distro depends on opentelemetry-instrumentation and is being moved as well. Neither of the two packages are really part of "core" Otel python anyway.
110 lines
2.9 KiB
Python
Executable File
110 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright The OpenTelemetry Authors
|
|
#
|
|
# 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.
|
|
|
|
import ast
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
import astor
|
|
import pkg_resources
|
|
from otel_packaging import (
|
|
get_instrumentation_packages,
|
|
root_path,
|
|
scripts_path,
|
|
)
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger("instrumentation_list_generator")
|
|
|
|
_auto_generation_msg = """
|
|
# DO NOT EDIT. THIS FILE WAS AUTOGENERATED FROM templates/{source}.
|
|
# RUN `python scripts/generate_setup.py` TO REGENERATE.
|
|
"""
|
|
|
|
_template = """
|
|
{header}
|
|
|
|
# DO NOT EDIT. THIS FILE WAS AUTOGENERATED FROM INSTRUMENTATION PACKAGES.
|
|
# RUN `python scripts/generate_instrumentation_bootstrap.py` TO REGENERATE.
|
|
|
|
{source}
|
|
"""
|
|
|
|
_source_tmpl = """
|
|
libraries = {}
|
|
default_instrumentations = []
|
|
"""
|
|
|
|
gen_path = os.path.join(
|
|
root_path,
|
|
"opentelemetry-instrumentation",
|
|
"src",
|
|
"opentelemetry",
|
|
"instrumentation",
|
|
"bootstrap_gen.py",
|
|
)
|
|
|
|
|
|
def main():
|
|
# pylint: disable=no-member
|
|
default_instrumentations = ast.List(elts=[])
|
|
libraries = ast.Dict(keys=[], values=[])
|
|
for pkg in get_instrumentation_packages():
|
|
if not pkg["instruments"]:
|
|
default_instrumentations.elts.append(ast.Str(pkg["requirement"]))
|
|
for target_pkg in pkg["instruments"]:
|
|
parsed = pkg_resources.Requirement.parse(target_pkg)
|
|
libraries.keys.append(ast.Str(parsed.name))
|
|
libraries.values.append(
|
|
ast.Dict(
|
|
keys=[ast.Str("library"), ast.Str("instrumentation")],
|
|
values=[ast.Str(target_pkg), ast.Str(pkg["requirement"])],
|
|
)
|
|
)
|
|
|
|
tree = ast.parse(_source_tmpl)
|
|
tree.body[0].value = libraries
|
|
tree.body[1].value = default_instrumentations
|
|
source = astor.to_source(tree)
|
|
|
|
with open(
|
|
os.path.join(scripts_path, "license_header.txt"), "r", encoding="utf-8"
|
|
) as header_file:
|
|
header = header_file.read()
|
|
source = _template.format(header=header, source=source)
|
|
|
|
with open(gen_path, "w", encoding="utf-8") as gen_file:
|
|
gen_file.write(source)
|
|
|
|
subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"scripts/eachdist.py",
|
|
"format",
|
|
"--path",
|
|
"opentelemetry-instrumentation/src",
|
|
],
|
|
check=True,
|
|
)
|
|
|
|
logger.info("generated %s", gen_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|