46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
import pathlib
|
|
import shutil
|
|
import string
|
|
|
|
from models import *
|
|
|
|
|
|
def format_filename(s: str) -> str:
|
|
valid_chars = "-_.() %s%s" % (string.ascii_letters, string.digits)
|
|
filename = ''.join(c for c in s if c in valid_chars)
|
|
# filename = filename.replace(' ', '_') # I don't like spaces in filenames.
|
|
return filename
|
|
|
|
|
|
def create_library(librarydir: pathlib.Path, browsedir: pathlib.Path):
|
|
shutil.rmtree(browsedir)
|
|
browsedir.mkdir()
|
|
print("title")
|
|
title_dir = browsedir / "title"
|
|
title_dir.mkdir(exist_ok=True)
|
|
for paper in Paper.select():
|
|
sourcefile = librarydir / "{}.pdf".format(paper.id)
|
|
targetfile = title_dir / "{}.pdf".format(format_filename(paper.title))
|
|
targetfile.symlink_to(sourcefile)
|
|
|
|
print("author")
|
|
author_dir = browsedir / "authors"
|
|
author_dir.mkdir(exist_ok=True)
|
|
for author in Author.select():
|
|
author_subdir = author_dir / format_filename(author.name)
|
|
author_subdir.mkdir()
|
|
for paper in Paper.select().join(PaperAuthors).where(PaperAuthors.author == author):
|
|
sourcefile = librarydir / "{}.pdf".format(paper.id)
|
|
targetfile = author_subdir / "{}.pdf".format(format_filename(paper.title))
|
|
targetfile.symlink_to(sourcefile)
|
|
|
|
print("keywords")
|
|
keywords_dir = browsedir / "keywords"
|
|
keywords_dir.mkdir(exist_ok=True)
|
|
for keyword in Keyword.select():
|
|
keyword_subdir = keywords_dir / format_filename(keyword.keyword)
|
|
keyword_subdir.mkdir()
|
|
for paper in Paper.select().join(PaperKeywords).where(PaperKeywords.keyword == keyword):
|
|
sourcefile = librarydir / "{}.pdf".format(paper.id)
|
|
targetfile = keyword_subdir / "{}.pdf".format(format_filename(paper.title))
|
|
targetfile.symlink_to(sourcefile)
|