diff --git a/noblogsmv/main.py b/noblogsmv/main.py index 02a59aa06df832a006ef053f171a73f20557e14f..a0175707da6b75f8d1d28875d9daecec10ffbdaf 100644 --- a/noblogsmv/main.py +++ b/noblogsmv/main.py @@ -166,7 +166,10 @@ def process_rm_data(blog_id, value, progress): return 'rm_db' -class NoblogsMoveStateMachine(state.StateMachine): +class NoblogsStateMachine(state.StateMachine): + dry_run = None + +class NoblogsMoveStateMachine(NoblogsStateMachine): states = { 'error': state.nop, @@ -176,9 +179,8 @@ class NoblogsMoveStateMachine(state.StateMachine): 'done': state.nop, } - dry_run = False -class NoblogsCleanStateMachine(state.StateMachine): +class NoblogsCleanStateMachine(NoblogsStateMachine): states = { 'error': state.nop, @@ -234,7 +236,7 @@ use the `--recover' option. if opts.clean: statemachine = NoblogsCleanStateMachine else: - statemachine = NoblogsStateMachine + statemachine = NoblogsMoveStateMachine inputfd = None if not opts.recover: