|
1 |
| -"""FastAPI application using PGStac.""" |
| 1 | +"""FastAPI application using PGStac. |
| 2 | +
|
| 3 | +Enables the extensions specified as a comma-delimited list in |
| 4 | +the ENABLED_EXTENSIONS environment variable (e.g. `transactions,sort,query`). |
| 5 | +If the variable is not set, enables all extensions. |
| 6 | +""" |
| 7 | + |
| 8 | +import os |
| 9 | + |
2 | 10 | from fastapi.responses import ORJSONResponse
|
3 | 11 |
|
4 | 12 | from stac_fastapi.api.app import StacApi
|
|
21 | 29 | from stac_fastapi.pgstac.types.search import PgstacSearch
|
22 | 30 |
|
23 | 31 | settings = Settings()
|
24 |
| -extensions = [ |
25 |
| - TransactionExtension( |
| 32 | +extensions_map = { |
| 33 | + "transaction": TransactionExtension( |
26 | 34 | client=TransactionsClient(),
|
27 | 35 | settings=settings,
|
28 | 36 | response_class=ORJSONResponse,
|
29 | 37 | ),
|
30 |
| - QueryExtension(), |
31 |
| - SortExtension(), |
32 |
| - FieldsExtension(), |
33 |
| - TokenPaginationExtension(), |
34 |
| - ContextExtension(), |
35 |
| - FilterExtension(client=FiltersClient()), |
36 |
| - BulkTransactionExtension(client=BulkTransactionsClient()), |
37 |
| -] |
| 38 | + "query": QueryExtension(), |
| 39 | + "sort": SortExtension(), |
| 40 | + "fields": FieldsExtension(), |
| 41 | + "pagination": TokenPaginationExtension(), |
| 42 | + "context": ContextExtension(), |
| 43 | + "filter": FilterExtension(client=FiltersClient()), |
| 44 | + "bulk_transactions": BulkTransactionExtension(client=BulkTransactionsClient()), |
| 45 | +} |
| 46 | + |
| 47 | +if enabled_extensions := os.getenv("ENABLED_EXTENSIONS"): |
| 48 | + extensions = [ |
| 49 | + extensions_map[extension_name] |
| 50 | + for extension_name in enabled_extensions.split(",") |
| 51 | + ] |
| 52 | +else: |
| 53 | + extensions = list(extensions_map.values()) |
38 | 54 |
|
39 | 55 | post_request_model = create_post_request_model(extensions, base_model=PgstacSearch)
|
40 | 56 |
|
|
0 commit comments