opentelemetry_tortoiseorm_instrumentation_quickstart_with_sqlite.py

1from tortoise import Tortoise, run_async
2from opentelemetry.instrumentation.tortoiseorm import TortoiseORMInstrumentor
3
4async def init():
5    # Instrument Tortoise ORM
6    TortoiseORMInstrumentor().instrument()
7
8    # Initialize Tortoise ORM as usual
9    await Tortoise.init(
10        db_url='sqlite://:memory:',
11        modules={'models': ['__main__']}
12    )
13    await Tortoise.generate_schemas()
14
15async def run():
16    await init()
17    # Your Tortoise ORM operations here will now be instrumented
18    await Tortoise.close_connections()
19
20if __name__ == "__main__":
21    run_async(run())