@@ 158,18 158,23 @@ def refresh(task):
@task(
domain='default',
inputs=(
- rio.number('cpus', required=True),
+ rio.string('name', required=False),
)
)
def migrate_diffs(task):
- from tshistory.migrate import migrate_add_diffstart_diffend
-
- cpus = task.input['cpus']
-
+ from tshistory.migrate import (
+ migrate_add_diffstart_diffend,
+ migrate_seriesdata_diffstart_diffend
+ )
+ namespaces = [
+ 'tsh', 'tsh.group', 'tsh-upstream',
+ 'tsh-formula-patch', 'tsh-cache'
+ ]
engine = task.engine
with task.capturelogs(std=True):
- migrate_add_diffstart_diffend(engine, 'tsh', False, True, cpus=cpus)
- migrate_add_diffstart_diffend(engine, 'tsh.group', False, True, cpus=cpus)
- migrate_add_diffstart_diffend(engine, 'tsh-upstream', False, True, cpus=cpus)
- migrate_add_diffstart_diffend(engine, 'tsh-formula-patch', False, True, cpus=cpus)
- migrate_add_diffstart_diffend(engine, 'tsh-cache', False, True, cpus=cpus)
+ inputs = task.input
+ for namespace in namespaces:
+ if 'name' in inputs:
+ migrate_seriesdata_diffstart_diffend(engine, namespace, inputs['name'])
+ else:
+ migrate_add_diffstart_diffend(engine, namespace, False, True, cpus=1)