wip

Author: owais

opentelemetry-auto-instrumentation/src/opentelemetry/auto_instrumentation/bootstrap.py

@@ -22,31 +22,32 @@
 
 logger = getLogger(__file__)
 
-_ACTION_INSTALL = "install"
-_ACTION_REQUIREMENTS = "requirements"
-
 
 # target library to desired instrumentor path/versioned package name
 instrumentations = {
-    "dbapi": "opentelemetry-ext-dbapi>=0.6b0",
-    "flask": "opentelemetry-ext-flask>=0.6b0",
-    "grpc": "opentelemetry-ext-grpc>=0.6b0",
-    "requests": "opentelemetry-ext-requests>=0.6b0",
-    "mysql": "opentelemetry-ext-mysql>=0.6b0",
-    "psycopg2": "opentelemetry-ext-psycopg2>=0.6b0",
-    "pymongo": "opentelemetry-ext-pymongo>=0.6b0",
-    "pymysql": "opentelemetry-ext-pymysql",
-    "redis": "opentelemetry-ext-redis",
-    "sqlalchemy": "opentelemetry-ext-sqlalchemy",
-    "wsgi": "opentelemetry-ext-wsgi>=0.6b0",
+    "dbapi": "opentelemetry-ext-dbapi>=0.7b1",
+    "dbapi": "opentelemetry-ext-djando>=0.7b1",
+    "flask": "opentelemetry-ext-flask>=0.7b1",
+    "grpc": "opentelemetry-ext-grpc>=0.7b1",
+    "requests": "opentelemetry-ext-requests>=0.7b1",
+    "jinja2": "opentelemetry-ext-jinja2>=0.7b1",
+    "mysql": "opentelemetry-ext-mysql>=0.7b1",
+    "psycopg2": "opentelemetry-ext-psycopg2>=0.7b1",
+    "pymongo": "opentelemetry-ext-pymongo>=0.7b1",
+    "pymysql": "opentelemetry-ext-pymysql>=0.7b1",
+    "redis": "opentelemetry-ext-redis>=0.7b1",
+    "sqlalchemy": "opentelemetry-ext-sqlalchemy>=0.7b1",
+    "wsgi": "opentelemetry-ext-wsgi>=0.7b1",
 }
 
 # relevant instrumentors and tracers to uninstall and check for conflicts for target libraries
 libraries = {
     "dbapi": ("opentelemetry-ext-dbapi",),
+    "django": ("opentelemetry-ext-django",),
     "flask": ("opentelemetry-ext-flask",),
     "grpc": ("opentelemetry-ext-grpc",),
     "requests": ("opentelemetry-ext-requests",),
+    "jinja2": ("opentelemetry-ext-jinja2",),
     "mysql": ("opentelemetry-ext-mysql",),
     "psycopg2": ("opentelemetry-ext-psycopg2",),
     "pymongo": ("opentelemetry-ext-pymongo",),
@@ -150,6 +151,9 @@ def _run_install(packages):
 
 
 def run() -> None:
+    action_install = "install"
+    action_requirements = "requirements"
+
     parser = argparse.ArgumentParser(
         description="""
         opentelemetry-bootstrap detects installed libraries and automatically
@@ -159,8 +163,8 @@ def run() -> None:
     parser.add_argument(
         "-a",
         "--action",
-        choices=[_ACTION_INSTALL, _ACTION_REQUIREMENTS],
-        default=_ACTION_REQUIREMENTS,
+        choices=[action_install, action_requirements],
+        default=action_requirements,
         help="""
         install - uses pip to install the new requirements using to the
                   currently active site-package.
@@ -171,7 +175,7 @@ def run() -> None:
     args = parser.parse_args()
 
     cmd = {
-        _ACTION_INSTALL: _run_install,
-        _ACTION_REQUIREMENTS: _run_requirements,
+        action_install: _run_install,
+        action_requirements: _run_requirements,
     }[args.action]
     cmd(_find_installed_libraries())