|
8 | 8 | import re |
9 | 9 | from datetime import date |
10 | 10 | from datetime import datetime as datetime_type |
| 11 | +from datetime import timezone |
11 | 12 | from typing import Dict, Optional, Union |
12 | 13 |
|
| 14 | +from stac_fastapi.core.utilities import get_bool_env |
13 | 15 | from stac_fastapi.types.rfc3339 import DateTimeType |
14 | 16 |
|
15 | 17 | logger = logging.getLogger(__name__) |
@@ -37,38 +39,103 @@ def return_date( |
37 | 39 | always containing 'gte' and 'lte' keys. |
38 | 40 | """ |
39 | 41 | result: Dict[str, Optional[str]] = {"gte": None, "lte": None} |
40 | | - |
| 42 | + use_datetime_nanos = get_bool_env("USE_DATETIME_NANOS", default=True) |
41 | 43 | if interval is None: |
42 | 44 | return result |
43 | 45 |
|
44 | | - if isinstance(interval, str): |
45 | | - if "/" in interval: |
46 | | - parts = interval.split("/") |
47 | | - result["gte"] = ( |
48 | | - parts[0] if parts[0] != ".." else datetime_type.min.isoformat() + "Z" |
49 | | - ) |
50 | | - result["lte"] = ( |
51 | | - parts[1] |
52 | | - if len(parts) > 1 and parts[1] != ".." |
53 | | - else datetime_type.max.isoformat() + "Z" |
| 46 | + if use_datetime_nanos: |
| 47 | + MIN_DATE_NANOS = datetime_type(1970, 1, 1, tzinfo=timezone.utc) |
| 48 | + MAX_DATE_NANOS = datetime_type( |
| 49 | + 2262, 4, 11, 23, 47, 16, 854775, tzinfo=timezone.utc |
| 50 | + ) |
| 51 | + |
| 52 | + if isinstance(interval, str): |
| 53 | + if "/" in interval: |
| 54 | + parts = interval.split("/") |
| 55 | + result["gte"] = ( |
| 56 | + parts[0] if parts[0] != ".." else MIN_DATE_NANOS.isoformat() + "Z" |
| 57 | + ) |
| 58 | + result["lte"] = ( |
| 59 | + parts[1] |
| 60 | + if len(parts) > 1 and parts[1] != ".." |
| 61 | + else MAX_DATE_NANOS.isoformat() + "Z" |
| 62 | + ) |
| 63 | + else: |
| 64 | + converted_time = interval if interval != ".." else None |
| 65 | + result["gte"] = result["lte"] = converted_time |
| 66 | + return result |
| 67 | + |
| 68 | + if isinstance(interval, datetime_type): |
| 69 | + dt_utc = ( |
| 70 | + interval.astimezone(timezone.utc) |
| 71 | + if interval.tzinfo |
| 72 | + else interval.replace(tzinfo=timezone.utc) |
54 | 73 | ) |
55 | | - else: |
56 | | - converted_time = interval if interval != ".." else None |
57 | | - result["gte"] = result["lte"] = converted_time |
| 74 | + if dt_utc < MIN_DATE_NANOS: |
| 75 | + dt_utc = MIN_DATE_NANOS |
| 76 | + elif dt_utc > MAX_DATE_NANOS: |
| 77 | + dt_utc = MAX_DATE_NANOS |
| 78 | + datetime_iso = dt_utc.isoformat() |
| 79 | + result["gte"] = result["lte"] = datetime_iso |
| 80 | + elif isinstance(interval, tuple): |
| 81 | + start, end = interval |
| 82 | + # Ensure datetimes are converted to UTC and formatted with 'Z' |
| 83 | + if start: |
| 84 | + start_utc = ( |
| 85 | + start.astimezone(timezone.utc) |
| 86 | + if start.tzinfo |
| 87 | + else start.replace(tzinfo=timezone.utc) |
| 88 | + ) |
| 89 | + if start_utc < MIN_DATE_NANOS: |
| 90 | + start_utc = MIN_DATE_NANOS |
| 91 | + elif start_utc > MAX_DATE_NANOS: |
| 92 | + start_utc = MAX_DATE_NANOS |
| 93 | + result["gte"] = start_utc.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
| 94 | + if end: |
| 95 | + end_utc = ( |
| 96 | + end.astimezone(timezone.utc) |
| 97 | + if end.tzinfo |
| 98 | + else end.replace(tzinfo=timezone.utc) |
| 99 | + ) |
| 100 | + if end_utc < MIN_DATE_NANOS: |
| 101 | + end_utc = MIN_DATE_NANOS |
| 102 | + elif end_utc > MAX_DATE_NANOS: |
| 103 | + end_utc = MAX_DATE_NANOS |
| 104 | + result["lte"] = end_utc.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
| 105 | + |
58 | 106 | return result |
59 | 107 |
|
60 | | - if isinstance(interval, datetime_type): |
61 | | - datetime_iso = interval.isoformat() |
62 | | - result["gte"] = result["lte"] = datetime_iso |
63 | | - elif isinstance(interval, tuple): |
64 | | - start, end = interval |
65 | | - # Ensure datetimes are converted to UTC and formatted with 'Z' |
66 | | - if start: |
67 | | - result["gte"] = start.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
68 | | - if end: |
69 | | - result["lte"] = end.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
70 | | - |
71 | | - return result |
| 108 | + else: |
| 109 | + if isinstance(interval, str): |
| 110 | + if "/" in interval: |
| 111 | + parts = interval.split("/") |
| 112 | + result["gte"] = ( |
| 113 | + parts[0] |
| 114 | + if parts[0] != ".." |
| 115 | + else datetime_type.min.isoformat() + "Z" |
| 116 | + ) |
| 117 | + result["lte"] = ( |
| 118 | + parts[1] |
| 119 | + if len(parts) > 1 and parts[1] != ".." |
| 120 | + else datetime_type.max.isoformat() + "Z" |
| 121 | + ) |
| 122 | + else: |
| 123 | + converted_time = interval if interval != ".." else None |
| 124 | + result["gte"] = result["lte"] = converted_time |
| 125 | + return result |
| 126 | + |
| 127 | + if isinstance(interval, datetime_type): |
| 128 | + datetime_iso = interval.isoformat() |
| 129 | + result["gte"] = result["lte"] = datetime_iso |
| 130 | + elif isinstance(interval, tuple): |
| 131 | + start, end = interval |
| 132 | + # Ensure datetimes are converted to UTC and formatted with 'Z' |
| 133 | + if start: |
| 134 | + result["gte"] = start.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
| 135 | + if end: |
| 136 | + result["lte"] = end.strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z" |
| 137 | + |
| 138 | + return result |
72 | 139 |
|
73 | 140 |
|
74 | 141 | def extract_date(date_str: str) -> date: |
|
0 commit comments