Skip to content

Commit c90edf2

Browse files
authored
use end date for interval as well (#310)
1 parent bacda1a commit c90edf2

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

ingestr/src/tiktok_ads/__init__.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -112,23 +112,25 @@ def custom_reports(
112112
datetime=(
113113
dlt.sources.incremental(
114114
incremental_loading_param,
115-
start_date,
115+
initial_value=start_date,
116+
end_value=end_date,
116117
range_end="closed",
117118
range_start="closed",
118119
)
119120
if is_incremental
120121
else None
121122
),
122123
) -> Iterable[TDataItem]:
123-
current_date = start_date.in_tz(timezone)
124+
start_date_tz_adjusted = start_date.in_tz(timezone)
125+
end_date_tz_adjusted = end_date.in_tz(timezone)
124126

125127
if datetime is not None:
126-
datetime_str = datetime.last_value
127-
current_date = ensure_pendulum_datetime(datetime_str).in_tz(timezone)
128+
start_date_tz_adjusted = ensure_pendulum_datetime(datetime.last_value).in_tz(timezone)
129+
end_date_tz_adjusted = ensure_pendulum_datetime(datetime.end_value).in_tz(timezone)
128130

129131
list_of_interval = find_intervals(
130-
current_date=current_date,
131-
end_date=end_date,
132+
current_date=start_date_tz_adjusted,
133+
end_date=end_date_tz_adjusted,
132134
interval_days=interval_days,
133135
)
134136

0 commit comments

Comments
 (0)