schedule_csv_partition.py
No OneTemporary
Actions

Size

1 KB

Subscribers

None

schedule_csv_partition.py
View Options

	#!/usr/bin/env python3

	# Copyright (C) 2018-2020 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information


	import click
	import random


	def _is_power_of_two(n: int) -> bool:
	return n > 0 and n & (n - 1) == 0


	@click.command()
	@click.option('--task-type', '-t', default='index-mimetype-partition',
	help='Add task type')
	@click.option('--policy', '-p', default='oneshot',
	help='Task policy, either oneshot or recurring')
	@click.option('--number-partitions', '-n', default=1024,
	help='(A power of 2) number of partitions to compute')
	def main(task_type, policy, number_partitions):
	"""Compute `number-partitions` for a given `task-type` with a given `policy`.

	Use sample:
	$ python3 -m schedule_csv_partition \
	--task-type index-mimetype-partition \
	--policy recurring \
	--number-ranges 1024 \| head -2
	index-mimetype-partition;oneshot;[122, 1024]
	index-mimetype-partition;oneshot;[720, 1024]

	Schedule the tasks (providing file.csv contains the result of the previous
	command):

	$ head -1 file.csv \| swh scheduler task schedule \
	-c type -c policy -c args --delimiter ';' -

	"""
	if not _is_power_of_two(number_partitions):
	# expected by the storage
	raise ValueError("Number of partitions must be a power of 2")

	partition_ids = [partition_id for partition_id in range(number_partitions)]
	random.shuffle(partition_ids)

	assert len(partition_ids) == number_partitions

	for partition_id in partition_ids:
	print(
	f"{task_type};{policy};[{partition_id}, {number_partitions}]"
	)


	if __name__ == '__main__':
	main()

File Metadata

Mime Type: text/x-python
Expires: Thu, Jul 3, 10:09 AM (2 w, 5 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3239585

schedule_csv_partition.pyNo OneTemporaryActions

schedule_csv_partition.pyView Options

File Metadata

Event Timeline

schedule_csv_partition.py
No OneTemporary
Actions

schedule_csv_partition.py
View Options