mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
Compare commits
44 Commits
Author | SHA1 | Date | |
---|---|---|---|
3f7c73df80 | |||
43100f69d5 | |||
d73c778b0a | |||
73c3eb0984 | |||
a519cbe05d | |||
b3ad9783c4 | |||
c1ccb6c53e | |||
51a713fa04 | |||
74ea039458 | |||
aaa6a2b6a4 | |||
e32a49480b | |||
be65051f9d | |||
3387bc5f20 | |||
3f0ae729d6 | |||
8e8c1c031a | |||
55678d7fee | |||
a57ee8b96b | |||
e367da925e | |||
77a109bb7e | |||
a3e1864a26 | |||
41cbcc509c | |||
77874b432b | |||
5c4c785e60 | |||
2aed4f6d1f | |||
34152fbe54 | |||
4047fe78f3 | |||
c28347122e | |||
5b38ab8cf1 | |||
bb25d32f03 | |||
ecaedea709 | |||
f05d1b1261 | |||
6aaa3071f9 | |||
c26c9352f1 | |||
d9ea688145 | |||
e8be6e498e | |||
e4b1fac045 | |||
402ae81335 | |||
52f31e2783 | |||
739522a151 | |||
6c034209b6 | |||
f6fbd5e4bb | |||
7024db1f13 | |||
23bfa42a0d | |||
fdb57884ed |
67
.github/workflows/package.yml
vendored
Normal file
67
.github/workflows/package.yml
vendored
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
name: Package Application with Pyinstaller
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- "*"
|
||||||
|
tags:
|
||||||
|
- "v*"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build:
|
||||||
|
|
||||||
|
runs-on: ${{ matrix.os }}
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
os: [ubuntu-latest, windows-latest]
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v2
|
||||||
|
|
||||||
|
- uses: actions/setup-python@v2
|
||||||
|
with:
|
||||||
|
python-version: '3.x'
|
||||||
|
|
||||||
|
- name: "Install dependencies"
|
||||||
|
run: "pip install setuptools pyinstaller rich requests beautifulsoup4 -f --upgrade"
|
||||||
|
|
||||||
|
- name: "Install sync_url.py"
|
||||||
|
run: "pyinstaller sync_url.py -F"
|
||||||
|
|
||||||
|
- uses: actions/upload-artifact@v2
|
||||||
|
with:
|
||||||
|
name: "Pferd Sync URL"
|
||||||
|
path: "dist/sync_url*"
|
||||||
|
|
||||||
|
release:
|
||||||
|
name: Release
|
||||||
|
|
||||||
|
needs: [build]
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
if: startsWith(github.ref, 'refs/tags/')
|
||||||
|
|
||||||
|
env:
|
||||||
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: "Checkout"
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
|
- name: "Download artifacts"
|
||||||
|
uses: actions/download-artifact@v2
|
||||||
|
with:
|
||||||
|
name: "Pferd Sync URL"
|
||||||
|
|
||||||
|
- name: "look at folder structure"
|
||||||
|
run: "ls -lah"
|
||||||
|
|
||||||
|
- name: "Create release"
|
||||||
|
uses: softprops/action-gh-release@v1
|
||||||
|
|
||||||
|
- name: "Upload release artifacts"
|
||||||
|
uses: softprops/action-gh-release@v1
|
||||||
|
with:
|
||||||
|
body: "Download sync_url (or sync_url.exe on Windows) and run it in the terminal or CMD."
|
||||||
|
files: |
|
||||||
|
sync_url
|
||||||
|
sync_url.exe
|
7
.gitignore
vendored
7
.gitignore
vendored
@ -1,7 +1,14 @@
|
|||||||
__pycache__/
|
__pycache__/
|
||||||
.venv/
|
.venv/
|
||||||
|
venv/
|
||||||
|
.idea/
|
||||||
|
build/
|
||||||
.mypy_cache/
|
.mypy_cache/
|
||||||
.tmp/
|
.tmp/
|
||||||
.env
|
.env
|
||||||
.vscode
|
.vscode
|
||||||
ilias_cookies.txt
|
ilias_cookies.txt
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
sync_url.spec
|
||||||
|
dist/
|
||||||
|
18
LICENSE
Normal file
18
LICENSE
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
Copyright 2019-2020 Garmelon, I-Al-Istannen, danstooamerican, pavelzw
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||||
|
this software and associated documentation files (the "Software"), to deal in
|
||||||
|
the Software without restriction, including without limitation the rights to
|
||||||
|
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
||||||
|
the Software, and to permit persons to whom the Software is furnished to do so,
|
||||||
|
subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||||||
|
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||||||
|
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
||||||
|
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
||||||
|
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
@ -22,7 +22,7 @@ class CookieJar:
|
|||||||
if cookie_file is None:
|
if cookie_file is None:
|
||||||
self._cookies = LWPCookieJar()
|
self._cookies = LWPCookieJar()
|
||||||
else:
|
else:
|
||||||
self._cookies = LWPCookieJar(cookie_file)
|
self._cookies = LWPCookieJar(str(cookie_file.resolve()))
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def cookies(self) -> LWPCookieJar:
|
def cookies(self) -> LWPCookieJar:
|
||||||
|
69
PFERD/download_summary.py
Normal file
69
PFERD/download_summary.py
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
"""
|
||||||
|
Provides a summary that keeps track of new modified or deleted files.
|
||||||
|
"""
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
|
||||||
|
class DownloadSummary:
|
||||||
|
"""
|
||||||
|
Keeps track of all new, modified or deleted files and provides a summary.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self._new_files: List[Path] = []
|
||||||
|
self._modified_files: List[Path] = []
|
||||||
|
self._deleted_files: List[Path] = []
|
||||||
|
|
||||||
|
@property
|
||||||
|
def new_files(self) -> List[Path]:
|
||||||
|
"""
|
||||||
|
Returns all new files.
|
||||||
|
"""
|
||||||
|
return self._new_files.copy()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def modified_files(self) -> List[Path]:
|
||||||
|
"""
|
||||||
|
Returns all modified files.
|
||||||
|
"""
|
||||||
|
return self._modified_files.copy()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def deleted_files(self) -> List[Path]:
|
||||||
|
"""
|
||||||
|
Returns all deleted files.
|
||||||
|
"""
|
||||||
|
return self._deleted_files.copy()
|
||||||
|
|
||||||
|
def merge(self, summary: 'DownloadSummary') -> None:
|
||||||
|
"""
|
||||||
|
Merges ourselves with the passed summary. Modifies this object, but not the passed one.
|
||||||
|
"""
|
||||||
|
self._new_files += summary.new_files
|
||||||
|
self._modified_files += summary.modified_files
|
||||||
|
self._deleted_files += summary.deleted_files
|
||||||
|
|
||||||
|
def add_deleted_file(self, path: Path) -> None:
|
||||||
|
"""
|
||||||
|
Registers a file as deleted.
|
||||||
|
"""
|
||||||
|
self._deleted_files.append(path)
|
||||||
|
|
||||||
|
def add_modified_file(self, path: Path) -> None:
|
||||||
|
"""
|
||||||
|
Registers a file as changed.
|
||||||
|
"""
|
||||||
|
self._modified_files.append(path)
|
||||||
|
|
||||||
|
def add_new_file(self, path: Path) -> None:
|
||||||
|
"""
|
||||||
|
Registers a file as new.
|
||||||
|
"""
|
||||||
|
self._new_files.append(path)
|
||||||
|
|
||||||
|
def has_updates(self) -> bool:
|
||||||
|
"""
|
||||||
|
Returns whether this summary has any updates.
|
||||||
|
"""
|
||||||
|
return bool(self._new_files or self._modified_files or self._deleted_files)
|
@ -67,7 +67,7 @@ class KitShibbolethAuthenticator(IliasAuthenticator):
|
|||||||
while not self._login_successful(soup):
|
while not self._login_successful(soup):
|
||||||
# Searching the form here so that this fails before asking for
|
# Searching the form here so that this fails before asking for
|
||||||
# credentials rather than after asking.
|
# credentials rather than after asking.
|
||||||
form = soup.find("form", {"class": "form2", "method": "post"})
|
form = soup.find("form", {"class": "full content", "method": "post"})
|
||||||
action = form["action"]
|
action = form["action"]
|
||||||
|
|
||||||
# Equivalent: Enter credentials in
|
# Equivalent: Enter credentials in
|
||||||
|
@ -38,6 +38,12 @@ class IliasElementType(Enum):
|
|||||||
FORUM = "FORUM"
|
FORUM = "FORUM"
|
||||||
EXTERNAL_LINK = "EXTERNAL_LINK"
|
EXTERNAL_LINK = "EXTERNAL_LINK"
|
||||||
|
|
||||||
|
def is_folder(self) -> bool:
|
||||||
|
"""
|
||||||
|
Returns whether this type is some kind of folder.
|
||||||
|
"""
|
||||||
|
return "FOLDER" in str(self.name)
|
||||||
|
|
||||||
|
|
||||||
IliasDirectoryFilter = Callable[[Path, IliasElementType], bool]
|
IliasDirectoryFilter = Callable[[Path, IliasElementType], bool]
|
||||||
|
|
||||||
@ -110,6 +116,16 @@ class IliasCrawler:
|
|||||||
|
|
||||||
return urlunsplit((scheme, netloc, path, new_query_string, fragment))
|
return urlunsplit((scheme, netloc, path, new_query_string, fragment))
|
||||||
|
|
||||||
|
def recursive_crawl_url(self, url: str) -> List[IliasDownloadInfo]:
|
||||||
|
"""
|
||||||
|
Crawls a given url *and all reachable elements in it*.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
url {str} -- the *full* url to crawl
|
||||||
|
"""
|
||||||
|
start_entries: List[IliasCrawlerEntry] = self._crawl_folder(Path(""), url)
|
||||||
|
return self._iterate_entries_to_download_infos(start_entries)
|
||||||
|
|
||||||
def crawl_course(self, course_id: str) -> List[IliasDownloadInfo]:
|
def crawl_course(self, course_id: str) -> List[IliasDownloadInfo]:
|
||||||
"""
|
"""
|
||||||
Starts the crawl process for a course, yielding a list of elements to (potentially)
|
Starts the crawl process for a course, yielding a list of elements to (potentially)
|
||||||
@ -128,7 +144,7 @@ class IliasCrawler:
|
|||||||
|
|
||||||
if not self._is_course_id_valid(root_url, course_id):
|
if not self._is_course_id_valid(root_url, course_id):
|
||||||
raise FatalException(
|
raise FatalException(
|
||||||
"Invalid course id? The URL the server returned did not contain my id."
|
"Invalid course id? I didn't find anything looking like a course!"
|
||||||
)
|
)
|
||||||
|
|
||||||
# And treat it as a folder
|
# And treat it as a folder
|
||||||
@ -137,7 +153,34 @@ class IliasCrawler:
|
|||||||
|
|
||||||
def _is_course_id_valid(self, root_url: str, course_id: str) -> bool:
|
def _is_course_id_valid(self, root_url: str, course_id: str) -> bool:
|
||||||
response: requests.Response = self._session.get(root_url)
|
response: requests.Response = self._session.get(root_url)
|
||||||
return course_id in response.url
|
# We were redirected ==> Non-existant ID
|
||||||
|
if course_id not in response.url:
|
||||||
|
return False
|
||||||
|
|
||||||
|
link_element: bs4.Tag = self._get_page(root_url, {}).find(id="current_perma_link")
|
||||||
|
if not link_element:
|
||||||
|
return False
|
||||||
|
# It wasn't a course but a category list, forum, etc.
|
||||||
|
return "crs_" in link_element.get("value")
|
||||||
|
|
||||||
|
def find_course_name(self, course_id: str) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
Returns the name of a given course. None if it is not a valid course
|
||||||
|
or it could not be found.
|
||||||
|
"""
|
||||||
|
course_url = self._url_set_query_param(
|
||||||
|
self._base_url + "/goto.php", "target", f"crs_{course_id}"
|
||||||
|
)
|
||||||
|
return self.find_element_name(course_url)
|
||||||
|
|
||||||
|
def find_element_name(self, url: str) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
Returns the name of the element at the given URL, if it can find one.
|
||||||
|
"""
|
||||||
|
focus_element: bs4.Tag = self._get_page(url, {}).find(id="il_mhead_t_focus")
|
||||||
|
if not focus_element:
|
||||||
|
return None
|
||||||
|
return focus_element.text
|
||||||
|
|
||||||
def crawl_personal_desktop(self) -> List[IliasDownloadInfo]:
|
def crawl_personal_desktop(self) -> List[IliasDownloadInfo]:
|
||||||
"""
|
"""
|
||||||
@ -167,7 +210,7 @@ class IliasCrawler:
|
|||||||
PRETTY.not_searching(entry.path, "forum")
|
PRETTY.not_searching(entry.path, "forum")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if not self.dir_filter(entry.path, entry.entry_type):
|
if entry.entry_type.is_folder() and not self.dir_filter(entry.path, entry.entry_type):
|
||||||
PRETTY.not_searching(entry.path, "user filter")
|
PRETTY.not_searching(entry.path, "user filter")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
@ -202,6 +245,15 @@ class IliasCrawler:
|
|||||||
"""
|
"""
|
||||||
soup = self._get_page(url, {})
|
soup = self._get_page(url, {})
|
||||||
|
|
||||||
|
if soup.find(id="headerimage"):
|
||||||
|
element: bs4.Tag = soup.find(id="headerimage")
|
||||||
|
if "opencast" in element.attrs["src"].lower():
|
||||||
|
PRETTY.warning(f"Switched to crawling a video at {folder_path}")
|
||||||
|
if not self.dir_filter(folder_path, IliasElementType.VIDEO_FOLDER):
|
||||||
|
PRETTY.not_searching(folder_path, "user filter")
|
||||||
|
return []
|
||||||
|
return self._crawl_video_directory(folder_path, url)
|
||||||
|
|
||||||
result: List[IliasCrawlerEntry] = []
|
result: List[IliasCrawlerEntry] = []
|
||||||
|
|
||||||
# Fetch all links and throw them to the general interpreter
|
# Fetch all links and throw them to the general interpreter
|
||||||
@ -425,7 +477,8 @@ class IliasCrawler:
|
|||||||
results: List[IliasCrawlerEntry] = []
|
results: List[IliasCrawlerEntry] = []
|
||||||
|
|
||||||
# We can download everything directly!
|
# We can download everything directly!
|
||||||
if len(direct_download_links) == len(video_links):
|
# FIXME: Sadly the download button is currently broken, so never do that
|
||||||
|
if False and len(direct_download_links) == len(video_links):
|
||||||
for link in direct_download_links:
|
for link in direct_download_links:
|
||||||
results += self._crawl_single_video(video_dir_path, link, True)
|
results += self._crawl_single_video(video_dir_path, link, True)
|
||||||
else:
|
else:
|
||||||
|
@ -2,6 +2,8 @@
|
|||||||
|
|
||||||
import datetime
|
import datetime
|
||||||
import logging
|
import logging
|
||||||
|
import math
|
||||||
|
import os
|
||||||
from pathlib import Path, PurePath
|
from pathlib import Path, PurePath
|
||||||
from typing import Callable, List, Optional, Union
|
from typing import Callable, List, Optional, Union
|
||||||
|
|
||||||
@ -82,9 +84,13 @@ class IliasDownloader:
|
|||||||
session: requests.Session,
|
session: requests.Session,
|
||||||
authenticator: IliasAuthenticator,
|
authenticator: IliasAuthenticator,
|
||||||
strategy: IliasDownloadStrategy,
|
strategy: IliasDownloadStrategy,
|
||||||
|
timeout: int = 5
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Create a new IliasDownloader.
|
Create a new IliasDownloader.
|
||||||
|
|
||||||
|
The timeout applies to the download request only, as bwcloud uses IPv6
|
||||||
|
and requests has a problem with that: https://github.com/psf/requests/issues/5522
|
||||||
"""
|
"""
|
||||||
|
|
||||||
self._tmp_dir = tmp_dir
|
self._tmp_dir = tmp_dir
|
||||||
@ -92,6 +98,7 @@ class IliasDownloader:
|
|||||||
self._session = session
|
self._session = session
|
||||||
self._authenticator = authenticator
|
self._authenticator = authenticator
|
||||||
self._strategy = strategy
|
self._strategy = strategy
|
||||||
|
self._timeout = timeout
|
||||||
|
|
||||||
def download_all(self, infos: List[IliasDownloadInfo]) -> None:
|
def download_all(self, infos: List[IliasDownloadInfo]) -> None:
|
||||||
"""
|
"""
|
||||||
@ -119,7 +126,15 @@ class IliasDownloader:
|
|||||||
LOGGER.info("Retrying download: %r", info)
|
LOGGER.info("Retrying download: %r", info)
|
||||||
self._authenticator.authenticate(self._session)
|
self._authenticator.authenticate(self._session)
|
||||||
|
|
||||||
self._organizer.accept_file(tmp_file, info.path)
|
dst_path = self._organizer.accept_file(tmp_file, info.path)
|
||||||
|
if dst_path and info.modification_date:
|
||||||
|
os.utime(
|
||||||
|
dst_path,
|
||||||
|
times=(
|
||||||
|
math.ceil(info.modification_date.timestamp()),
|
||||||
|
math.ceil(info.modification_date.timestamp())
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def _try_download(self, info: IliasDownloadInfo, target: Path) -> bool:
|
def _try_download(self, info: IliasDownloadInfo, target: Path) -> bool:
|
||||||
url = info.url()
|
url = info.url()
|
||||||
@ -127,10 +142,11 @@ class IliasDownloader:
|
|||||||
PRETTY.warning(f"Could not download {str(info.path)!r} as I got no URL :/")
|
PRETTY.warning(f"Could not download {str(info.path)!r} as I got no URL :/")
|
||||||
return True
|
return True
|
||||||
|
|
||||||
with self._session.get(url, stream=True) as response:
|
with self._session.get(url, stream=True, timeout=self._timeout) as response:
|
||||||
content_type = response.headers["content-type"]
|
content_type = response.headers["content-type"]
|
||||||
|
has_content_disposition = "content-disposition" in response.headers
|
||||||
|
|
||||||
if content_type.startswith("text/html"):
|
if content_type.startswith("text/html") and not has_content_disposition:
|
||||||
if self._is_logged_in(soupify(response)):
|
if self._is_logged_in(soupify(response)):
|
||||||
raise ContentTypeException("Attempting to download a web page, not a file")
|
raise ContentTypeException("Attempting to download a web page, not a file")
|
||||||
|
|
||||||
|
@ -3,14 +3,18 @@ Contains a few logger utility functions and implementations.
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from typing import Optional
|
from pathlib import Path
|
||||||
|
from typing import List, Optional
|
||||||
|
|
||||||
|
from rich import print as rich_print
|
||||||
from rich._log_render import LogRender
|
from rich._log_render import LogRender
|
||||||
from rich.console import Console
|
from rich.console import Console
|
||||||
|
from rich.panel import Panel
|
||||||
from rich.style import Style
|
from rich.style import Style
|
||||||
from rich.text import Text
|
from rich.text import Text
|
||||||
from rich.theme import Theme
|
from rich.theme import Theme
|
||||||
|
|
||||||
|
from .download_summary import DownloadSummary
|
||||||
from .utils import PathLike, to_path
|
from .utils import PathLike, to_path
|
||||||
|
|
||||||
STYLE = "{"
|
STYLE = "{"
|
||||||
@ -111,6 +115,15 @@ class PrettyLogger:
|
|||||||
f"[bold green]Created {self._format_path(path)}.[/bold green]"
|
f"[bold green]Created {self._format_path(path)}.[/bold green]"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def deleted_file(self, path: PathLike) -> None:
|
||||||
|
"""
|
||||||
|
A file has been deleted.
|
||||||
|
"""
|
||||||
|
|
||||||
|
self.logger.info(
|
||||||
|
f"[bold red]Deleted {self._format_path(path)}.[/bold red]"
|
||||||
|
)
|
||||||
|
|
||||||
def ignored_file(self, path: PathLike, reason: str) -> None:
|
def ignored_file(self, path: PathLike, reason: str) -> None:
|
||||||
"""
|
"""
|
||||||
File was not downloaded or modified.
|
File was not downloaded or modified.
|
||||||
@ -138,6 +151,23 @@ class PrettyLogger:
|
|||||||
f"([/dim]{reason}[dim]).[/dim]"
|
f"([/dim]{reason}[dim]).[/dim]"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def summary(self, download_summary: DownloadSummary) -> None:
|
||||||
|
"""
|
||||||
|
Prints a download summary.
|
||||||
|
"""
|
||||||
|
self.logger.info("")
|
||||||
|
self.logger.info("[bold cyan]Download Summary[/bold cyan]")
|
||||||
|
if not download_summary.has_updates():
|
||||||
|
self.logger.info("[bold dim]Nothing changed![/bold dim]")
|
||||||
|
return
|
||||||
|
|
||||||
|
for new_file in download_summary.new_files:
|
||||||
|
self.new_file(new_file)
|
||||||
|
for modified_file in download_summary.modified_files:
|
||||||
|
self.modified_file(modified_file)
|
||||||
|
for deleted_files in download_summary.deleted_files:
|
||||||
|
self.deleted_file(deleted_files)
|
||||||
|
|
||||||
def starting_synchronizer(
|
def starting_synchronizer(
|
||||||
self,
|
self,
|
||||||
target_directory: PathLike,
|
target_directory: PathLike,
|
||||||
|
@ -5,10 +5,12 @@ A organizer is bound to a single directory.
|
|||||||
|
|
||||||
import filecmp
|
import filecmp
|
||||||
import logging
|
import logging
|
||||||
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
from pathlib import Path, PurePath
|
from pathlib import Path, PurePath
|
||||||
from typing import List, Set
|
from typing import List, Optional, Set
|
||||||
|
|
||||||
|
from .download_summary import DownloadSummary
|
||||||
from .location import Location
|
from .location import Location
|
||||||
from .logging import PrettyLogger
|
from .logging import PrettyLogger
|
||||||
from .utils import prompt_yes_no
|
from .utils import prompt_yes_no
|
||||||
@ -32,10 +34,28 @@ class Organizer(Location):
|
|||||||
# Keep the root dir
|
# Keep the root dir
|
||||||
self._known_files.add(path.resolve())
|
self._known_files.add(path.resolve())
|
||||||
|
|
||||||
def accept_file(self, src: Path, dst: PurePath) -> None:
|
self.download_summary = DownloadSummary()
|
||||||
"""Move a file to this organizer and mark it."""
|
|
||||||
src_absolute = src.resolve()
|
def accept_file(self, src: Path, dst: PurePath) -> Optional[Path]:
|
||||||
dst_absolute = self.resolve(dst)
|
"""
|
||||||
|
Move a file to this organizer and mark it.
|
||||||
|
|
||||||
|
Returns the path the file was moved to, to allow the caller to adjust the metadata.
|
||||||
|
As you might still need to adjust the metadata when the file was identical
|
||||||
|
(e.g. update the timestamp), the path is also returned in this case.
|
||||||
|
In all other cases (ignored, not overwritten, etc.) this method returns None.
|
||||||
|
"""
|
||||||
|
# Windows limits the path length to 260 for *some* historical reason
|
||||||
|
# If you want longer paths, you will have to add the "\\?\" prefix in front of
|
||||||
|
# your path...
|
||||||
|
# See:
|
||||||
|
# https://docs.microsoft.com/en-us/windows/win32/fileio/naming-a-file#maximum-path-length-limitation
|
||||||
|
if os.name == 'nt':
|
||||||
|
src_absolute = Path("\\\\?\\" + str(src.resolve()))
|
||||||
|
dst_absolute = Path("\\\\?\\" + str(self.resolve(dst)))
|
||||||
|
else:
|
||||||
|
src_absolute = src.resolve()
|
||||||
|
dst_absolute = self.resolve(dst)
|
||||||
|
|
||||||
if not src_absolute.exists():
|
if not src_absolute.exists():
|
||||||
raise FileAcceptException("Source file does not exist")
|
raise FileAcceptException("Source file does not exist")
|
||||||
@ -49,7 +69,7 @@ class Organizer(Location):
|
|||||||
PRETTY.warning(f"File {str(dst_absolute)!r} was already written!")
|
PRETTY.warning(f"File {str(dst_absolute)!r} was already written!")
|
||||||
if not prompt_yes_no(f"Overwrite file?", default=False):
|
if not prompt_yes_no(f"Overwrite file?", default=False):
|
||||||
PRETTY.ignored_file(dst_absolute, "file was written previously")
|
PRETTY.ignored_file(dst_absolute, "file was written previously")
|
||||||
return
|
return None
|
||||||
|
|
||||||
# Destination file is directory
|
# Destination file is directory
|
||||||
if dst_absolute.exists() and dst_absolute.is_dir():
|
if dst_absolute.exists() and dst_absolute.is_dir():
|
||||||
@ -57,7 +77,7 @@ class Organizer(Location):
|
|||||||
shutil.rmtree(dst_absolute)
|
shutil.rmtree(dst_absolute)
|
||||||
else:
|
else:
|
||||||
PRETTY.warning(f"Could not add file {str(dst_absolute)!r}")
|
PRETTY.warning(f"Could not add file {str(dst_absolute)!r}")
|
||||||
return
|
return None
|
||||||
|
|
||||||
# Destination file exists
|
# Destination file exists
|
||||||
if dst_absolute.exists() and dst_absolute.is_file():
|
if dst_absolute.exists() and dst_absolute.is_file():
|
||||||
@ -65,10 +85,12 @@ class Organizer(Location):
|
|||||||
# Bail out, nothing more to do
|
# Bail out, nothing more to do
|
||||||
PRETTY.ignored_file(dst_absolute, "same file contents")
|
PRETTY.ignored_file(dst_absolute, "same file contents")
|
||||||
self.mark(dst)
|
self.mark(dst)
|
||||||
return
|
return dst_absolute
|
||||||
|
|
||||||
|
self.download_summary.add_modified_file(dst_absolute)
|
||||||
PRETTY.modified_file(dst_absolute)
|
PRETTY.modified_file(dst_absolute)
|
||||||
else:
|
else:
|
||||||
|
self.download_summary.add_new_file(dst_absolute)
|
||||||
PRETTY.new_file(dst_absolute)
|
PRETTY.new_file(dst_absolute)
|
||||||
|
|
||||||
# Create parent dir if needed
|
# Create parent dir if needed
|
||||||
@ -80,6 +102,8 @@ class Organizer(Location):
|
|||||||
|
|
||||||
self.mark(dst)
|
self.mark(dst)
|
||||||
|
|
||||||
|
return dst_absolute
|
||||||
|
|
||||||
def mark(self, path: PurePath) -> None:
|
def mark(self, path: PurePath) -> None:
|
||||||
"""Mark a file as used so it will not get cleaned up."""
|
"""Mark a file as used so it will not get cleaned up."""
|
||||||
absolute_path = self.resolve(path)
|
absolute_path = self.resolve(path)
|
||||||
@ -115,9 +139,9 @@ class Organizer(Location):
|
|||||||
if start_dir.resolve() not in self._known_files and dir_empty:
|
if start_dir.resolve() not in self._known_files and dir_empty:
|
||||||
start_dir.rmdir()
|
start_dir.rmdir()
|
||||||
|
|
||||||
@staticmethod
|
def _delete_file_if_confirmed(self, path: Path) -> None:
|
||||||
def _delete_file_if_confirmed(path: Path) -> None:
|
|
||||||
prompt = f"Do you want to delete {path}"
|
prompt = f"Do you want to delete {path}"
|
||||||
|
|
||||||
if prompt_yes_no(prompt, False):
|
if prompt_yes_no(prompt, False):
|
||||||
|
self.download_summary.add_deleted_file(path)
|
||||||
path.unlink()
|
path.unlink()
|
||||||
|
100
PFERD/pferd.py
100
PFERD/pferd.py
@ -9,6 +9,7 @@ from typing import Callable, List, Optional, Union
|
|||||||
from .cookie_jar import CookieJar
|
from .cookie_jar import CookieJar
|
||||||
from .diva import (DivaDownloader, DivaDownloadStrategy, DivaPlaylistCrawler,
|
from .diva import (DivaDownloader, DivaDownloadStrategy, DivaPlaylistCrawler,
|
||||||
diva_download_new)
|
diva_download_new)
|
||||||
|
from .download_summary import DownloadSummary
|
||||||
from .errors import FatalException, swallow_and_print_errors
|
from .errors import FatalException, swallow_and_print_errors
|
||||||
from .ilias import (IliasAuthenticator, IliasCrawler, IliasDirectoryFilter,
|
from .ilias import (IliasAuthenticator, IliasCrawler, IliasDirectoryFilter,
|
||||||
IliasDownloader, IliasDownloadInfo, IliasDownloadStrategy,
|
IliasDownloader, IliasDownloadInfo, IliasDownloadStrategy,
|
||||||
@ -42,6 +43,7 @@ class Pferd(Location):
|
|||||||
):
|
):
|
||||||
super().__init__(Path(base_dir))
|
super().__init__(Path(base_dir))
|
||||||
|
|
||||||
|
self._download_summary = DownloadSummary()
|
||||||
self._tmp_dir = TmpDir(self.resolve(tmp_dir))
|
self._tmp_dir = TmpDir(self.resolve(tmp_dir))
|
||||||
self._test_run = test_run
|
self._test_run = test_run
|
||||||
|
|
||||||
@ -70,7 +72,8 @@ class Pferd(Location):
|
|||||||
dir_filter: IliasDirectoryFilter,
|
dir_filter: IliasDirectoryFilter,
|
||||||
transform: Transform,
|
transform: Transform,
|
||||||
download_strategy: IliasDownloadStrategy,
|
download_strategy: IliasDownloadStrategy,
|
||||||
clean: bool = True
|
timeout: int,
|
||||||
|
clean: bool = True,
|
||||||
) -> Organizer:
|
) -> Organizer:
|
||||||
# pylint: disable=too-many-locals
|
# pylint: disable=too-many-locals
|
||||||
cookie_jar = CookieJar(to_path(cookies) if cookies else None)
|
cookie_jar = CookieJar(to_path(cookies) if cookies else None)
|
||||||
@ -79,7 +82,8 @@ class Pferd(Location):
|
|||||||
organizer = Organizer(self.resolve(to_path(target)))
|
organizer = Organizer(self.resolve(to_path(target)))
|
||||||
|
|
||||||
crawler = IliasCrawler(base_url, session, authenticator, dir_filter)
|
crawler = IliasCrawler(base_url, session, authenticator, dir_filter)
|
||||||
downloader = IliasDownloader(tmp_dir, organizer, session, authenticator, download_strategy)
|
downloader = IliasDownloader(tmp_dir, organizer, session,
|
||||||
|
authenticator, download_strategy, timeout)
|
||||||
|
|
||||||
cookie_jar.load_cookies()
|
cookie_jar.load_cookies()
|
||||||
info = crawl_function(crawler)
|
info = crawl_function(crawler)
|
||||||
@ -110,6 +114,7 @@ class Pferd(Location):
|
|||||||
password: Optional[str] = None,
|
password: Optional[str] = None,
|
||||||
download_strategy: IliasDownloadStrategy = download_modified_or_new,
|
download_strategy: IliasDownloadStrategy = download_modified_or_new,
|
||||||
clean: bool = True,
|
clean: bool = True,
|
||||||
|
timeout: int = 5,
|
||||||
) -> Organizer:
|
) -> Organizer:
|
||||||
"""
|
"""
|
||||||
Synchronizes a folder with the ILIAS instance of the KIT.
|
Synchronizes a folder with the ILIAS instance of the KIT.
|
||||||
@ -135,11 +140,14 @@ class Pferd(Location):
|
|||||||
be downloaded. Can save bandwidth and reduce the number of requests.
|
be downloaded. Can save bandwidth and reduce the number of requests.
|
||||||
(default: {download_modified_or_new})
|
(default: {download_modified_or_new})
|
||||||
clean {bool} -- Whether to clean up when the method finishes.
|
clean {bool} -- Whether to clean up when the method finishes.
|
||||||
|
timeout {int} -- The download timeout for opencast videos. Sadly needed due to a
|
||||||
|
requests bug.
|
||||||
"""
|
"""
|
||||||
# This authenticator only works with the KIT ilias instance.
|
# This authenticator only works with the KIT ilias instance.
|
||||||
authenticator = KitShibbolethAuthenticator(username=username, password=password)
|
authenticator = KitShibbolethAuthenticator(username=username, password=password)
|
||||||
PRETTY.starting_synchronizer(target, "ILIAS", course_id)
|
PRETTY.starting_synchronizer(target, "ILIAS", course_id)
|
||||||
return self._ilias(
|
|
||||||
|
organizer = self._ilias(
|
||||||
target=target,
|
target=target,
|
||||||
base_url="https://ilias.studium.kit.edu/",
|
base_url="https://ilias.studium.kit.edu/",
|
||||||
crawl_function=lambda crawler: crawler.crawl_course(course_id),
|
crawl_function=lambda crawler: crawler.crawl_course(course_id),
|
||||||
@ -149,8 +157,19 @@ class Pferd(Location):
|
|||||||
transform=transform,
|
transform=transform,
|
||||||
download_strategy=download_strategy,
|
download_strategy=download_strategy,
|
||||||
clean=clean,
|
clean=clean,
|
||||||
|
timeout=timeout
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._download_summary.merge(organizer.download_summary)
|
||||||
|
|
||||||
|
return organizer
|
||||||
|
|
||||||
|
def print_summary(self) -> None:
|
||||||
|
"""
|
||||||
|
Prints the accumulated download summary.
|
||||||
|
"""
|
||||||
|
PRETTY.summary(self._download_summary)
|
||||||
|
|
||||||
@swallow_and_print_errors
|
@swallow_and_print_errors
|
||||||
def ilias_kit_personal_desktop(
|
def ilias_kit_personal_desktop(
|
||||||
self,
|
self,
|
||||||
@ -162,6 +181,7 @@ class Pferd(Location):
|
|||||||
password: Optional[str] = None,
|
password: Optional[str] = None,
|
||||||
download_strategy: IliasDownloadStrategy = download_modified_or_new,
|
download_strategy: IliasDownloadStrategy = download_modified_or_new,
|
||||||
clean: bool = True,
|
clean: bool = True,
|
||||||
|
timeout: int = 5,
|
||||||
) -> Organizer:
|
) -> Organizer:
|
||||||
"""
|
"""
|
||||||
Synchronizes a folder with the ILIAS instance of the KIT. This method will crawl the ILIAS
|
Synchronizes a folder with the ILIAS instance of the KIT. This method will crawl the ILIAS
|
||||||
@ -186,11 +206,14 @@ class Pferd(Location):
|
|||||||
be downloaded. Can save bandwidth and reduce the number of requests.
|
be downloaded. Can save bandwidth and reduce the number of requests.
|
||||||
(default: {download_modified_or_new})
|
(default: {download_modified_or_new})
|
||||||
clean {bool} -- Whether to clean up when the method finishes.
|
clean {bool} -- Whether to clean up when the method finishes.
|
||||||
|
timeout {int} -- The download timeout for opencast videos. Sadly needed due to a
|
||||||
|
requests bug.
|
||||||
"""
|
"""
|
||||||
# This authenticator only works with the KIT ilias instance.
|
# This authenticator only works with the KIT ilias instance.
|
||||||
authenticator = KitShibbolethAuthenticator(username=username, password=password)
|
authenticator = KitShibbolethAuthenticator(username=username, password=password)
|
||||||
PRETTY.starting_synchronizer(target, "ILIAS", "Personal Desktop")
|
PRETTY.starting_synchronizer(target, "ILIAS", "Personal Desktop")
|
||||||
return self._ilias(
|
|
||||||
|
organizer = self._ilias(
|
||||||
target=target,
|
target=target,
|
||||||
base_url="https://ilias.studium.kit.edu/",
|
base_url="https://ilias.studium.kit.edu/",
|
||||||
crawl_function=lambda crawler: crawler.crawl_personal_desktop(),
|
crawl_function=lambda crawler: crawler.crawl_personal_desktop(),
|
||||||
@ -200,8 +223,77 @@ class Pferd(Location):
|
|||||||
transform=transform,
|
transform=transform,
|
||||||
download_strategy=download_strategy,
|
download_strategy=download_strategy,
|
||||||
clean=clean,
|
clean=clean,
|
||||||
|
timeout=timeout
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._download_summary.merge(organizer.download_summary)
|
||||||
|
|
||||||
|
return organizer
|
||||||
|
|
||||||
|
@swallow_and_print_errors
|
||||||
|
def ilias_kit_folder(
|
||||||
|
self,
|
||||||
|
target: PathLike,
|
||||||
|
full_url: str,
|
||||||
|
dir_filter: IliasDirectoryFilter = lambda x, y: True,
|
||||||
|
transform: Transform = lambda x: x,
|
||||||
|
cookies: Optional[PathLike] = None,
|
||||||
|
username: Optional[str] = None,
|
||||||
|
password: Optional[str] = None,
|
||||||
|
download_strategy: IliasDownloadStrategy = download_modified_or_new,
|
||||||
|
clean: bool = True,
|
||||||
|
timeout: int = 5,
|
||||||
|
) -> Organizer:
|
||||||
|
"""
|
||||||
|
Synchronizes a folder with a given folder on the ILIAS instance of the KIT.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
target {Path} -- the target path to write the data to
|
||||||
|
full_url {str} -- the full url of the folder/videos/course to crawl
|
||||||
|
|
||||||
|
Keyword Arguments:
|
||||||
|
dir_filter {IliasDirectoryFilter} -- A filter for directories. Will be applied on the
|
||||||
|
crawler level, these directories and all of their content is skipped.
|
||||||
|
(default: {lambdax:True})
|
||||||
|
transform {Transform} -- A transformation function for the output paths. Return None
|
||||||
|
to ignore a file. (default: {lambdax:x})
|
||||||
|
cookies {Optional[Path]} -- The path to store and load cookies from.
|
||||||
|
(default: {None})
|
||||||
|
username {Optional[str]} -- The SCC username. If none is given, it will prompt
|
||||||
|
the user. (default: {None})
|
||||||
|
password {Optional[str]} -- The SCC password. If none is given, it will prompt
|
||||||
|
the user. (default: {None})
|
||||||
|
download_strategy {DownloadStrategy} -- A function to determine which files need to
|
||||||
|
be downloaded. Can save bandwidth and reduce the number of requests.
|
||||||
|
(default: {download_modified_or_new})
|
||||||
|
clean {bool} -- Whether to clean up when the method finishes.
|
||||||
|
timeout {int} -- The download timeout for opencast videos. Sadly needed due to a
|
||||||
|
requests bug.
|
||||||
|
"""
|
||||||
|
# This authenticator only works with the KIT ilias instance.
|
||||||
|
authenticator = KitShibbolethAuthenticator(username=username, password=password)
|
||||||
|
PRETTY.starting_synchronizer(target, "ILIAS", "An ILIAS element by url")
|
||||||
|
|
||||||
|
if not full_url.startswith("https://ilias.studium.kit.edu"):
|
||||||
|
raise FatalException("Not a valid KIT ILIAS URL")
|
||||||
|
|
||||||
|
organizer = self._ilias(
|
||||||
|
target=target,
|
||||||
|
base_url="https://ilias.studium.kit.edu/",
|
||||||
|
crawl_function=lambda crawler: crawler.recursive_crawl_url(full_url),
|
||||||
|
authenticator=authenticator,
|
||||||
|
cookies=cookies,
|
||||||
|
dir_filter=dir_filter,
|
||||||
|
transform=transform,
|
||||||
|
download_strategy=download_strategy,
|
||||||
|
clean=clean,
|
||||||
|
timeout=timeout
|
||||||
|
)
|
||||||
|
|
||||||
|
self._download_summary.merge(organizer.download_summary)
|
||||||
|
|
||||||
|
return organizer
|
||||||
|
|
||||||
@swallow_and_print_errors
|
@swallow_and_print_errors
|
||||||
def diva_kit(
|
def diva_kit(
|
||||||
self,
|
self,
|
||||||
|
@ -7,8 +7,7 @@ from types import TracebackType
|
|||||||
from typing import Optional, Type
|
from typing import Optional, Type
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
from rich.console import Console, ConsoleOptions, Control, RenderResult
|
from rich.console import Console
|
||||||
from rich.live_render import LiveRender
|
|
||||||
from rich.progress import (BarColumn, DownloadColumn, Progress, TaskID,
|
from rich.progress import (BarColumn, DownloadColumn, Progress, TaskID,
|
||||||
TextColumn, TimeRemainingColumn,
|
TextColumn, TimeRemainingColumn,
|
||||||
TransferSpeedColumn)
|
TransferSpeedColumn)
|
||||||
@ -23,7 +22,8 @@ _progress: Progress = Progress(
|
|||||||
TransferSpeedColumn(),
|
TransferSpeedColumn(),
|
||||||
"•",
|
"•",
|
||||||
TimeRemainingColumn(),
|
TimeRemainingColumn(),
|
||||||
console=Console(file=sys.stdout)
|
console=Console(file=sys.stdout),
|
||||||
|
transient=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -61,18 +61,6 @@ def progress_for(settings: Optional[ProgressSettings]) -> 'ProgressContextManage
|
|||||||
return ProgressContextManager(settings)
|
return ProgressContextManager(settings)
|
||||||
|
|
||||||
|
|
||||||
class _OneLineUp(LiveRender):
|
|
||||||
"""
|
|
||||||
Render a control code for moving one line upwards.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self) -> None:
|
|
||||||
super().__init__("not rendered")
|
|
||||||
|
|
||||||
def __console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
|
|
||||||
yield Control(f"\r\x1b[1A")
|
|
||||||
|
|
||||||
|
|
||||||
class ProgressContextManager:
|
class ProgressContextManager:
|
||||||
"""
|
"""
|
||||||
A context manager used for displaying progress.
|
A context manager used for displaying progress.
|
||||||
@ -113,9 +101,6 @@ class ProgressContextManager:
|
|||||||
_progress.stop()
|
_progress.stop()
|
||||||
_progress.refresh()
|
_progress.refresh()
|
||||||
|
|
||||||
# And we existed, so remove the line above (remove_task leaves one behind)
|
|
||||||
Console().print(_OneLineUp())
|
|
||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def advance(self, amount: float) -> None:
|
def advance(self, amount: float) -> None:
|
||||||
|
213
README.md
213
README.md
@ -2,35 +2,63 @@
|
|||||||
|
|
||||||
**P**rogramm zum **F**lotten, **E**infachen **R**unterladen von **D**ateien
|
**P**rogramm zum **F**lotten, **E**infachen **R**unterladen von **D**ateien
|
||||||
|
|
||||||
|
- [Quickstart with `sync_url`](#quickstart-with-sync_url)
|
||||||
|
- [Installation](#installation)
|
||||||
|
- [Upgrading from 2.0.0 to 2.1.0+](#upgrading-from-200-to-210)
|
||||||
|
- [Example setup](#example-setup)
|
||||||
|
- [Usage](#usage)
|
||||||
|
- [General concepts](#general-concepts)
|
||||||
|
- [Constructing transforms](#constructing-transforms)
|
||||||
|
- [Transform creators](#transform-creators)
|
||||||
|
- [Transform combinators](#transform-combinators)
|
||||||
|
- [A short, but commented example](#a-short-but-commented-example)
|
||||||
|
|
||||||
|
## Quickstart with `sync_url`
|
||||||
|
|
||||||
|
The `sync_url` program allows you to just synchronize a given ILIAS URL (of a
|
||||||
|
course, a folder, your personal desktop, etc.) without any extra configuration
|
||||||
|
or setting up. Download the program, open ILIAS, copy the URL from the address
|
||||||
|
bar and pass it to sync_url.
|
||||||
|
|
||||||
|
It bundles everything it needs in one executable and is easy to
|
||||||
|
use, but doesn't expose all the configuration options and tweaks a full install
|
||||||
|
does.
|
||||||
|
|
||||||
|
1. Download the `sync_url` binary from the [latest release](https://github.com/Garmelon/PFERD/releases/latest).
|
||||||
|
2. Run the binary in your terminal (`./sync_url` or `sync_url.exe` in the CMD) to see the help and use it
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
Ensure that you have at least Python 3.8 installed.
|
Ensure that you have at least Python 3.8 installed.
|
||||||
|
|
||||||
To install PFERD or update your installation to the latest version, run this
|
To install PFERD or update your installation to the latest version, run this
|
||||||
wherever you want to install/have installed PFERD:
|
wherever you want to install or have already installed PFERD:
|
||||||
```
|
```
|
||||||
$ pip install git+https://github.com/Garmelon/PFERD@v2.0.0
|
$ pip install git+https://github.com/Garmelon/PFERD@v2.4.0
|
||||||
```
|
```
|
||||||
|
|
||||||
The use of [venv](https://docs.python.org/3/library/venv.html) is recommended.
|
The use of [venv] is recommended.
|
||||||
|
|
||||||
|
[venv]: https://docs.python.org/3/library/venv.html
|
||||||
|
|
||||||
|
### Upgrading from 2.0.0 to 2.1.0+
|
||||||
|
|
||||||
|
- The `IliasDirectoryType` type was renamed to `IliasElementType` and is now far more detailed.
|
||||||
|
The new values are: `REGULAR_FOLDER`, `VIDEO_FOLDER`, `EXERCISE_FOLDER`, `REGULAR_FILE`, `VIDEO_FILE`, `FORUM`, `EXTERNAL_LINK`.
|
||||||
|
- Forums and external links are skipped automatically if you use the `kit_ilias` helper.
|
||||||
|
|
||||||
## Example setup
|
## Example setup
|
||||||
|
|
||||||
In this example, `python3` refers to at least Python 3.8.
|
In this example, `python3` refers to at least Python 3.8.
|
||||||
|
|
||||||
If you just want to get started and crawl *your entire ILIAS Desktop* instead
|
|
||||||
of a given set of courses, please replace `example_config.py` with
|
|
||||||
`example_config_personal_desktop.py` in all of the instructions below (`curl` call and
|
|
||||||
`python3` run command).
|
|
||||||
|
|
||||||
A full example setup and initial use could look like:
|
A full example setup and initial use could look like:
|
||||||
```
|
```
|
||||||
$ mkdir Vorlesungen
|
$ mkdir Vorlesungen
|
||||||
$ cd Vorlesungen
|
$ cd Vorlesungen
|
||||||
$ python3 -m venv .venv
|
$ python3 -m venv .venv
|
||||||
$ .venv/bin/activate
|
$ .venv/bin/activate
|
||||||
$ pip install git+https://github.com/Garmelon/PFERD@v2.0.0
|
$ pip install git+https://github.com/Garmelon/PFERD@v2.4.0
|
||||||
$ curl -O https://raw.githubusercontent.com/Garmelon/PFERD/v2.0.0/example_config.py
|
$ curl -O https://raw.githubusercontent.com/Garmelon/PFERD/v2.4.0/example_config.py
|
||||||
$ python3 example_config.py
|
$ python3 example_config.py
|
||||||
$ deactivate
|
$ deactivate
|
||||||
```
|
```
|
||||||
@ -43,50 +71,93 @@ $ python3 example_config.py
|
|||||||
$ deactivate
|
$ deactivate
|
||||||
```
|
```
|
||||||
|
|
||||||
|
If you just want to get started and crawl *your entire ILIAS Desktop* instead
|
||||||
|
of a given set of courses, please replace `example_config.py` with
|
||||||
|
`example_config_personal_desktop.py` in all of the instructions below (`curl` call and
|
||||||
|
`python3` run command).
|
||||||
|
|
||||||
## Usage
|
## Usage
|
||||||
|
|
||||||
|
### General concepts
|
||||||
|
|
||||||
A PFERD config is a normal python file that starts multiple *synchronizers*
|
A PFERD config is a normal python file that starts multiple *synchronizers*
|
||||||
which do all the heavy lifting. While you can create and wire them up manually,
|
which do all the heavy lifting. While you can create and wire them up manually,
|
||||||
you are encouraged to use the helper methods provided in `PFERD.Pferd`.
|
you are encouraged to use the helper methods provided in `PFERD.Pferd`.
|
||||||
|
|
||||||
The synchronizers take some input arguments specific to their service and a
|
The synchronizers take some input arguments specific to their service and a
|
||||||
*transformer*. The transformer receives the computed path of an element in
|
*transform*. The transform receives the computed path of an element in ILIAS and
|
||||||
ILIAS and can return either an output path (so you can rename files or move
|
can return either an output path (so you can rename files or move them around as
|
||||||
them around as you wish) or `None` if you do not want to save the given file.
|
you wish) or `None` if you do not want to save the given file.
|
||||||
|
|
||||||
Additionally the ILIAS synchronizer allows you to define a *crawl filter*. This
|
Additionally the ILIAS synchronizer allows you to define a *crawl filter*. This
|
||||||
filter also receives the computed path as the input, but is only called or
|
filter also receives the computed path as the input, but is only called for
|
||||||
*directoties*. If you return `True`, the directory will be crawled and
|
*directories*. If you return `True`, the directory will be crawled and
|
||||||
searched. If you return `False` the directory will be ignored and nothing in it
|
searched. If you return `False` the directory will be ignored and nothing in it
|
||||||
will be passed to the transformer.
|
will be passed to the transform.
|
||||||
|
|
||||||
In order to help you with writing your own transformers and filters, PFERD
|
### Constructing transforms
|
||||||
ships with a few powerful building blocks:
|
|
||||||
|
|
||||||
| Method | Description |
|
While transforms are just normal python functions, writing them by hand can
|
||||||
|--------|-------------|
|
quickly become tedious. In order to help you with writing your own transforms
|
||||||
| `glob` | Returns a transform that returns `None` if the glob does not match and the unmodified path otherwise. |
|
and filters, PFERD defines a few useful transform creators and combinators in
|
||||||
| `predicate` | Returns a transform that returns `None` if the predicate does not match the path and the unmodified path otherwise. |
|
the `PFERD.transform` module:
|
||||||
| `move_dir(source, target)` | Returns a transform that moves all files from the `source` to the `target` dir. |
|
|
||||||
| `move(source, target)` | Returns a transform that moves the `source` file to `target`. |
|
|
||||||
| `rename(old, new)` | Renames a single file. |
|
|
||||||
| `re_move(regex, sub)` | Moves all files matching the given regular expression. The different captured groups are available under their index and can be used together with normal python format methods: `re_move(r"Blatt (\d+)\.pdf", "Blätter/Blatt_{1:0>2}.pdf"),`. |
|
|
||||||
| `re_rename(old, new)` | Same as `re_move` but operates on the path *names* instead of the full path. |
|
|
||||||
|
|
||||||
And PFERD also offers a few combinator functions:
|
#### Transform creators
|
||||||
|
|
||||||
* **`keep`**
|
These methods let you create a few basic transform building blocks:
|
||||||
`keep` just returns the input path unchanged. It can be very useful as the
|
|
||||||
last argument in an `attempt` call, to leave everything not matching a rule
|
- **`glob(glob)`**
|
||||||
unchanged.
|
Creates a transform that returns the unchanged path if the glob matches the path and `None` otherwise.
|
||||||
* **`optionally(transformer)`**
|
See also [Path.match].
|
||||||
Wraps a given transformer and returns its result if it is not `None`.
|
Example: `glob("Übung/*.pdf")`
|
||||||
|
- **`predicate(pred)`**
|
||||||
|
Creates a transform that returns the unchanged path if `pred(path)` returns a truthy value.
|
||||||
|
Returns `None` otherwise.
|
||||||
|
Example: `predicate(lambda path: len(path.parts) == 3)`
|
||||||
|
- **`move_dir(source, target)`**
|
||||||
|
Creates a transform that moves all files from the `source` to the `target` directory.
|
||||||
|
Example: `move_dir("Übung/", "Blätter/")`
|
||||||
|
- **`move(source, target)`**
|
||||||
|
Creates a transform that moves the `source` file to `target`.
|
||||||
|
Example: `move("Vorlesung/VL02_Automten.pdf", "Vorlesung/VL02_Automaten.pdf")`
|
||||||
|
- **`rename(source, target)`**
|
||||||
|
Creates a transform that renames all files named `source` to `target`.
|
||||||
|
This transform works on the file names, not paths, and thus works no matter where the file is located.
|
||||||
|
Example: `rename("VL02_Automten.pdf", "VL02_Automaten.pdf")`
|
||||||
|
- **`re_move(regex, target)`**
|
||||||
|
Creates a transform that moves all files matching `regex` to `target`.
|
||||||
|
The transform `str.format` on the `target` string with the contents of the capturing groups before returning it.
|
||||||
|
The capturing groups can be accessed via their index.
|
||||||
|
See also [Match.group].
|
||||||
|
Example: `re_move(r"Übung/Blatt (\d+)\.pdf", "Blätter/Blatt_{1:0>2}.pdf")`
|
||||||
|
- **`re_rename(regex, target)`**
|
||||||
|
Creates a transform that renames all files matching `regex` to `target`.
|
||||||
|
This transform works on the file names, not paths, and thus works no matter where the file is located.
|
||||||
|
Example: `re_rename(r"VL(\d+)(.*)\.pdf", "Vorlesung_Nr_{1}__{2}.pdf")`
|
||||||
|
|
||||||
|
All movement or rename transforms above return `None` if a file doesn't match
|
||||||
|
their movement or renaming criteria. This enables them to be used as building
|
||||||
|
blocks to build up more complex transforms.
|
||||||
|
|
||||||
|
In addition, `PFERD.transform` also defines the `keep` transform which returns its input path unchanged.
|
||||||
|
This behaviour can be very useful when creating more complex transforms.
|
||||||
|
See below for example usage.
|
||||||
|
|
||||||
|
[Path.match]: https://docs.python.org/3/library/pathlib.html#pathlib.Path.match
|
||||||
|
[Match.group]: https://docs.python.org/3/library/re.html#re.Match.group
|
||||||
|
|
||||||
|
#### Transform combinators
|
||||||
|
|
||||||
|
These methods let you combine transforms into more complex transforms:
|
||||||
|
|
||||||
|
- **`optionally(transform)`**
|
||||||
|
Wraps a given transform and returns its result if it is not `None`.
|
||||||
Otherwise returns the input path unchanged.
|
Otherwise returns the input path unchanged.
|
||||||
* **`do(transformers)`**
|
See below for example usage.
|
||||||
`do` accepts a series of transformers and applies them in the given order to
|
* **`do(transforms)`**
|
||||||
the result of the previous one. If any transformer returns `None`, do
|
Accepts a series of transforms and applies them in the given order to the result of the previous one.
|
||||||
short-circuits and also returns `None`. This can be used to perform multiple
|
If any transform returns `None`, `do` short-circuits and also returns `None`.
|
||||||
renames in a row:
|
This can be used to perform multiple renames in a row:
|
||||||
```py
|
```py
|
||||||
do(
|
do(
|
||||||
# Move them
|
# Move them
|
||||||
@ -95,13 +166,12 @@ And PFERD also offers a few combinator functions:
|
|||||||
optionally(re_rename("(.*).m4v.mp4", "{1}.mp4")),
|
optionally(re_rename("(.*).m4v.mp4", "{1}.mp4")),
|
||||||
# Remove the 'dbs' prefix (if they have any)
|
# Remove the 'dbs' prefix (if they have any)
|
||||||
optionally(re_rename("(?i)dbs-(.+)", "{1}")),
|
optionally(re_rename("(?i)dbs-(.+)", "{1}")),
|
||||||
),
|
)
|
||||||
```
|
```
|
||||||
* **`attempt(transformers)`**
|
- **`attempt(transforms)`**
|
||||||
`attempt` applies the passed transformers in the given order until it finds
|
Applies the passed transforms in the given order until it finds one that does not return `None`.
|
||||||
one that does not return `None`. If it does not find any, it returns `None`.
|
If it does not find any, it returns `None`.
|
||||||
This can be used to give a list of possible transformations and it will
|
This can be used to give a list of possible transformations and automatically pick the first one that fits:
|
||||||
automatically pick the first one that fits:
|
|
||||||
```py
|
```py
|
||||||
attempt(
|
attempt(
|
||||||
# Move all videos. If a video is passed in, this `re_move` will succeed
|
# Move all videos. If a video is passed in, this `re_move` will succeed
|
||||||
@ -114,17 +184,26 @@ And PFERD also offers a few combinator functions:
|
|||||||
)
|
)
|
||||||
```
|
```
|
||||||
|
|
||||||
All of these combinators are used in the provided example config, if you want
|
All of these combinators are used in the provided example configs, if you want
|
||||||
to see some more true-to-live usages.
|
to see some more real-life usages.
|
||||||
|
|
||||||
### A short, but commented example
|
### A short, but commented example
|
||||||
|
|
||||||
```py
|
```py
|
||||||
def filter_course(path: PurePath) -> bool:
|
from pathlib import Path, PurePath
|
||||||
# Note that glob returns a Transformer
|
from PFERD import Pferd
|
||||||
# - a function from PurePath -> Optional[PurePath]
|
from PFERD.ilias import IliasElementType
|
||||||
# So we need to apply the result of 'glob' to our input path.
|
from PFERD.transform import *
|
||||||
# We need to crawl the 'Tutorien' folder as it contains the one we want.
|
|
||||||
|
# This filter will later be used by the ILIAS crawler to decide whether it
|
||||||
|
# should crawl a directory (or directory-like structure).
|
||||||
|
def filter_course(path: PurePath, type: IliasElementType) -> bool:
|
||||||
|
# Note that glob returns a Transform, which is a function from PurePath ->
|
||||||
|
# Optional[PurePath]. Because of this, we need to apply the result of
|
||||||
|
# 'glob' to our input path. The returned value will be truthy (a Path) if
|
||||||
|
# the transform succeeded, or `None` if it failed.
|
||||||
|
|
||||||
|
# We need to crawl the 'Tutorien' folder as it contains one that we want.
|
||||||
if glob("Tutorien/")(path):
|
if glob("Tutorien/")(path):
|
||||||
return True
|
return True
|
||||||
# If we found 'Tutorium 10', keep it!
|
# If we found 'Tutorium 10', keep it!
|
||||||
@ -137,21 +216,35 @@ def filter_course(path: PurePath) -> bool:
|
|||||||
# All other dirs (including subdirs of 'Tutorium 10') should be searched :)
|
# All other dirs (including subdirs of 'Tutorium 10') should be searched :)
|
||||||
return True
|
return True
|
||||||
|
|
||||||
enable_logging() # needed once before calling a Pferd method
|
|
||||||
# Create a Pferd instance rooted in the same directory as the script file
|
# This transform will later be used to rename a few files. It can also be used
|
||||||
# This is not a test run, so files will be downloaded (default, can be omitted)
|
# to ignore some files.
|
||||||
|
transform_course = attempt(
|
||||||
|
# We don't care about the other tuts and would instead prefer a cleaner
|
||||||
|
# directory structure.
|
||||||
|
move_dir("Tutorien/Tutorium 10/", "Tutorium/"),
|
||||||
|
# We don't want to modify any other files, so we're going to keep them
|
||||||
|
# exactly as they are.
|
||||||
|
keep
|
||||||
|
)
|
||||||
|
|
||||||
|
# Enable and configure the text output. Needs to be called before calling any
|
||||||
|
# other PFERD methods.
|
||||||
|
Pferd.enable_logging()
|
||||||
|
# Create a Pferd instance rooted in the same directory as the script file. This
|
||||||
|
# is not a test run, so files will be downloaded (default, can be omitted).
|
||||||
pferd = Pferd(Path(__file__).parent, test_run=False)
|
pferd = Pferd(Path(__file__).parent, test_run=False)
|
||||||
|
|
||||||
# Use the ilias_kit helper to synchronize an ILIAS course
|
# Use the ilias_kit helper to synchronize an ILIAS course
|
||||||
pferd.ilias_kit(
|
pferd.ilias_kit(
|
||||||
# The folder all of the course's content should be placed in
|
# The directory that all of the downloaded files should be placed in
|
||||||
Path("My cool course"),
|
"My_cool_course/",
|
||||||
# The course ID (found in the URL when on the course page in ILIAS)
|
# The course ID (found in the URL when on the course page in ILIAS)
|
||||||
"course id",
|
"course id",
|
||||||
# A path to a cookie jar. If you synchronize multiple ILIAS courses, setting this
|
# A path to a cookie jar. If you synchronize multiple ILIAS courses,
|
||||||
# to a common value requires you to only login once.
|
# setting this to a common value requires you to only log in once.
|
||||||
cookies=Path("ilias_cookies.txt"),
|
cookies=Path("ilias_cookies.txt"),
|
||||||
# A transform to apply to all found paths
|
# A transform can rename, move or filter out certain files
|
||||||
transform=transform_course,
|
transform=transform_course,
|
||||||
# A crawl filter limits what paths the cralwer searches
|
# A crawl filter limits what paths the cralwer searches
|
||||||
dir_filter=filter_course,
|
dir_filter=filter_course,
|
||||||
|
@ -124,6 +124,8 @@ def main() -> None:
|
|||||||
cookies="ilias_cookies.txt",
|
cookies="ilias_cookies.txt",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Prints a summary listing all new, modified or deleted files
|
||||||
|
pferd.print_summary()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
@ -30,6 +30,9 @@ def main() -> None:
|
|||||||
cookies="ilias_cookies.txt",
|
cookies="ilias_cookies.txt",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Prints a summary listing all new, modified or deleted files
|
||||||
|
pferd.print_summary()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
4
setup.py
4
setup.py
@ -2,12 +2,12 @@ from setuptools import find_packages, setup
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name="PFERD",
|
name="PFERD",
|
||||||
version="2.1.0",
|
version="2.4.0",
|
||||||
packages=find_packages(),
|
packages=find_packages(),
|
||||||
install_requires=[
|
install_requires=[
|
||||||
"requests>=2.21.0",
|
"requests>=2.21.0",
|
||||||
"beautifulsoup4>=4.7.1",
|
"beautifulsoup4>=4.7.1",
|
||||||
"rich>=1.0.0"
|
"rich>=2.1.0"
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
67
sync_url.py
Executable file
67
sync_url.py
Executable file
@ -0,0 +1,67 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
"""
|
||||||
|
A simple script to download a course by name from ILIAS.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
from pathlib import Path
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
|
from PFERD import Pferd
|
||||||
|
from PFERD.cookie_jar import CookieJar
|
||||||
|
from PFERD.ilias import (IliasCrawler, IliasElementType,
|
||||||
|
KitShibbolethAuthenticator)
|
||||||
|
from PFERD.utils import to_path
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument("--test-run", action="store_true")
|
||||||
|
parser.add_argument('-c', '--cookies', nargs='?', default=None, help="File to store cookies in")
|
||||||
|
parser.add_argument('-f', '--no-videos', nargs='?', default=None, help="Don't download videos")
|
||||||
|
parser.add_argument('url', help="URL to the course page")
|
||||||
|
parser.add_argument('folder', nargs='?', default=None, help="Folder to put stuff into")
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
url = urlparse(args.url)
|
||||||
|
|
||||||
|
cookie_jar = CookieJar(to_path(args.cookies) if args.cookies else None)
|
||||||
|
session = cookie_jar.create_session()
|
||||||
|
authenticator = KitShibbolethAuthenticator()
|
||||||
|
crawler = IliasCrawler(url.scheme + '://' + url.netloc, session,
|
||||||
|
authenticator, lambda x, y: True)
|
||||||
|
|
||||||
|
cookie_jar.load_cookies()
|
||||||
|
|
||||||
|
if args.folder is not None:
|
||||||
|
folder = args.folder
|
||||||
|
# Initialize pferd at the *parent of the passed folder*
|
||||||
|
# This is needed so Pferd's internal protections against escaping the working directory
|
||||||
|
# do not trigger (e.g. if somebody names a file in ILIAS '../../bad thing.txt')
|
||||||
|
pferd = Pferd(Path(Path(__file__).parent, folder).parent, test_run=args.test_run)
|
||||||
|
else:
|
||||||
|
# fetch course name from ilias
|
||||||
|
folder = crawler.find_element_name(args.url)
|
||||||
|
cookie_jar.save_cookies()
|
||||||
|
|
||||||
|
# Initialize pferd at the location of the script
|
||||||
|
pferd = Pferd(Path(__file__).parent, test_run=args.test_run)
|
||||||
|
|
||||||
|
def dir_filter(_: Path, element: IliasElementType) -> bool:
|
||||||
|
if args.no_videos:
|
||||||
|
return element not in [IliasElementType.VIDEO_FILE, IliasElementType.VIDEO_FOLDER]
|
||||||
|
return True
|
||||||
|
|
||||||
|
pferd.enable_logging()
|
||||||
|
# fetch
|
||||||
|
pferd.ilias_kit_folder(
|
||||||
|
target=folder,
|
||||||
|
full_url=args.url,
|
||||||
|
cookies=args.cookies,
|
||||||
|
dir_filter=dir_filter
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Reference in New Issue
Block a user