mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-12 05:08:28 +03:00
3b87dffe6b
The [browsertrix-crawler] utility is a browser-based crawler that can crawl one or more pages. browsertrix-crawler creates archives in the [WACZ] format which is essentially a standardized ZIP file (similar to DOCX, EPUB, JAR, etc) which can then be replayed using the [ReplayWeb.page] web component, or unzipped to get the original WARC data (the ISO standard format used by the Internet Archive Wayback Machine). This PR adds browsertrix-crawler to archiver classes where screenshots are made made. The WACZ is uploaded to storage and then added to a new column in the spreadsheet. A column can be added that will display the WACZ, loaded from cloud storage (S3, digitalocean, etc) using the client side ReplayWeb page. You can see an example of the spreadsheet here: https://docs.google.com/spreadsheets/d/1Tk-iJWzT9Sx2-YccuPttL9HcMdZEnhv_OR7Bc6tfeu8/edit#gid=0 browsertrix-crawler requires Docker to be installed. If Docker is not installed an error message will be logged and things continue as normal. [browsertrix-crawler]: https://github.com/webrecorder/browsertrix-crawler [WACZ]: https://specs.webrecorder.net/wacz/latest/ [ReplayWeb.page]: https://replayweb.page
73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
import os, traceback
|
|
import tiktok_downloader
|
|
from loguru import logger
|
|
|
|
from .base_archiver import Archiver, ArchiveResult
|
|
from storages import Storage
|
|
|
|
|
|
class TiktokArchiver(Archiver):
|
|
name = "tiktok"
|
|
|
|
def download(self, url, check_if_exists=False):
|
|
if 'tiktok.com' not in url:
|
|
return False
|
|
|
|
status = 'success'
|
|
|
|
try:
|
|
info = tiktok_downloader.info_post(url)
|
|
key = self.get_key(f'{info.id}.mp4')
|
|
filename = os.path.join(Storage.TMP_FOLDER, key)
|
|
logger.info(f'found video {key=}')
|
|
|
|
if check_if_exists and self.storage.exists(key):
|
|
status = 'already archived'
|
|
|
|
media = tiktok_downloader.snaptik(url).get_media()
|
|
|
|
if len(media) <= 0:
|
|
if status == 'already archived':
|
|
return ArchiveResult(status='Could not download media, but already archived', cdn_url=self.storage.get_cdn_url(key))
|
|
else:
|
|
return ArchiveResult(status='Could not download media')
|
|
|
|
logger.info(f'downloading video {key=}')
|
|
media[0].download(filename)
|
|
|
|
if status != 'already archived':
|
|
logger.info(f'uploading video {key=}')
|
|
self.storage.upload(filename, key)
|
|
|
|
try:
|
|
key_thumb, thumb_index = self.get_thumbnails(filename, key, duration=info.duration)
|
|
except Exception as e:
|
|
logger.error(e)
|
|
key_thumb = ''
|
|
thumb_index = 'error creating thumbnails'
|
|
|
|
hash = self.get_hash(filename)
|
|
screenshot = self.get_screenshot(url)
|
|
wacz = self.get_wacz(url)
|
|
|
|
try: os.remove(filename)
|
|
except FileNotFoundError:
|
|
logger.info(f'tmp file not found thus not deleted {filename}')
|
|
cdn_url = self.storage.get_cdn_url(key)
|
|
timestamp = info.create.isoformat() if hasattr(info, "create") else None
|
|
|
|
return ArchiveResult(status=status, cdn_url=cdn_url, thumbnail=key_thumb,
|
|
thumbnail_index=thumb_index, duration=getattr(info, "duration", 0), title=getattr(info, "caption", ""),
|
|
timestamp=timestamp, hash=hash, screenshot=screenshot, wacz=wacz)
|
|
|
|
except tiktok_downloader.Except.InvalidUrl as e:
|
|
status = 'Invalid URL'
|
|
logger.warning(f'Invalid URL on {url} {e}\n{traceback.format_exc()}')
|
|
return ArchiveResult(status=status)
|
|
|
|
except:
|
|
error = traceback.format_exc()
|
|
status = 'Other Tiktok error: ' + str(error)
|
|
logger.warning(f'Other Tiktok error' + str(error))
|
|
return ArchiveResult(status=status)
|