scrapyd: populate SCRAPY_SLOT environment variable with the scrapyd slot number

This commit is contained in:
Pablo Hoffman 2010-12-24 12:47:59 -02:00
parent d9a3df45c6
commit 1c8d74eb5b
2 changed files with 3 additions and 0 deletions

View File

@ -21,6 +21,7 @@ class Environment(object):
def get_environment(self, message, slot, eggpath):
project = message['_project']
env = self.initenv.copy()
env['SCRAPY_SLOT'] = str(slot)
env['SCRAPY_PROJECT'] = project
if eggpath:
env['SCRAPY_EGGFILE'] = eggpath

View File

@ -26,6 +26,7 @@ class EggStorageTest(unittest.TestCase):
slot = 3
env = self.environ.get_environment(msg, slot, '/path/to/file.egg')
self.assertEqual(env['SCRAPY_PROJECT'], 'mybot')
self.assertEqual(env['SCRAPY_SLOT'], '3')
self.assert_(env['SCRAPY_SQLITE_DB'].endswith('mybot.db'))
self.assert_(env['SCRAPY_LOG_FILE'].endswith('/mybot/myspider/ID.log'))
self.assert_(env['SCRAPY_EGGFILE'].endswith('/path/to/file.egg'))
@ -36,6 +37,7 @@ class EggStorageTest(unittest.TestCase):
slot = 3
env = self.environ.get_environment(msg, slot, None)
self.assertEqual(env['SCRAPY_PROJECT'], 'newbot')
self.assertEqual(env['SCRAPY_SLOT'], '3')
self.assert_(env['SCRAPY_SQLITE_DB'].endswith('newbot.db'))
self.assert_(env['SCRAPY_LOG_FILE'].endswith('/newbot/myspider/ID.log'))
self.assertEqual(env['SCRAPY_SETTINGS_MODULE'], 'newbot.settings')