crawl.py 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. import os
  2. from scrapy.commands import ScrapyCommand
  3. from scrapy.utils.conf import arglist_to_dict
  4. from scrapy.utils.python import without_none_values
  5. from scrapy.exceptions import UsageError
  6. class Command(ScrapyCommand):
  7. requires_project = True
  8. def syntax(self):
  9. return "[options] <spider>"
  10. def short_desc(self):
  11. return "Run a spider"
  12. def add_options(self, parser):
  13. ScrapyCommand.add_options(self, parser)
  14. parser.add_option("-a", dest="spargs", action="append", default=[], metavar="NAME=VALUE",
  15. help="set spider argument (may be repeated)")
  16. parser.add_option("-o", "--output", metavar="FILE",
  17. help="dump scraped items into FILE (use - for stdout)")
  18. parser.add_option("-t", "--output-format", metavar="FORMAT",
  19. help="format to use for dumping items with -o")
  20. def process_options(self, args, opts):
  21. ScrapyCommand.process_options(self, args, opts)
  22. try:
  23. opts.spargs = arglist_to_dict(opts.spargs)
  24. except ValueError:
  25. raise UsageError("Invalid -a value, use -a NAME=VALUE", print_help=False)
  26. if opts.output:
  27. if opts.output == '-':
  28. self.settings.set('FEED_URI', 'stdout:', priority='cmdline')
  29. else:
  30. self.settings.set('FEED_URI', opts.output, priority='cmdline')
  31. feed_exporters = without_none_values(
  32. self.settings.getwithbase('FEED_EXPORTERS'))
  33. valid_output_formats = feed_exporters.keys()
  34. if not opts.output_format:
  35. opts.output_format = os.path.splitext(opts.output)[1].replace(".", "")
  36. if opts.output_format not in valid_output_formats:
  37. raise UsageError("Unrecognized output format '%s', set one"
  38. " using the '-t' switch or as a file extension"
  39. " from the supported list %s" % (opts.output_format,
  40. tuple(valid_output_formats)))
  41. self.settings.set('FEED_FORMAT', opts.output_format, priority='cmdline')
  42. def run(self, args, opts):
  43. if len(args) < 1:
  44. raise UsageError()
  45. elif len(args) > 1:
  46. raise UsageError("running 'scrapy crawl' with more than one spider is no longer supported")
  47. spname = args[0]
  48. self.crawler_process.crawl(spname, **opts.spargs)
  49. self.crawler_process.start()
  50. if self.crawler_process.bootstrap_failed:
  51. self.exitcode = 1