Fetching the queue name and parameters from within the script without…

… external dependencies
ruecarlo · Nov 21, 2017 · cde64ac · cde64ac
1 parent 0007261
commit cde64ac
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/src/queue_processor.py b/src/queue_processor.py
@@ -1,10 +1,15 @@
 import boto3
 import time
 import subprocess
-import config
+import argparse
+
+session = boto3.session.Session()
+REGION = session.region_name
+parser = argparse.ArgumentParser(description='Process some integers.')
+parser.add_argument('--queue', dest='queue',default='workshop', help='Amazon SQS')
+args = parser.parse_args()
+QUEUE = args.queue
 
-REGION = str(config.REGION)
-QUEUE = str(config.QUEUE)
 
 sleepTime = 5
 

diff --git a/templates/spotlabworker.sh b/templates/spotlabworker.sh
@@ -19,10 +19,10 @@
 	wget $WEBURL/static/queue_processor.py
 	wget $WEBURL/static/worker.py
 
-	echo "Configuring worker inputs"
-    echo "QUEUE = 'Workshop'" > config.py
-	echo "REGION = '$REGION'" >> config.py
-	echo "Debug: "$(config.py)
+#	echo "Configuring worker inputs"
+#    echo "QUEUE = 'Workshop'" > config.py
+#	echo "REGION = '$REGION'" >> config.py
+#	echo "Debug: "$(config.py)
 
 	echo 'Starting the worker processor'
 	python /home/ec2-user/spotlabworker/queue_processor.py > stdout.txt 2>&1