use*_*629 3 python sqlalchemy scrapy
在使用可抓取的蜘蛛抓取一些数据之后:
class Test_Spider(Spider):
name = "test"
def start_requests(self):
for i in range(900,902,1):
........
yield item
Run Code Online (Sandbox Code Playgroud)
我将数据传递给使用sqlalchemy写入sqllite表的管道对象:
class SQLlitePipeline(object):
def __init__(self):
_engine = create_engine("sqlite:///data.db")
_connection = _engine.connect()
_metadata = MetaData()
_stack_items = Table("table1", _metadata,
Column("id", Integer, primary_key=True),
Column("detail_url", Text),
_metadata.create_all(_engine)
self.connection = _connection
self.stack_items = _stack_items
def process_item(self, item, spider):
is_valid = True
Run Code Online (Sandbox Code Playgroud)
我希望能够将表名设置为变量,而不是像现在这样被硬编码(“ table1”)。如何才能做到这一点?
一种更简单的方法是将参数传递给crawl:
scrapy crawl -a table=table1
Run Code Online (Sandbox Code Playgroud)
然后通过以下方式获取值spider.table:
class TestScrapyPipeline(object):
def process_item(self, item, spider):
table = spider.table
Run Code Online (Sandbox Code Playgroud)
假设您通过命令行(例如-s table="table1")传递此参数,请定义一个from_crawler方法。
@classmethod
def from_crawler(cls, crawler):
# Here, you get whatever value was passed through the "table" parameter
settings = crawler.settings
table = settings.get('table')
# Instantiate the pipeline with your table
return cls(table)
def __init__(self, table):
_engine = create_engine("sqlite:///data.db")
_connection = _engine.connect()
_metadata = MetaData()
_stack_items = Table(table, _metadata,
Column("id", Integer, primary_key=True),
Column("detail_url", Text),
_metadata.create_all(_engine)
self.connection = _connection
self.stack_items = _stack_items
Run Code Online (Sandbox Code Playgroud)
class SQLlitePipeline(object):
def __init__(self, table_name):
_engine = create_engine("sqlite:///data.db")
_connection = _engine.connect()
_metadata = MetaData()
_stack_items = Table(table_name, _metadata,
Column("id", Integer, primary_key=True),
Column("detail_url", Text),
_metadata.create_all(_engine)
self.connection = _connection
self.stack_items = _stack_items
@classmethod
def from_crawler(cls, crawler):
table_name = getattr(crawler.spider, 'table_name')
return cls(table_name)
Run Code Online (Sandbox Code Playgroud)
随着from_crawler你可以创建或实例化一个管道对象与您指定的参数。