Differential D3641 Diff 12855 swh/storage/cassandra/cql.py

Changeset View

Standalone View

swh/storage/cassandra/cql.py

Show First 20 Lines • Show All 750 Lines • ▼ Show 20 Lines	) -> ResultSet:
args.append(limit)		args.append(limit)
else:		else:
limit_name = "no_limit"		limit_name = "no_limit"

method_name = f"_origin_visit_get_{page_name}_{order.value}_{limit_name}"		method_name = f"_origin_visit_get_{page_name}_{order.value}_{limit_name}"
origin_visit_get_method = getattr(self, method_name)		origin_visit_get_method = getattr(self, method_name)
return origin_visit_get_method(*args)		return origin_visit_get_method(*args)

		@_prepared_statement(
		"SELECT * FROM origin_visit_status WHERE origin = ? "
		"AND visit = ? AND date >= ? "
		"ORDER BY date ASC "
		"LIMIT ?"
		)
		def _origin_visit_status_get_with_date_asc_limit(
		self,
		origin: str,
		visit: int,
		date_from: datetime.datetime,
		limit: int,
		*,
		statement,
		) -> ResultSet:
		return self._execute_with_retries(statement, [origin, visit, date_from, limit])

		@_prepared_statement(
		"SELECT * FROM origin_visit_status WHERE origin = ? "
		"AND visit = ? AND date <= ? "
		"ORDER BY visit DESC "
		"LIMIT ?"
		)
		def _origin_visit_status_get_with_date_desc_limit(
		self,
		origin: str,
		visit: int,
		date_from: datetime.datetime,
		limit: int,
		*,
		statement,
		) -> ResultSet:
		return self._execute_with_retries(statement, [origin, visit, date_from, limit])

		@_prepared_statement(
		"SELECT * FROM origin_visit_status WHERE origin = ? AND visit = ? "
		"ORDER BY visit ASC "
		"LIMIT ?"
		)
		def _origin_visit_status_get_with_no_date_asc_limit(
		self, origin: str, visit: int, limit: int, *, statement
		) -> ResultSet:
		return self._execute_with_retries(statement, [origin, visit, limit])

		@_prepared_statement(
		"SELECT * FROM origin_visit_status WHERE origin = ? AND visit = ? "
		"ORDER BY visit DESC "
		"LIMIT ?"
		)
		def _origin_visit_status_get_with_no_date_desc_limit(
		self, origin: str, visit: int, limit: int, *, statement
		) -> ResultSet:
		return self._execute_with_retries(statement, [origin, visit, limit])

		def origin_visit_status_get_range(
		self,
		origin: str,
		visit: int,
		date_from: Optional[datetime.datetime],
		limit: int,
		order: ListOrder,
		) -> ResultSet:
		args: List[Any] = [origin, visit]

		if date_from is not None:
		date_name = "date"
		vlorentzUnsubmitted Done Inline Actions why the name `page_name`? vlorentz: why the name `page_name`?
		ardumontAuthorUnsubmitted Done Inline Actions i don't know, i kept the naming logic from the function just before. i adapted the value from pagination to token as you seemed keen on that (from the test). What do you propose instead (nothing comes to mind for my part)? ardumont: i don't know, i kept the naming logic from the function just before. i adapted the value from…
		ardumontAuthorUnsubmitted Done Inline Actions It's related to `pagination` so `page` for short. ardumont: It's related to `pagination` so `page` for short.
		ardumontAuthorUnsubmitted Done Inline Actions I'll try something... date_name = "date" ... date_name = "no_date" ardumont: I'll try something... ``` date_name = "date" ... date_name = "no_date" ```
		args.append(date_from)
		else:
		date_name = "no_date"

		args.append(limit)

		method_name = f"_origin_visit_status_get_with_{date_name}_{order.value}_limit"
		origin_visit_status_get_method = getattr(self, method_name)
		return origin_visit_status_get_method(*args)

@_prepared_insert_statement("origin_visit", _origin_visit_keys)		@_prepared_insert_statement("origin_visit", _origin_visit_keys)
def origin_visit_add_one(self, visit: OriginVisit, *, statement) -> None:		def origin_visit_add_one(self, visit: OriginVisit, *, statement) -> None:
self._add_one(statement, "origin_visit", visit, self._origin_visit_keys)		self._add_one(statement, "origin_visit", visit, self._origin_visit_keys)

_origin_visit_status_keys = [		_origin_visit_status_keys = [
"origin",		"origin",
"visit",		"visit",
"date",		"date",
▲ Show 20 Lines • Show All 210 Lines • Show Last 20 Lines