56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
#!/usr/bin/python3
|
|
import re
|
|
import subprocess
|
|
import warnings
|
|
from pathlib import Path
|
|
from window_subprocess import open_pdf, edit
|
|
from enum import Enum
|
|
|
|
|
|
class FileType(Enum):
|
|
pdf = 'pdf'
|
|
tex = 'tex'
|
|
|
|
|
|
class FileHandle:
|
|
def __init__(self, file_path: Path, file_type: FileType = FileType.pdf):
|
|
self.path = file_path
|
|
self.file_type = file_type
|
|
match = re.match(r'[\D]*(\d+)[\D]*\.' + file_type.value, file_path.name)
|
|
if match is None:
|
|
warnings.warn(f'Invalid format in file {str(file_path)}: Could not parse number.')
|
|
self.number = -1
|
|
else:
|
|
self.number = int(match.group(1))
|
|
|
|
def open(self):
|
|
if self.file_type == FileType.pdf:
|
|
open_pdf(self.path)
|
|
return 0
|
|
elif self.file_type == FileType.tex:
|
|
edit(self.path)
|
|
return 0
|
|
return 1
|
|
|
|
def edit(self):
|
|
if self.file_type == FileType.tex:
|
|
edit(self.path)
|
|
return 0
|
|
else:
|
|
return 1
|
|
|
|
|
|
class Files(list):
|
|
def __init__(self, root_path: Path, pattern: str = "*"):
|
|
self.root: Path = root_path
|
|
list.__init__(self, self.read_files(pattern))
|
|
|
|
def read_files(self, pattern):
|
|
files = self.root.glob(pattern)
|
|
return sorted((FileHandle(f) for f in files), key=lambda f: f.number)
|
|
|
|
def unite_files(self, name):
|
|
result = subprocess.run(
|
|
['pdfunite'] + list(map(lambda f: str(f.path), self)) + [str(self.root / name)]
|
|
)
|
|
print(result.stdout)
|