Merge branch 'main' into odysee-refactor

cisticola/scraper/base.py

@@ -10,7 +10,8 @@ from loguru import logger
 import ffmpeg
 from sqlalchemy.orm import sessionmaker
 import yt_dlp
-from  sqlalchemy.sql.expression import func
+from sqlalchemy.sql.expression import func
+from pathlib import Path
 
 from cisticola.base import Channel, ScraperResult, mapper_registry
 from cisticola.utils import make_request
@@ -181,13 +182,20 @@ class Scraper:
         content_type = 'video/mp4'
 
         with tempfile.TemporaryDirectory() as temp_dir:
+            cookiefile = Path(temp_dir)/self.cookiefilename
+            with open(cookiefile, 'w') as f:
+                f.write(self.cookiestring)
+                
             ydl_opts = {
                 "format": "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best",
                 "merge_output_format": "mp4",
                 "outtmpl": f"{temp_dir}/%(id)s.%(ext)s",
                 "noplaylist": True,
-                'quiet': True,
-                "verbose": False,}
+                "quiet": True,
+                "verbose": False,
+                "retries": 5,
+                "cookiefile": cookiefile}
+
             ydl = yt_dlp.YoutubeDL(ydl_opts)
 
             try:
@@ -421,7 +429,7 @@ class ScraperController:
 
         # this query is really slow (~2.5 minutes) because of the shuffle. shuffling is so that multiple media archivers could work
         # simultaneously with low risk of collision (at least while the number of unarchived items is very large)
-        posts = session.query(ScraperResult).where(ScraperResult.media_archived == None).order_by(func.random()).limit(10000).all()
+        posts = session.query(ScraperResult).where(ScraperResult.media_archived == None).order_by(func.random()).limit(4000).all()
 
         logger.info(f"Found {len(posts)} posts without media. Archiving now")
 

cisticola/scraper/bitchute.py

@@ -16,7 +16,7 @@ from cisticola.scraper.base import Scraper
 class BitchuteScraper(Scraper):
     """An implementation of a Scraper for Bitchute, using classes from the 4cat
     library"""
-    __version__ = "BitchuteScraper 0.0.0"
+    __version__ = "BitchuteScraper 0.0.1"
 
     def get_username_from_url(self, url):
         username = url.split('bitchute.com/channel/')[-1].strip('/')

cisticola/scraper/rumble.py

@@ -5,6 +5,7 @@ from urllib.parse import urlparse
 from loguru import logger
 
 from bs4 import BeautifulSoup
+import os
 
 from cisticola.base import Channel, ScraperResult, RawChannelInfo
 from cisticola.scraper import Scraper, make_request
@@ -13,7 +14,10 @@ BASE_URL = 'https://rumble.com'
 
 class RumbleScraper(Scraper):
     """An implementation of a Scraper for Rumble, using custom functions"""
-    __version__ = "RumbleScraper 0.0.0"
+    __version__ = "RumbleScraper 0.0.1"
+
+    cookiestring = os.environ["YOUTUBE_COOKIESTRING"].replace(r'\n', '\n').replace(r'\t', '\t')
+    cookiefilename = 'cookiefile.txt'
 
     @logger.catch
     def get_posts(self, channel: Channel, since: ScraperResult = None, archive_media: bool = True) -> Generator[ScraperResult, None, None]:

cisticola/scraper/youtube.py

@@ -13,7 +13,10 @@ from cisticola.scraper import Scraper
 
 class YoutubeScraper(Scraper):
     """An implementation of a Scraper for Youtube, using youtube-dl"""
-    __version__ = "YoutubeScraper 0.0.0"
+    __version__ = "YoutubeScraper 0.0.1"
+
+    cookiestring = os.environ["YOUTUBE_COOKIESTRING"].replace(r'\n', '\n').replace(r'\t', '\t')
+    cookiefilename = 'cookiefile.txt'
 
     @logger.catch
     def get_posts(self, channel: Channel, since: ScraperResult = None, archive_media: bool = True) -> Generator[ScraperResult, None, None]:
@@ -29,13 +32,21 @@ class YoutubeScraper(Scraper):
 
         with tempfile.TemporaryDirectory() as temp_dir:
 
+            cookiefile = Path(temp_dir)/self.cookiefilename
+            with open(cookiefile, 'w') as f:
+                f.write(self.cookiestring)
+
             daterange = yt_dlp.utils.DateRange(start = start_date)
 
             ydl_opts = {
                 "format": "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best",
                 "merge_output_format": "mp4",
                 "outtmpl": f"{temp_dir}/%(id)s.%(ext)s",
-                "daterange" : daterange}
+                "daterange" : daterange,
+                "quiet": True,
+                "verbose": False,
+                "retries": 5,
+                "cookiefile": cookiefile}
 
             ydl = yt_dlp.YoutubeDL(ydl_opts)
 
@@ -92,10 +103,18 @@ class YoutubeScraper(Scraper):
 
                 with tempfile.TemporaryDirectory() as temp_dir:
 
+                    cookiefile = Path(temp_dir)/self.cookiefilename
+                    with open(cookiefile, 'w') as f:
+                        f.write(self.cookiestring)
+
                     ydl_opts = {
                         "format": "bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best",
                         "merge_output_format": "mp4",
-                        "outtmpl": f"{temp_dir}/%(id)s.%(ext)s"}
+                        "outtmpl": f"{temp_dir}/%(id)s.%(ext)s",
+                        "quiet": True,
+                        "verbose": False,
+                        "retries": 5,
+                        "cookiefile": cookiefile}
 
                     ydl = yt_dlp.YoutubeDL(ydl_opts)
 
@@ -104,7 +123,7 @@ class YoutubeScraper(Scraper):
                     except yt_dlp.utils.DownloadError as e:
                         raise e
                         
-                    files = os.listdir(temp_dir)
+                    files = [file for file in os.listdir(temp_dir) if file != self.cookiefilename]
                     if len(files) != 1:
                         logger.warning(f'{len(files)} files downloaded for video: {url}')
                     key = files[0]
@@ -120,7 +139,12 @@ class YoutubeScraper(Scraper):
         return result
 
     def get_profile(self, channel: Channel) -> RawChannelInfo:
-        ydl_opts = {}
+        
+        ydl_opts = {
+            "quiet": True,
+            "verbose": False,
+            "retries": 5}
+
         ydl = yt_dlp.YoutubeDL(ydl_opts)
 
         meta = None

cisticola/utils.py

@@ -31,21 +31,15 @@ def make_request(url, headers = None, max_retries = 5, break_codes = None):
 
     r = None
 
-    for n_retries in range(max_retries):
-        try:
-            r = request_until_200(
-                url = url, 
-                headers = headers, 
-                max_retries = max_retries,
-                break_codes = break_codes)
-            logger.debug(f"Request for url: {url} succeeded on attempt: {n_retries}/{max_retries}")
-        except Exception as e:
-            logger.warning(f"Request for url: {url} raised exception: [{e}] on attempt: {n_retries}/{max_retries}")
-            continue 
-        else:
-            break 
-    else:
-        logger.error(f"Request for url: {url} failed after {max_retries} attempts")
+    try:
+        r = request_until_200(
+            url = url, 
+            headers = headers, 
+            max_retries = max_retries,
+            break_codes = break_codes)
+        logger.debug(f"Request for url: {url} succeeded")
+    except Exception as e:
+        logger.warning(f"Request for url: {url} raised exception: [{e}]")
 
     return r