|
59 | 59 | {% set datetime = modules.datetime %} |
60 | 60 | {% set pytz = modules.pytz %} |
61 | 61 |
|
62 | | - {# In dbt-fusion, pytz.localize() produces incorrect results (dbt-labs/dbt-fusion#143). |
63 | | - Use datetime.timezone.utc and replace(tzinfo=) instead. #} |
64 | 62 | {% if elementary.is_dbt_fusion() %} |
65 | | - {% set utc_tz = datetime.timezone.utc %} |
| 63 | + {# dbt-fusion's pytz and timezone-aware datetime operations have known issues |
| 64 | + (dbt-labs/dbt-fusion#143). Use naive UTC datetimes with manual offset |
| 65 | + calculation to avoid broken localize() and datetime comparison. #} |
| 66 | + {% set utc_tz = pytz.timezone("UTC") %} |
66 | 67 | {% set target_tz = pytz.timezone(timezone) %} |
67 | 68 |
|
68 | | - {% set now_utc = datetime.datetime.now(utc_tz) %} |
69 | | - {% set now_local = now_utc.astimezone(target_tz) %} |
70 | | - {% set target_date_local = now_local.date() %} |
71 | | - |
72 | | - {# Use replace(tzinfo=) instead of localize() for fusion compatibility #} |
73 | | - {% set day_start_naive = datetime.datetime.combine( |
74 | | - target_date_local, datetime.time(0, 0, 0) |
| 69 | + {# Get current UTC time as naive datetime - reliable across environments #} |
| 70 | + {% set now_utc = datetime.datetime.utcnow() %} |
| 71 | + |
| 72 | + {# Determine today's date and UTC offset in target timezone. |
| 73 | + Use localize+astimezone only to probe the offset, not for final values. #} |
| 74 | + {% set probe = utc_tz.localize(now_utc).astimezone(target_tz) %} |
| 75 | + {% set target_date_local = probe.date() %} |
| 76 | + {% set tz_offset = probe.utcoffset() %} |
| 77 | + {% set now_local = probe %} |
| 78 | + |
| 79 | + {# Build all datetimes as naive local, then convert to naive UTC |
| 80 | + by subtracting the timezone offset. This avoids tz-aware comparison. #} |
| 81 | + {% set day_start_utc = ( |
| 82 | + datetime.datetime.combine( |
| 83 | + target_date_local, datetime.time(0, 0, 0) |
| 84 | + ) |
| 85 | + - tz_offset |
75 | 86 | ) %} |
76 | | - {% set day_start_local = day_start_naive.replace(tzinfo=target_tz) %} |
77 | | - {% set day_start_utc = day_start_local.astimezone(utc_tz) %} |
78 | 87 |
|
79 | | - {% set day_end_naive = datetime.datetime.combine( |
80 | | - target_date_local, datetime.time(23, 59, 59) |
| 88 | + {% set day_end_utc = ( |
| 89 | + datetime.datetime.combine( |
| 90 | + target_date_local, datetime.time(23, 59, 59) |
| 91 | + ) |
| 92 | + - tz_offset |
81 | 93 | ) %} |
82 | | - {% set day_end_local = day_end_naive.replace(tzinfo=target_tz) %} |
83 | | - {% set day_end_utc = day_end_local.astimezone(utc_tz) %} |
84 | 94 |
|
85 | | - {% set sla_deadline_naive = datetime.datetime.combine( |
86 | | - target_date_local, datetime.time(sla_hour, sla_minute, 0) |
| 95 | + {% set sla_deadline_utc = ( |
| 96 | + datetime.datetime.combine( |
| 97 | + target_date_local, datetime.time(sla_hour, sla_minute, 0) |
| 98 | + ) |
| 99 | + - tz_offset |
87 | 100 | ) %} |
88 | | - {% set sla_deadline_local = sla_deadline_naive.replace(tzinfo=target_tz) %} |
89 | | - {% set sla_deadline_utc = sla_deadline_local.astimezone(utc_tz) %} |
| 101 | + |
| 102 | + {# Compare naive UTC datetimes #} |
| 103 | + {% set deadline_passed = now_utc > sla_deadline_utc %} |
90 | 104 | {% else %} |
91 | 105 | {# Standard dbt-core path using pytz.localize() #} |
92 | 106 | {% set utc_tz = pytz.timezone("UTC") %} |
|
115 | 129 | sla_deadline_naive, is_dst=False |
116 | 130 | ) %} |
117 | 131 | {% set sla_deadline_utc = sla_deadline_local.astimezone(utc_tz) %} |
118 | | - {% endif %} |
119 | 132 |
|
120 | | - {# Check if deadline has passed #} |
121 | | - {% set deadline_passed = now_utc > sla_deadline_utc %} |
| 133 | + {% set deadline_passed = now_utc > sla_deadline_utc %} |
| 134 | + {% endif %} |
122 | 135 |
|
123 | 136 | {# Format for SQL #} |
124 | 137 | {% set sla_deadline_utc_str = sla_deadline_utc.strftime("%Y-%m-%d %H:%M:%S") %} |
|
0 commit comments