tasks: migration 0.20 from tshistory is possible for one seriesname
1 files changed, 15 insertions(+), 10 deletions(-)

M tshistory_refinery/tasks.py
M tshistory_refinery/tasks.py +15 -10
@@ 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)