Adding the SQLAlchemyPipeline to the spider pipelines

.gitignore

@@ -0,0 +1 @@
+/*.db

scrarls.py

@@ -2,6 +2,7 @@
 
 import re
 import sys
+
 from datetime import datetime
 
 import scrapy
@@ -9,24 +10,69 @@ from scrapy.crawler import CrawlerProcess
 from scrapy.spiders import CrawlSpider, Rule
 from scrapy.linkextractors import LinkExtractor
 
-from sqlalchemy import create_engine, Column, Integer, String, Text, DateTime
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import create_engine, func, select, Engine, Integer, String, Text, DateTime
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 from sqlalchemy.orm import sessionmaker
 
-Base = declarative_base()
+
+class Base(DeclarativeBase):
+    """Base déclarative pour SQLAlchemy 2.0."""
+    pass
 
 
-class TvShow(Base):
-    __tablename__ = 'tvshows'
-    id = Column(Integer, primary_key=True)
-    rlsbb_id = Column(Integer, nullable=False)
-    article_title = Column(String(255), nullable=False)
-    title = Column(String(255), nullable=False)
-    date = Column(DateTime, nullable=False)
-    summary = Column(Text, nullable=True)
-    image_url = Column(String(length=255), nullable=True)
-    download_url = Column(String(length=255), nullable=True)
+class TvShowDB(Base):
+    """Modèle pour le stockage des épisodes (SQLAlchemy 2.0)."""
 
+    __tablename__: str = "tvshows"
+
+    id: Mapped[int] = mapped_column(
+        Integer,
+        primary_key=True,
+        autoincrement=True
+    )
+    post_id: Mapped[str] = mapped_column(
+        String(length=255),
+        nullable=False,
+        unique=True,
+        index=True
+    )
+    post_title: Mapped[str] = mapped_column(
+        String(255),
+        nullable=False
+    )
+    title: Mapped[str] = mapped_column(
+        String(255),
+        nullable=False,
+        index=True
+    )
+    date: Mapped[datetime] = mapped_column(
+        DateTime,
+        nullable=False,
+        index=True
+    )
+    summary: Mapped[str | None] = mapped_column(
+        Text,
+        nullable=True
+    )
+    image_url: Mapped[str | None] = mapped_column(
+        String(255),
+        nullable=True
+    )
+    download_url: Mapped[str | None] = mapped_column(
+        String(255),
+        nullable=True
+    )
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime,
+        server_default=func.datetime('now'),
+        nullable=False
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime,
+        server_default=func.datetime('now'),
+        onupdate=func.datetime('now'),
+        nullable=False
+    )
 
 class TvShowItem(scrapy.Item):
     post_id: scrapy.Field = scrapy.Field()
@@ -38,6 +84,33 @@ class TvShowItem(scrapy.Item):
     download_url: scrapy.Field = scrapy.Field()
 
 
+class SQLAlchemyPipeline:
+    def __init__(self):
+        self.engine: Engine = create_engine('sqlite:///tvshows.db', echo=True)
+        Base.metadata.create_all(self.engine)
+        self.Session = sessionmaker(bind=self.engine)
+
+    def process_item(self, item, spider):
+        session = self.Session()
+        try:
+            stmt = select(TvShowDB).where(TvShowDB.post_id == item["post_id"])
+            show = session.scalars(stmt).first()
+            print(f"{show=}")
+            if not show:
+                show = TvShowDB(**item)
+                session.add(show)
+            else:
+                for key, value in item.items():
+                    setattr(show, key, value)
+            session.commit()
+        except Exception as e:
+            session.rollback()
+            raise
+        finally:
+            session.close()
+        return item
+
+
 class TvShow(CrawlSpider):
     name: str = "rlsb_tvshow"
     allowed_domains: list[str] = ["rlsbb.ru"]
@@ -46,6 +119,9 @@ class TvShow(CrawlSpider):
         'USER_AGENT': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36',
         'AUTOTHROTTLE_ENABLED': True,
         'DOWNLOAD_DELAY': 10,
+        'ITEM_PIPELINES': {
+            '__main__.SQLAlchemyPipeline': 300,
+        },
     }
     rules: list[Rule] = [
         Rule(LinkExtractor(allow=r"/tv-shows/page/"), callback="parse", follow=True)
@@ -54,13 +130,13 @@ class TvShow(CrawlSpider):
     def parse(self, response):
         for article in response.css("article"):
             item = TvShowItem()
-            item['article_id'] = article.attrib['id'],
-            item['article_title'] = article.css('h1.entry-title > a::text').get(),
-            item['title'] = article.css('.entry-summary > p:nth-child(4) > strong::text').get(),
-            item['date'] = self.parse_date(article.css('.entry-meta-header-before::text').getall()[1].strip()),
-            item['summary'] = article.xpath('.//div[@class="entry-summary"]/node()').extract(),
-            item['image_url'] = article.css('.entry-summary > p > img::attr(src)').get(),
-            item['download_url'] = article.css('.entry-summary > p > a[href ^= "https://rapidgator"]::attr(href)').get()
+            item['post_id'] = article.attrib['id']
+            item['post_title'] = article.css('h1.entry-title > a::text').get()
+            item['title'] = article.css('.entry-summary > p:nth-child(4) > strong::text').get()
+            item['date'] = self.parse_date(article.css('.entry-meta-header-before::text').getall()[1].strip())
+            item['summary'] = "".join(article.xpath('.//div[@class="entry-summary"]/node()').extract())
+            item['image_url'] = article.css('.entry-summary > p > img::attr(src)').get()
+            item['download_url'] = "#".join(article.css('.entry-summary > p > a[href ^= "https://rapidgator"]::attr(href)').extract())
             yield item
 
     def parse_date(self, formatted_date: str):