mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2024-11-28 12:33:59 +03:00
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""Run 'Bundles' example application."""
|
|
|
|
import sqlite3
|
|
import boto3
|
|
|
|
from dependency_injector import containers
|
|
from dependency_injector import providers
|
|
|
|
from bundles.users import Users
|
|
from bundles.photos import Photos
|
|
|
|
|
|
class Core(containers.DeclarativeContainer):
|
|
"""Core container."""
|
|
|
|
config = providers.Configuration('config')
|
|
sqlite = providers.Singleton(sqlite3.connect, config.database.dsn)
|
|
s3 = providers.Singleton(
|
|
boto3.client, 's3',
|
|
aws_access_key_id=config.aws.access_key_id,
|
|
aws_secret_access_key=config.aws.secret_access_key)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Initializing containers
|
|
core = Core()
|
|
core.config.update({'database': {'dsn': ':memory:'},
|
|
'aws': {'access_key_id': 'KEY',
|
|
'secret_access_key': 'SECRET'}})
|
|
users = Users(database=core.sqlite)
|
|
photos = Photos(database=core.sqlite, file_storage=core.s3)
|
|
|
|
# Fetching few users
|
|
user_repository = users.user_repository()
|
|
user1 = user_repository.get(id=1)
|
|
user2 = user_repository.get(id=2)
|
|
|
|
# Making some checks
|
|
assert user1.id == 1
|
|
assert user2.id == 2
|
|
assert user_repository.db is core.sqlite()
|