mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-12 21:28:29 +03:00
Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
53494c961e | ||
|
|
f7839a99cc | ||
|
|
7a2119e6e9 | ||
|
|
3ae25e51e7 |
@@ -27,7 +27,7 @@ class WaczArchiverEnricher(Enricher, Archiver):
|
|||||||
def configs() -> dict:
|
def configs() -> dict:
|
||||||
return {
|
return {
|
||||||
"profile": {"default": None, "help": "browsertrix-profile (for profile generation see https://github.com/webrecorder/browsertrix-crawler#creating-and-using-browser-profiles)."},
|
"profile": {"default": None, "help": "browsertrix-profile (for profile generation see https://github.com/webrecorder/browsertrix-crawler#creating-and-using-browser-profiles)."},
|
||||||
"browsertrix_home": {"default": None, "help": "path to use when calling docker run with a volume, by default it will be the tmp folder generated during execution, but setting this option is needed when running the auto-archiver in a docker container that calls another container via DooD."},
|
"docker_commands": {"default": None, "help":"if a custom docker invocation is needed"},
|
||||||
"timeout": {"default": 120, "help": "timeout for WACZ generation in seconds"},
|
"timeout": {"default": 120, "help": "timeout for WACZ generation in seconds"},
|
||||||
"extract_media": {"default": True, "help": "If enabled all the images/videos/audio present in the WACZ archive will be extracted into separate Media. The .wacz file will be kept untouched."}
|
"extract_media": {"default": True, "help": "If enabled all the images/videos/audio present in the WACZ archive will be extracted into separate Media. The .wacz file will be kept untouched."}
|
||||||
}
|
}
|
||||||
@@ -47,51 +47,44 @@ class WaczArchiverEnricher(Enricher, Archiver):
|
|||||||
url = to_enrich.get_url()
|
url = to_enrich.get_url()
|
||||||
|
|
||||||
collection = str(uuid.uuid4())[0:8]
|
collection = str(uuid.uuid4())[0:8]
|
||||||
browsertrix_home = self.browsertrix_home or os.path.abspath(ArchivingContext.get_tmp_dir())
|
browsertrix_home_host = os.environ.get('BROWSERTRIX_HOME_HOST') or os.path.abspath(ArchivingContext.get_tmp_dir())
|
||||||
|
browsertrix_home_container = os.environ.get('BROWSERTRIX_HOME_CONTAINER') or browsertrix_home_host
|
||||||
|
|
||||||
if os.getenv('RUNNING_IN_DOCKER'):
|
cmd = [
|
||||||
|
"crawl",
|
||||||
|
"--url", url,
|
||||||
|
"--scopeType", "page",
|
||||||
|
"--generateWACZ",
|
||||||
|
"--text",
|
||||||
|
"--screenshot", "fullPage",
|
||||||
|
"--collection", collection,
|
||||||
|
"--id", collection,
|
||||||
|
"--saveState", "never",
|
||||||
|
"--behaviors", "autoscroll,autoplay,autofetch,siteSpecific",
|
||||||
|
"--behaviorTimeout", str(self.timeout),
|
||||||
|
"--timeout", str(self.timeout)]
|
||||||
|
|
||||||
|
# call docker if explicitly enabled or we are running on the host (not in docker)
|
||||||
|
use_docker = os.environ.get('WACZ_ENABLE_DOCKER') or not os.environ.get('RUNNING_IN_DOCKER')
|
||||||
|
|
||||||
|
if use_docker:
|
||||||
|
logger.debug(f"generating WACZ in Docker for {url=}")
|
||||||
|
logger.debug(f"{browsertrix_home_host=} {browsertrix_home_container=}")
|
||||||
|
if not self.docker_commands:
|
||||||
|
self.docker_commands = ["docker", "run", "--rm", "-v", f"{browsertrix_home_host}:/crawls/", "webrecorder/browsertrix-crawler"]
|
||||||
|
cmd = self.docker_commands + cmd
|
||||||
|
|
||||||
|
if self.profile:
|
||||||
|
profile_fn = os.path.join(browsertrix_home_container, "profile.tar.gz")
|
||||||
|
logger.debug(f"copying {self.profile} to {profile_fn}")
|
||||||
|
shutil.copyfile(self.profile, profile_fn)
|
||||||
|
cmd.extend(["--profile", os.path.join("/crawls", "profile.tar.gz")])
|
||||||
|
|
||||||
|
else:
|
||||||
logger.debug(f"generating WACZ without Docker for {url=}")
|
logger.debug(f"generating WACZ without Docker for {url=}")
|
||||||
|
|
||||||
cmd = [
|
|
||||||
"crawl",
|
|
||||||
"--url", url,
|
|
||||||
"--scopeType", "page",
|
|
||||||
"--generateWACZ",
|
|
||||||
"--text",
|
|
||||||
"--screenshot", "fullPage",
|
|
||||||
"--collection", collection,
|
|
||||||
"--id", collection,
|
|
||||||
"--saveState", "never",
|
|
||||||
"--behaviors", "autoscroll,autoplay,autofetch,siteSpecific",
|
|
||||||
"--behaviorTimeout", str(self.timeout),
|
|
||||||
"--timeout", str(self.timeout)]
|
|
||||||
|
|
||||||
if self.profile:
|
if self.profile:
|
||||||
cmd.extend(["--profile", os.path.join("/app", str(self.profile))])
|
cmd.extend(["--profile", os.path.join("/app", str(self.profile))])
|
||||||
else:
|
|
||||||
logger.debug(f"generating WACZ in Docker for {url=}")
|
|
||||||
|
|
||||||
cmd = [
|
|
||||||
"docker", "run",
|
|
||||||
"--rm", # delete container once it has completed running
|
|
||||||
"-v", f"{browsertrix_home}:/crawls/",
|
|
||||||
# "-it", # this leads to "the input device is not a TTY"
|
|
||||||
"webrecorder/browsertrix-crawler", "crawl",
|
|
||||||
"--url", url,
|
|
||||||
"--scopeType", "page",
|
|
||||||
"--generateWACZ",
|
|
||||||
"--text",
|
|
||||||
"--screenshot", "fullPage",
|
|
||||||
"--collection", collection,
|
|
||||||
"--behaviors", "autoscroll,autoplay,autofetch,siteSpecific",
|
|
||||||
"--behaviorTimeout", str(self.timeout),
|
|
||||||
"--timeout", str(self.timeout)
|
|
||||||
]
|
|
||||||
|
|
||||||
if self.profile:
|
|
||||||
profile_fn = os.path.join(browsertrix_home, "profile.tar.gz")
|
|
||||||
shutil.copyfile(self.profile, profile_fn)
|
|
||||||
cmd.extend(["--profile", os.path.join("/crawls", "profile.tar.gz")])
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
logger.info(f"Running browsertrix-crawler: {' '.join(cmd)}")
|
logger.info(f"Running browsertrix-crawler: {' '.join(cmd)}")
|
||||||
@@ -100,10 +93,10 @@ class WaczArchiverEnricher(Enricher, Archiver):
|
|||||||
logger.error(f"WACZ generation failed: {e}")
|
logger.error(f"WACZ generation failed: {e}")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if os.getenv('RUNNING_IN_DOCKER'):
|
if use_docker:
|
||||||
filename = os.path.join("collections", collection, f"{collection}.wacz")
|
filename = os.path.join(browsertrix_home_container, "collections", collection, f"{collection}.wacz")
|
||||||
else:
|
else:
|
||||||
filename = os.path.join(browsertrix_home, "collections", collection, f"{collection}.wacz")
|
filename = os.path.join("collections", collection, f"{collection}.wacz")
|
||||||
|
|
||||||
if not os.path.exists(filename):
|
if not os.path.exists(filename):
|
||||||
logger.warning(f"Unable to locate and upload WACZ {filename=}")
|
logger.warning(f"Unable to locate and upload WACZ {filename=}")
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ _MAJOR = "0"
|
|||||||
_MINOR = "6"
|
_MINOR = "6"
|
||||||
# On main and in a nightly release the patch should be one ahead of the last
|
# On main and in a nightly release the patch should be one ahead of the last
|
||||||
# released build.
|
# released build.
|
||||||
_PATCH = "8"
|
_PATCH = "10"
|
||||||
# This is mainly for nightly builds which have the suffix ".dev$DATE". See
|
# This is mainly for nightly builds which have the suffix ".dev$DATE". See
|
||||||
# https://semver.org/#is-v123-a-semantic-version for the semantics.
|
# https://semver.org/#is-v123-a-semantic-version for the semantics.
|
||||||
_SUFFIX = ""
|
_SUFFIX = ""
|
||||||
|
|||||||
Reference in New Issue
Block a user