Skip to content

Commit

Permalink
implement collab_add/remove/primary
Browse files Browse the repository at this point in the history
  • Loading branch information
tschaume committed May 16, 2015
1 parent 7f169e7 commit 1375b78
Showing 1 changed file with 18 additions and 3 deletions.
21 changes: 18 additions & 3 deletions scripts/mgc
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,28 @@ def delete(args):
)

def collab_add(args):
pass
with MPRester(API_KEY, endpoint=ENDPOINT) as m:
cids = [] if args.to is None else args.to
if args.to is None:
for contrib in m.query_contribs():
cids.append(contrib['contribution_id'])
print m.update_collaborators(args.collaborators, cids, 'add')

def collab_remove(args):
pass
with MPRester(API_KEY, endpoint=ENDPOINT) as m:
cids = [] if args.from is None else args.from
if args.from is None:
for contrib in m.query_contribs():
cids.append(contrib['contribution_id'])
print m.update_collaborators(args.collaborators, cids, 'remove')

def collab_primary(args):
pass
with MPRester(API_KEY, endpoint=ENDPOINT) as m:
cids = [] if args.for is None else args.for
if args.for is None:
for contrib in m.query_contribs():
cids.append(contrib['contribution_id'])
print m.update_collaborators([args.collaborator], cids, 'primary')

def main():
parser = argparse.ArgumentParser(
Expand Down

0 comments on commit 1375b78

Please sign in to comment.