Skip to content

Commit f7f9868

Browse files
fix: apply ruff formatting
Co-Authored-By: Aaron <AJ> Steers <[email protected]>
1 parent 1b2a724 commit f7f9868

File tree

6 files changed

+34
-14
lines changed

6 files changed

+34
-14
lines changed

airbyte_cdk/sources/declarative/datetime/datetime_parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def parse(self, date: Union[str, int], format: str) -> datetime.datetime:
5151
elif "%_ms" in format:
5252
# Convert custom millisecond format to standard format
5353
format = format.replace("%_ms", "%f")
54-
54+
5555
# For standard formats, use ab_datetime_parse with the specific format
5656
try:
5757
result = ab_datetime_parse(date, formats=[format], disallow_other_formats=True)

airbyte_cdk/sources/declarative/datetime/min_max_datetime.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,18 +72,22 @@ def get_datetime(
7272
**additional_parameters,
7373
)
7474
),
75-
formats=[datetime_format] if datetime_format else None
75+
formats=[datetime_format] if datetime_format else None,
7676
).to_datetime() # type: ignore # datetime is always cast to an interpolated string
7777

7878
if self.min_datetime:
7979
min_time = str(self.min_datetime.eval(config, **additional_parameters)) # type: ignore # min_datetime is always cast to an interpolated string
8080
if min_time:
81-
min_datetime = ab_datetime_parse(min_time, formats=[datetime_format] if datetime_format else None).to_datetime() # type: ignore # min_datetime is always cast to an interpolated string
81+
min_datetime = ab_datetime_parse(
82+
min_time, formats=[datetime_format] if datetime_format else None
83+
).to_datetime() # type: ignore # min_datetime is always cast to an interpolated string
8284
time = max(time, min_datetime)
8385
if self.max_datetime:
8486
max_time = str(self.max_datetime.eval(config, **additional_parameters)) # type: ignore # max_datetime is always cast to an interpolated string
8587
if max_time:
86-
max_datetime = ab_datetime_parse(max_time, formats=[datetime_format] if datetime_format else None).to_datetime()
88+
max_datetime = ab_datetime_parse(
89+
max_time, formats=[datetime_format] if datetime_format else None
90+
).to_datetime()
8791
time = min(time, max_datetime)
8892
return time
8993

airbyte_cdk/sources/declarative/incremental/datetime_based_cursor.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,9 +310,13 @@ def _get_date(
310310
def parse_date(self, date: str) -> datetime.datetime:
311311
try:
312312
# Try each format in the list, falling back to the default format
313-
return ab_datetime_parse(date, formats=self.cursor_datetime_formats + [self.datetime_format]).to_datetime()
313+
return ab_datetime_parse(
314+
date, formats=self.cursor_datetime_formats + [self.datetime_format]
315+
).to_datetime()
314316
except ValueError:
315-
raise ValueError(f"No format in {self.cursor_datetime_formats + [self.datetime_format]} matching {date}")
317+
raise ValueError(
318+
f"No format in {self.cursor_datetime_formats + [self.datetime_format]} matching {date}"
319+
)
316320

317321
@classmethod
318322
def _parse_timedelta(cls, time_str: Optional[str]) -> Union[datetime.timedelta, Duration]:

airbyte_cdk/sources/declarative/interpolation/macros.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,7 @@ def day_delta(num_days: int, format: str = "%Y-%m-%dT%H:%M:%S.%f%z") -> str:
144144
:return: datetime formatted as RFC3339
145145
"""
146146
return ab_datetime_format(
147-
ab_datetime_now().to_datetime() + datetime.timedelta(days=num_days),
148-
format
147+
ab_datetime_now().to_datetime() + datetime.timedelta(days=num_days), format
149148
)
150149

151150

@@ -176,10 +175,14 @@ def format_datetime(
176175
return ab_datetime_format(dt, format)
177176

178177
if isinstance(dt, int):
179-
dt_datetime = ab_datetime_parse(dt, formats=[input_format] if input_format else None).to_datetime()
178+
dt_datetime = ab_datetime_parse(
179+
dt, formats=[input_format] if input_format else None
180+
).to_datetime()
180181
else:
181182
dt_datetime = (
182-
ab_datetime_parse(dt, formats=[input_format] if input_format else None).to_datetime() if input_format else str_to_datetime(dt)
183+
ab_datetime_parse(dt, formats=[input_format] if input_format else None).to_datetime()
184+
if input_format
185+
else str_to_datetime(dt)
183186
)
184187
if dt_datetime.tzinfo is None:
185188
dt_datetime = dt_datetime.replace(tzinfo=pytz.utc)

airbyte_cdk/sources/file_based/stream/concurrent/cursor/file_based_concurrent_cursor.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,9 @@ def _compute_prev_sync_cursor(self, value: Optional[StreamState]) -> Tuple[datet
111111
)
112112
cursor_str = min(prev_cursor_str, earliest_file_cursor_value)
113113
cursor_dt, cursor_uri = cursor_str.split("_", 1)
114-
return ab_datetime_parse(cursor_dt, formats=[self.DATE_TIME_FORMAT]).to_datetime(), cursor_uri
114+
return ab_datetime_parse(
115+
cursor_dt, formats=[self.DATE_TIME_FORMAT]
116+
).to_datetime(), cursor_uri
115117

116118
def _get_cursor_key_from_file(self, file: Optional[RemoteFile]) -> str:
117119
if file:
@@ -126,7 +128,9 @@ def _compute_earliest_file_in_history(self) -> Optional[RemoteFile]:
126128
)
127129
return RemoteFile(
128130
uri=filename,
129-
last_modified=ab_datetime_parse(last_modified, formats=[self.DATE_TIME_FORMAT]).to_datetime(),
131+
last_modified=ab_datetime_parse(
132+
last_modified, formats=[self.DATE_TIME_FORMAT]
133+
).to_datetime(),
130134
)
131135
else:
132136
return None
@@ -211,7 +215,9 @@ def _compute_latest_file_in_history(self) -> Optional[RemoteFile]:
211215
)
212216
return RemoteFile(
213217
uri=filename,
214-
last_modified=ab_datetime_parse(last_modified, formats=[self.DATE_TIME_FORMAT]).to_datetime(),
218+
last_modified=ab_datetime_parse(
219+
last_modified, formats=[self.DATE_TIME_FORMAT]
220+
).to_datetime(),
215221
)
216222
else:
217223
return None

airbyte_cdk/sources/file_based/stream/cursor/default_file_based_cursor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,10 @@ def _compute_earliest_file_in_history(self) -> Optional[RemoteFile]:
133133
self._file_to_datetime_history.items(), key=lambda f: (f[1], f[0])
134134
)
135135
return RemoteFile(
136-
uri=filename, last_modified=ab_datetime_parse(last_modified, formats=[self.DATE_TIME_FORMAT]).to_datetime()
136+
uri=filename,
137+
last_modified=ab_datetime_parse(
138+
last_modified, formats=[self.DATE_TIME_FORMAT]
139+
).to_datetime(),
137140
)
138141
else:
139142
return None

0 commit comments

Comments
 (0)