Skip to content

Commit c610c93

Browse files
author
Sam Ruby
committed
Load configuration only once
1 parent 5d7bd64 commit c610c93

File tree

7 files changed

+18
-15
lines changed

7 files changed

+18
-15
lines changed

planet.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,14 +41,17 @@
4141
else:
4242
config_file = arg
4343

44+
from planet import config
45+
config.load(config_file)
46+
4447
if verbose:
4548
import planet
4649
planet.getLogger('DEBUG')
4750

4851
if not offline:
4952
from planet import spider
50-
spider.spiderPlanet(config_file)
53+
spider.spiderPlanet()
5154

5255
from planet import splice
53-
doc = splice.splice(config_file)
56+
doc = splice.splice()
5457
splice.apply(doc.toxml('utf-8'))

planet/spider.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,8 @@ def spiderFeed(feed):
180180
write(output, cache_file)
181181
os.utime(cache_file, (mtime, mtime))
182182

183-
def spiderPlanet(configFile):
183+
def spiderPlanet():
184184
""" Spider (fetch) an entire planet """
185-
config.load(configFile)
186185
log = planet.getLogger(config.log_level())
187186
planet.setTimeout(config.feed_timeout())
188187

planet/splice.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,9 @@
55
from reconstitute import createTextElement, date
66
from spider import filename
77

8-
def splice(configFile):
8+
def splice():
99
""" Splice together a planet from a cache of entries """
1010
import planet
11-
config.load(configFile)
1211
log = planet.getLogger(config.log_level())
1312

1413
log.info("Loading cached data")

spider.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,13 @@
88

99
if __name__ == '__main__':
1010

11+
config.load(sys.argv[1])
12+
1113
if len(sys.argv) == 2:
1214
# spider all feeds
13-
spider.spiderPlanet(sys.argv[1])
15+
spider.spiderPlanet()
1416
elif len(sys.argv) > 2:
1517
# spider selected feeds
16-
config.load(sys.argv[1])
1718
for feed in sys.argv[2:]:
1819
spider.spiderFeed(feed)
1920
else:

splice.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,13 @@
55

66
import os.path
77
import sys
8-
from planet import splice
8+
from planet import splice, config
99

1010
if __name__ == '__main__':
1111

1212
if len(sys.argv) == 2 and os.path.isfile(sys.argv[1]):
13-
# at the moment, we don't have template support, so we cheat and
14-
# simply insert a XSLT processing instruction
15-
doc = splice.splice(sys.argv[1])
13+
config.load(sys.argv[1])
14+
doc = splice.splice()
1615
splice.apply(doc.toxml('utf-8'))
1716
else:
1817
print "Usage:"

tests/test_spider.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,8 @@ def test_spiderUpdate(self):
6666
self.test_spiderFeed()
6767

6868
def test_spiderPlanet(self):
69-
spiderPlanet(configfile)
69+
config.load(configfile)
70+
spiderPlanet()
7071
files = glob.glob(workdir+"/*")
7172

7273
# verify that exactly eight files + 1 source dir were produced

tests/test_splice.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,15 @@
11
#!/usr/bin/env python
22

33
import unittest
4-
from planet.splice import splice
4+
from planet.splice import splice, config
55

66
configfile = 'tests/data/splice/config.ini'
77

88
class SpliceTest(unittest.TestCase):
99

1010
def test_splice(self):
11-
doc = splice(configfile)
11+
config.load(configfile)
12+
doc = splice()
1213
self.assertEqual(12,len(doc.getElementsByTagName('entry')))
1314
self.assertEqual(4,len(doc.getElementsByTagName('planet:source')))
1415
self.assertEqual(16,len(doc.getElementsByTagName('planet:name')))

0 commit comments

Comments
 (0)