euscanwww: scan_upstream imports euscan.scan

scan_upstream command imports euscan.scan and simply calls a function,
it doesn't call a subprocess

Signed-off-by: volpino <fox91@anche.no>
This commit is contained in:
volpino
2012-05-31 18:42:37 +02:00
parent e8718748b4
commit 47ec539e1d
3 changed files with 26 additions and 11 deletions

View File

@ -67,4 +67,4 @@ ROBOTS_TXT_BLACKLIST_DOMAINS = [
]
from out import EuscanOutput
output = out.EuscanOutput(CONFIG)
output = EuscanOutput(CONFIG)

View File

@ -95,7 +95,7 @@ class EuscanOutput(object):
"metadata": {},
}
def get_formatted_output(self):
def get_formatted_output(self, format_=None):
data = {}
for query in self.queries:
@ -105,8 +105,11 @@ class EuscanOutput(object):
"messages": self.queries[query]["output"].getvalue(),
}
if self.config["format"].lower() == "json":
format_ = format_ or self.config["format"]
if format_.lower() == "json":
return json.dumps(data, indent=self.config["indent"])
elif format_.lower() == "dict":
return data
else:
raise TypeError("Invalid output format")
@ -133,7 +136,7 @@ class EuscanOutput(object):
print "%s: %s" % (key.capitalize(), value)
def __getattr__(self, key):
if not self.config["quiet"]:
if not self.config["quiet"] and self.current_query is not None:
output = self.queries[self.current_query]["output"]
return getattr(output, key)
else: