4
4
import logging
5
5
import sys
6
6
import django
7
- from django .db import transaction
8
7
django .setup ()
9
8
10
9
from osf .models import AbstractNode
11
10
from scripts import utils as script_utils
12
11
from website import settings
13
- from website .app import init_app
12
+ from website .app import setup_django
14
13
from website .project .tasks import on_registration_updated
15
14
16
15
17
16
logger = logging .getLogger (__name__ )
18
17
19
18
def migrate (registrations ):
20
- print registrations
21
19
assert settings .SHARE_URL , 'SHARE_URL must be set to migrate.'
22
20
assert settings .SHARE_API_TOKEN , 'SHARE_API_TOKEN must be set to migrate.'
23
21
registrations_count = len (registrations )
@@ -43,13 +41,12 @@ def main():
43
41
'--targets' ,
44
42
action = 'store' ,
45
43
dest = 'targets' ,
46
- help = 'List of targets, of form {"data": ["registration_id", ...]} ' ,
44
+ help = 'List of targets, of form ["registration_id", ...]' ,
47
45
)
48
46
pargs = parser .parse_args ()
49
47
script_utils .add_file_logger (logger , __file__ )
50
- init_app (set_backends = True , routes = False )
51
- with transaction .atomic ():
52
- migrate (json .loads (pargs .targets )['data' ])
48
+ setup_django ()
49
+ migrate (json .loads (pargs .targets ))
53
50
54
51
if __name__ == "__main__" :
55
52
main ()
0 commit comments