jpt-archive/oyster
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

--immediate

Browse Source
This commit is contained in:
James Turk 2012-05-15 16:58:50 -04:00
parent c4b7597772
commit f14fc1cd2b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
oyster/scripts/signal.py
View File

@ -14,6 +14,7 @@ def main():
parser.add_argument('doc_class', type=str, parser.add_argument('doc_class', type=str,
help='doc_class to apply function to') help='doc_class to apply function to')
parser.add_argument('--sample', action='store_true') parser.add_argument('--sample', action='store_true')
parser.add_argument('--immediate', action='store_true')
args = parser.parse_args() args = parser.parse_args()
@ -25,11 +26,13 @@ def main():
if args.sample: if args.sample:
print 'sampling 100 documents' print 'sampling 100 documents'
docs = docs.limit(100) docs = docs.limit(100)
args.immediate = True
if args.immediate:
module, name = args.task.rsplit('.', 1) module, name = args.task.rsplit('.', 1)
task = getattr(__import__(module, fromlist=[name]), name) task = getattr(__import__(module, fromlist=[name]), name)
for doc in docs: for doc in docs:
task.apply((doc['_id'],), throw=True) task.apply((doc['_id'],), throw=True)
else: else:
for doc in docs: for doc in docs:
send_task(args.task, (doc['_id'], )) send_task(args.task, (doc['_id'], ))