def __attrs_post_init__(self):
"""Post-init hook.
Responsible for setting up the application upon instantiation of the class.
Returns:
None
"""
# inject settings
self.client.extensions = self.extensions
fields_ext = self.get_extension(FieldsExtension)
if fields_ext:
self.settings.default_includes = fields_ext.default_includes
Settings.set(self.settings)
self.register_core()
# register extensions
for ext in self.extensions:
ext.register(self.app)
# add health check
self.add_health_check()
# register exception handlers
add_exception_handlers(self.app, status_codes=self.exceptions)
# customize openapi
self.app.openapi = self.customize_openapi
def __attrs_post_init__(self):
"""Post-init hook.
Responsible for setting up the application upon instantiation of the class.
Returns:
None
"""
# inject settings
self.client.extensions = self.extensions
self.client.stac_version = self.stac_version
self.client.title = self.title
self.client.description = self.description
fields_ext = self.get_extension(FieldsExtension)
if fields_ext:
self.settings.default_includes = fields_ext.default_includes
Settings.set(self.settings)
self.app.state.settings = self.settings
# Register core STAC endpoints
self.register_core()
self.app.include_router(self.router)
# register extensions
for ext in self.extensions:
ext.register(self.app)
# add health check
self.add_health_check()
# register exception handlers
add_exception_handlers(self.app, status_codes=self.exceptions)
# customize openapi
self.app.openapi = self.customize_openapi
# add middlewares
for middleware in self.middlewares:
self.app.add_middleware(middleware)
BulkTransactionsClient,
TransactionsClient,
)
from stac_fastapi.sqlalchemy.types.search import SQLAlchemySTACSearch
from stac_fastapi.types.config import Settings
DATA_DIR = os.path.join(os.path.dirname(__file__), "data")
class TestSettings(SqlalchemySettings):
class Config:
env_file = ".env.test"
settings = TestSettings()
Settings.set(settings)
@pytest.fixture(autouse=True)
def cleanup(postgres_core: CoreCrudClient,
postgres_transactions: TransactionsClient):
yield
collections = postgres_core.all_collections(request=MockStarletteRequest)
for coll in collections:
if coll.id.split("-")[0] == "test":
# Delete the items
items = postgres_core.item_collection(coll.id,
limit=100,
request=MockStarletteRequest)
for feat in items.features:
postgres_transactions.delete_item(feat.id,